1/***********************************************************************
2*                                                                      *
3*               This software is part of the ast package               *
4*          Copyright (c) 1982-2010 AT&T Intellectual Property          *
5*                      and is licensed under the                       *
6*                  Common Public License, Version 1.0                  *
7*                    by AT&T Intellectual Property                     *
8*                                                                      *
9*                A copy of the License is available at                 *
10*            http://www.opensource.org/licenses/cpl1.0.txt             *
11*         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
12*                                                                      *
13*              Information and Software Systems Research               *
14*                            AT&T Research                             *
15*                           Florham Park NJ                            *
16*                                                                      *
17*                  David Korn <dgk@research.att.com>                   *
18*                                                                      *
19***********************************************************************/
20#pragma prototyped
21/*
22 * break [n]
23 * continue [n]
24 * return [n]
25 * exit [n]
26 *
27 *   David Korn
28 *   AT&T Labs
29 *   dgk@research.att.com
30 *
31 */
32
33#include	"defs.h"
34#include	<ast.h>
35#include	<error.h>
36#include	"shnodes.h"
37#include	"builtins.h"
38
39/*
40 * return and exit
41 */
42#if 0
43    /* for the dictionary generator */
44    int	b_exit(int n, register char *argv[],void *extra){}
45#endif
46int	b_return(register int n, register char *argv[],void *extra)
47{
48	register char *arg;
49	register Shell_t *shp = ((Shbltin_t*)extra)->shp;
50	struct checkpt *pp = (struct checkpt*)shp->jmplist;
51	const char *options = (**argv=='r'?sh_optreturn:sh_optexit);
52	while((n = optget(argv,options))) switch(n)
53	{
54	    case ':':
55		if(!strmatch(argv[opt_info.index],"[+-]+([0-9])"))
56			errormsg(SH_DICT,2, "%s", opt_info.arg);
57		goto done;
58	    case '?':
59		errormsg(SH_DICT,ERROR_usage(0), "%s", opt_info.arg);
60		return(2);
61	}
62done:
63	if(error_info.errors)
64		errormsg(SH_DICT,ERROR_usage(2),"%s",optusage((char*)0));
65	pp->mode = (**argv=='e'?SH_JMPEXIT:SH_JMPFUN);
66	argv += opt_info.index;
67	n = (((arg= *argv)?(int)strtol(arg, (char**)0, 10)&SH_EXITMASK:shp->oldexit));
68	/* return outside of function, dotscript and profile is exit */
69	if(shp->fn_depth==0 && shp->dot_depth==0 && !sh_isstate(SH_PROFILE))
70		pp->mode = SH_JMPEXIT;
71	sh_exit(shp->savexit=n);
72	return(1);
73}
74
75
76/*
77 * break and continue
78 */
79#if 0
80    /* for the dictionary generator */
81    int	b_continue(int n, register char *argv[],void *extra){}
82#endif
83int	b_break(register int n, register char *argv[],void *extra)
84{
85	char *arg;
86	register int cont= **argv=='c';
87	register Shell_t *shp = ((Shbltin_t*)extra)->shp;
88	while((n = optget(argv,cont?sh_optcont:sh_optbreak))) switch(n)
89	{
90	    case ':':
91		errormsg(SH_DICT,2, "%s", opt_info.arg);
92		break;
93	    case '?':
94		errormsg(SH_DICT,ERROR_usage(0), "%s", opt_info.arg);
95		return(2);
96	}
97	if(error_info.errors)
98		errormsg(SH_DICT,ERROR_usage(2),"%s",optusage((char*)0));
99	argv += opt_info.index;
100	n=1;
101	if(arg= *argv)
102	{
103		n = strtol(arg,&arg,10);
104		if(n<=0 || *arg)
105			errormsg(SH_DICT,ERROR_exit(1),e_nolabels,*argv);
106	}
107	if(shp->st.loopcnt)
108	{
109		shp->st.execbrk = shp->st.breakcnt = n;
110		if(shp->st.breakcnt > shp->st.loopcnt)
111			shp->st.breakcnt = shp->st.loopcnt;
112		if(cont)
113			shp->st.breakcnt = -shp->st.breakcnt;
114	}
115	return(0);
116}
117
118