nail.h: fix sorting of #include<inttypes.h>
[s-mailx.git] / accmacvar.c
blob4a1d0f427b2acc7cabcf2665ff86181c08d8a8be
1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
2 *@ Account, macro and variable handling.
3 *@ HOWTO add a new non-dynamic boolean or value option:
4 *@ - add an entry to nail.h:enum okeys
5 *@ - run make-okey-map.pl
6 *@ - update the manual!
7 *@ TODO . Improve support for chains so that we can apply the property checks
8 *@ TODO of the base variable to -HOST and -USER@HOST variants! Add some!!
9 *@ TODO . once we can have non-fatal !0 returns for commands, we should
10 *@ TODO return error if "(environ)? unset" goes for non-existent.
11 *@ TODO . should be recursive environment based.
12 *@ TODO Otherwise, the `localopts' should be an attribute of the go.c
13 *@ TODO command context, so that it belongs to the execution context
14 *@ TODO we are running in, instead of being global data. See, e.g.,
15 *@ TODO the a_GO_SPLICE comment in go.c.
17 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
18 * Copyright (c) 2012 - 2017 Steffen (Daode) Nurpmeso <steffen@sdaoden.eu>.
21 * Copyright (c) 1980, 1993
22 * The Regents of the University of California. All rights reserved.
24 * Redistribution and use in source and binary forms, with or without
25 * modification, are permitted provided that the following conditions
26 * are met:
27 * 1. Redistributions of source code must retain the above copyright
28 * notice, this list of conditions and the following disclaimer.
29 * 2. Redistributions in binary form must reproduce the above copyright
30 * notice, this list of conditions and the following disclaimer in the
31 * documentation and/or other materials provided with the distribution.
32 * 3. Neither the name of the University nor the names of its contributors
33 * may be used to endorse or promote products derived from this software
34 * without specific prior written permission.
36 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
37 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
39 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
40 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
41 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
42 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
43 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
44 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
45 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
46 * SUCH DAMAGE.
48 #undef n_FILE
49 #define n_FILE accmacvar
51 #ifndef HAVE_AMALGAMATION
52 # include "nail.h"
53 #endif
55 #if !defined HAVE_SETENV && !defined HAVE_PUTENV
56 # error Exactly one of HAVE_SETENV and HAVE_PUTENV
57 #endif
59 /* Positional parameter maximum (macro arguments, `vexpr' "splitifs") */
60 #define a_AMV_POSPAR_MAX SI16_MAX
62 /* Special "pseudo macro" that stabs you from the back */
63 #define a_AMV_MACKY_MACK ((struct a_amv_mac*)-1)
65 /* Note: changing the hash function must be reflected in `vexpr' "hash",
66 * because that is used by the hashtable creator scripts! */
67 #define a_AMV_PRIME HSHSIZE
68 #define a_AMV_NAME2HASH(N) n_torek_hash(N)
69 #define a_AMV_HASH2PRIME(H) ((H) % a_AMV_PRIME)
71 enum a_amv_mac_flags{
72 a_AMV_MF_NONE = 0,
73 a_AMV_MF_ACCOUNT = 1u<<0, /* This macro is an `account' */
74 a_AMV_MF_TYPE_MASK = a_AMV_MF_ACCOUNT,
75 a_AMV_MF_UNDEF = 1u<<1, /* Unlink after lookup */
76 a_AMV_MF_DELETE = 1u<<7, /* Delete in progress, free once refcnt==0 */
77 a_AMV_MF__MAX = 0xFFu
80 enum a_amv_loflags{
81 a_AMV_LF_NONE = 0,
82 a_AMV_LF_SCOPE = 1u<<0, /* Current scope `localopts' on */
83 a_AMV_LF_SCOPE_FIXATE = 1u<<1, /* Ditto, but fixated */
84 a_AMV_LF_SCOPE_MASK = a_AMV_LF_SCOPE | a_AMV_LF_SCOPE_FIXATE,
85 a_AMV_LF_CALL = 1u<<2, /* `localopts' on for `call'ed scopes */
86 a_AMV_LF_CALL_FIXATE = 1u<<3, /* Ditto, but fixated */
87 a_AMV_LF_CALL_MASK = a_AMV_LF_CALL | a_AMV_LF_CALL_FIXATE,
88 a_AMV_LF_CALL_TO_SCOPE_SHIFT = 2
91 /* make-okey-map.pl ensures that _VIRT implies _RDONLY and _NODEL, and that
92 * _IMPORT implies _ENV; it doesn't verify anything... */
93 enum a_amv_var_flags{
94 a_AMV_VF_NONE = 0,
96 /* The basic set of flags, also present in struct a_amv_var_map.avm_flags */
97 a_AMV_VF_BOOL = 1u<<0, /* ok_b_* */
98 a_AMV_VF_VIRT = 1u<<1, /* "Stateless" automatic variable */
99 a_AMV_VF_CHAIN = 1u<<2, /* Is a variable chain (-USER{,@HOST} variants) */
100 a_AMV_VF_VIP = 1u<<3, /* Wants _var_check_vips() evaluation */
101 a_AMV_VF_RDONLY = 1u<<4, /* May not be set by user */
102 a_AMV_VF_NODEL = 1u<<5, /* May not be deleted */
103 a_AMV_VF_I3VAL = 1u<<6, /* Has an initial value */
104 a_AMV_VF_DEFVAL = 1u<<7, /* Has a default value */
105 a_AMV_VF_IMPORT = 1u<<8, /* Import ONLY from environ (pre n_PSO_STARTED) */
106 a_AMV_VF_ENV = 1u<<9, /* Update environment on change */
107 a_AMV_VF_NOLOPTS = 1u<<10, /* May not be tracked by `localopts' */
108 a_AMV_VF_NOTEMPTY = 1u<<11, /* May not be assigned an empty value */
109 a_AMV_VF_NOCNTRLS = 1u<<12, /* Value may not contain control characters */
110 a_AMV_VF_NUM = 1u<<13, /* Value must be a 32-bit number */
111 a_AMV_VF_POSNUM = 1u<<14, /* Value must be positive 32-bit number */
112 a_AMV_VF_LOWER = 1u<<15, /* Values will be stored in a lowercase version */
113 a_AMV_VF__MASK = (1u<<(15+1)) - 1,
115 /* Extended flags, not part of struct a_amv_var_map.avm_flags */
116 a_AMV_VF_EXT_LINKED = 1u<<24, /* `environ' link'ed */
117 a_AMV_VF_EXT_FROZEN = 1u<<25, /* Has been set by -S,.. */
118 a_AMV_VF_EXT_FROZEN_UNSET = 1u<<26, /* ..and was used to unset a variable */
119 a_AMV_VF_EXT__FROZEN_MASK = a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET,
120 a_AMV_VF_EXT__MASK = (1u<<(26+1)) - 1
123 /* We support some special parameter names for one(+)-letter variable names;
124 * note these have counterparts in the code that manages shell expansion!
125 * All these special variables are solely backed by n_var_vlook(), and besides
126 * there is only a_amv_var_revlookup() which knows about them */
127 enum a_amv_var_special_category{
128 a_AMV_VSC_NONE, /* Normal variable, no special treatment */
129 a_AMV_VSC_GLOBAL, /* ${[?!]} are specially mapped, but global */
130 a_AMV_VSC_MULTIPLEX, /* ${^.*} circumflex accent multiplexer */
131 a_AMV_VSC_POSPAR, /* ${[1-9][0-9]*} positional parameters */
132 a_AMV_VSC_POSPAR_ENV /* ${[*@#]} positional parameter support variables */
135 enum a_amv_var_special_type{
136 /* _VSC_GLOBAL */
137 a_AMV_VST_QM, /* ? */
138 a_AMV_VST_EM, /* ! */
139 /* _VSC_MULTIPLEX */
140 /* This is special in that it is a multiplex indicator, the ^ is followed by
141 * a normal variable */
142 a_AMV_VST_CACC, /* ^ (circumflex accent) */
143 /* _VSC_POSPAR_ENV */
144 a_AMV_VST_STAR, /* * */
145 a_AMV_VST_AT, /* @ */
146 a_AMV_VST_NOSIGN /* # */
149 enum a_amv_var_vip_mode{
150 a_AMV_VIP_SET_PRE,
151 a_AMV_VIP_SET_POST,
152 a_AMV_VIP_CLEAR
155 struct a_amv_pospar{
156 ui16_t app_maxcount; /* == slots in .app_dat */
157 ui16_t app_count; /* Maximum a_AMV_POSPAR_MAX */
158 ui16_t app_idx; /* `shift' moves this one, decs .app_count */
159 bool_t app_not_heap; /* .app_dat stuff not on dynamically allocated */
160 ui8_t app__dummy[1];
161 char const **app_dat; /* NULL terminated (for "$@" explosion support) */
163 n_CTA(a_AMV_POSPAR_MAX <= SI16_MAX, "Limit exceeds datatype capabilities");
165 struct a_amv_mac{
166 struct a_amv_mac *am_next;
167 ui32_t am_maxlen; /* of any line in .am_line_dat */
168 ui32_t am_line_cnt; /* of *.am_line_dat (but NULL terminated) */
169 struct a_amv_mac_line **am_line_dat; /* TODO use deque? */
170 struct a_amv_var *am_lopts; /* `localopts' unroll list */
171 ui32_t am_refcnt; /* 0-based for `un{account,define}' purposes */
172 ui8_t am_flags; /* enum a_amv_mac_flags */
173 char am_name[n_VFIELD_SIZE(3)]; /* of this macro */
175 n_CTA(a_AMV_MF__MAX <= UI8_MAX, "Enumeration excesses storage datatype");
177 struct a_amv_mac_line{
178 ui32_t aml_len;
179 ui32_t aml_prespc; /* Number of leading SPC, for display purposes */
180 char aml_dat[n_VFIELD_SIZE(0)];
183 struct a_amv_mac_call_args{
184 char const *amca_name; /* For MACKY_MACK, this is *0*! */
185 struct a_amv_mac *amca_amp; /* "const", but for am_refcnt */
186 struct a_amv_var **amca_unroller;
187 void (*amca_hook_pre)(void *);
188 void *amca_hook_arg;
189 ui8_t amca_loflags;
190 bool_t amca_ps_hook_mask;
191 bool_t amca_no_xcall; /* We want n_GO_INPUT_NO_XCALL for this */
192 ui8_t amca__pad[5];
193 struct a_amv_pospar amca_pospar;
196 struct a_amv_lostack{
197 struct a_amv_lostack *as_global_saved; /* Saved global XXX due to jump */
198 struct a_amv_mac_call_args *as_amcap;
199 struct a_amv_lostack *as_up; /* Outer context */
200 struct a_amv_var *as_lopts;
201 ui8_t as_loflags; /* enum a_amv_mac_loflags */
202 ui8_t avs__pad[7];
205 struct a_amv_var{
206 struct a_amv_var *av_link;
207 char *av_value;
208 #ifdef HAVE_PUTENV
209 char *av_env; /* Actively managed putenv(3) memory, or NULL */
210 #endif
211 ui32_t av_flags; /* enum a_amv_var_flags inclusive extended bits */
212 char av_name[n_VFIELD_SIZE(4)];
214 n_CTA(a_AMV_VF_EXT__MASK <= UI32_MAX, "Enumeration excesses storage datatype");
216 struct a_amv_var_map{
217 ui32_t avm_hash;
218 ui16_t avm_keyoff;
219 ui16_t avm_flags; /* enum a_amv_var_flags without extended bits */
221 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
223 struct a_amv_var_virt{
224 ui32_t avv_okey;
225 ui8_t avv__dummy[4];
226 struct a_amv_var const *avv_var;
229 struct a_amv_var_defval{
230 ui32_t avdv_okey;
231 ui8_t avdv__pad[4];
232 char const *avdv_value; /* Only for !BOOL (otherwise plain existence) */
235 struct a_amv_var_carrier{
236 char const *avc_name;
237 ui32_t avc_hash;
238 ui32_t avc_prime;
239 struct a_amv_var *avc_var;
240 struct a_amv_var_map const *avc_map;
241 enum okeys avc_okey;
242 ui8_t avc__pad[1];
243 ui8_t avc_special_cat;
244 /* Numerical parameter name if .avc_special_cat=a_AMV_VSC_POSPAR,
245 * otherwise a enum a_amv_var_special_type */
246 ui16_t avc_special_prop;
249 /* Include constant make-okey-map.pl output, and the generated version data */
250 #include <gen-version.h>
251 #include <gen-okeys.h>
253 /* The currently active account */
254 static struct a_amv_mac *a_amv_acc_curr;
256 static struct a_amv_mac *a_amv_macs[a_AMV_PRIME]; /* TODO dynamically spaced */
258 /* Unroll list of currently running macro stack */
259 static struct a_amv_lostack *a_amv_lopts;
261 static struct a_amv_var *a_amv_vars[a_AMV_PRIME]; /* TODO dynamically spaced */
263 /* Global (i.e., non-local) a_AMV_VSC_POSPAR stack */
264 static struct a_amv_pospar a_amv_pospar;
266 /* TODO We really deserve localopts support for *folder-hook*s, so hack it in
267 * TODO today via a static lostack, it should be a field in mailbox, once that
268 * TODO is a real multi-instance object */
269 static struct a_amv_var *a_amv_folder_hook_lopts;
271 /* TODO Rather ditto (except for storage -> cmd_ctx), compose hooks */
272 static struct a_amv_var *a_amv_compose_lopts;
274 /* Lookup for macros/accounts: if newamp is not NULL it will be linked in the
275 * map, if _MF_UNDEF is set a possibly existing entry will be removed (first).
276 * Returns NULL if a lookup failed, or if newamp was set, the found entry in
277 * plain lookup cases or when _UNDEF was performed on a currently active entry
278 * (the entry will have been unlinked, and the _MF_DELETE will be honoured once
279 * the reference count reaches 0), and (*)-1 if an _UNDEF was performed */
280 static struct a_amv_mac *a_amv_mac_lookup(char const *name,
281 struct a_amv_mac *newamp, enum a_amv_mac_flags amf);
283 /* `call', `call_if' (and `xcall' via go.c -> c_call()) */
284 static int a_amv_mac_call(void *v, bool_t silent_nexist);
286 /* Execute a macro; amcap must reside in LOFI memory */
287 static bool_t a_amv_mac_exec(struct a_amv_mac_call_args *amcap);
289 static void a_amv_mac__finalize(void *vp);
291 /* User display helpers */
292 static bool_t a_amv_mac_show(enum a_amv_mac_flags amf);
294 /* _def() returns error for faulty definitions and already existing * names,
295 * _undef() returns error if a named thing doesn't exist */
296 static bool_t a_amv_mac_def(char const *name, enum a_amv_mac_flags amf);
297 static bool_t a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf);
299 /* */
300 static void a_amv_mac_free(struct a_amv_mac *amp);
302 /* Update replay-log */
303 static void a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
304 struct a_amv_var *oavp);
305 static void a_amv_lopts_unroll(struct a_amv_var **avpp);
307 /* Special cased value string allocation */
308 static char *a_amv_var_copy(char const *str);
309 static void a_amv_var_free(char *cp);
311 /* Check for special housekeeping. _VIP_SET_POST and _VIP_CLEAR do not fail
312 * (or propagate errors), _VIP_SET_PRE may and should case abortion */
313 static bool_t a_amv_var_check_vips(enum a_amv_var_vip_mode avvm,
314 enum okeys okey, char const *val);
316 /* _VF_NOCNTRLS, _VF_NUM / _VF_POSNUM */
317 static bool_t a_amv_var_check_nocntrls(char const *val);
318 static bool_t a_amv_var_check_num(char const *val, bool_t posnum);
320 /* If a variable name begins with a lowercase-character and contains at
321 * least one '@', it is converted to all-lowercase. This is necessary
322 * for lookups of names based on email addresses.
323 * Following the standard, only the part following the last '@' should
324 * be lower-cased, but practice has established otherwise here */
325 static char const *a_amv_var_canonify(char const *vn);
327 /* Try to reverse lookup an option name to an enum okeys mapping.
328 * Updates .avc_name and .avc_hash; .avc_map is NULL if none found */
329 static bool_t a_amv_var_revlookup(struct a_amv_var_carrier *avcp,
330 char const *name);
332 /* Lookup a variable from .avc_(map|name|hash), return whether it was found.
333 * Sets .avc_prime; .avc_var is NULL if not found.
334 * Here it is where we care for _I3VAL and _DEFVAL.
335 * An _I3VAL will be "consumed" as necessary anyway, but it won't be used to
336 * create a new variable if i3val_nonew is true; if i3val_nonew is TRUM1 then
337 * we set .avc_var to -1 and return true if that was the case, otherwise we'll
338 * return FAL0, then! */
339 static bool_t a_amv_var_lookup(struct a_amv_var_carrier *avcp,
340 bool_t i3val_nonew);
342 /* Lookup functions for special category variables, _pospar drives all
343 * positional parameter etc. special categories */
344 static char const *a_amv_var_vsc_global(struct a_amv_var_carrier *avcp);
345 static char const *a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp);
346 static char const *a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp);
348 /* Set var from .avc_(map|name|hash), return success */
349 static bool_t a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
350 bool_t force_env);
352 static bool_t a_amv_var__putenv(struct a_amv_var_carrier *avcp,
353 struct a_amv_var *avp);
355 /* Clear var from .avc_(map|name|hash); sets .avc_var=NULL, return success */
356 static bool_t a_amv_var_clear(struct a_amv_var_carrier *avcp, bool_t force_env);
358 static bool_t a_amv_var__clearenv(char const *name, struct a_amv_var *avp);
360 /* List all variables */
361 static void a_amv_var_show_all(void);
363 static int a_amv_var__show_cmp(void const *s1, void const *s2);
365 /* Actually do print one, return number of lines written */
366 static size_t a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp);
368 /* Shared c_set() and c_environ():set impl, return success */
369 static bool_t a_amv_var_c_set(char **ap, bool_t issetenv);
371 static struct a_amv_mac *
372 a_amv_mac_lookup(char const *name, struct a_amv_mac *newamp,
373 enum a_amv_mac_flags amf){
374 struct a_amv_mac *amp, **ampp;
375 ui32_t h;
376 enum a_amv_mac_flags save_amf;
377 NYD2_ENTER;
379 save_amf = amf;
380 amf &= a_AMV_MF_TYPE_MASK;
381 h = a_AMV_NAME2HASH(name);
382 h = a_AMV_HASH2PRIME(h);
383 ampp = &a_amv_macs[h];
385 for(amp = *ampp; amp != NULL; ampp = &(*ampp)->am_next, amp = amp->am_next){
386 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf &&
387 !strcmp(amp->am_name, name)){
388 if(n_LIKELY((save_amf & a_AMV_MF_UNDEF) == 0))
389 goto jleave;
391 *ampp = amp->am_next;
393 if(amp->am_refcnt > 0){
394 amp->am_flags |= a_AMV_MF_DELETE;
395 if(n_poption & n_PO_D_V)
396 n_err(_("Delayed deletion of currently active %s: %s\n"),
397 (amp->am_flags & a_AMV_MF_ACCOUNT ? "account" : "define"),
398 name);
399 }else{
400 a_amv_mac_free(amp);
401 amp = (struct a_amv_mac*)-1;
403 break;
407 if(newamp != NULL){
408 ampp = &a_amv_macs[h];
409 newamp->am_next = *ampp;
410 *ampp = newamp;
411 amp = NULL;
413 jleave:
414 NYD2_LEAVE;
415 return amp;
418 static int
419 a_amv_mac_call(void *v, bool_t silent_nexist){
420 struct a_amv_mac *amp;
421 int rv;
422 char const *name;
423 struct n_cmd_arg_ctx *cacp;
424 NYD_ENTER;
426 cacp = v;
428 if(cacp->cac_no == 0){
429 n_err(_("Synopsis: call(_if)?: name [:<arguments>:]\n"));
430 n_pstate_err_no = n_ERR_INVAL;
431 rv = 1;
432 goto jleave;
435 name = cacp->cac_arg->ca_arg.ca_str.s;
437 if(n_UNLIKELY(cacp->cac_no > a_AMV_POSPAR_MAX)){
438 n_err(_("Too many arguments to macro `call': %s\n"), name);
439 n_pstate_err_no = n_ERR_OVERFLOW;
440 rv = 1;
441 }else if(n_UNLIKELY((amp = a_amv_mac_lookup(name, NULL, a_AMV_MF_NONE)
442 ) == NULL)){
443 if(!silent_nexist)
444 n_err(_("Undefined macro called: %s\n"), n_shexp_quote_cp(name, FAL0));
445 n_pstate_err_no = n_ERR_NOENT;
446 rv = 1;
447 }else{
448 char const **argv;
449 struct a_amv_mac_call_args *amcap;
450 size_t argc;
452 argc = cacp->cac_no + 1;
453 amcap = n_lofi_alloc(sizeof *amcap + (argc * sizeof *argv));
454 argv = (void*)&amcap[1];
456 for(argc = 0; (cacp->cac_arg = cacp->cac_arg->ca_next) != NULL; ++argc)
457 argv[argc] = cacp->cac_arg->ca_arg.ca_str.s;
458 argv[argc] = NULL;
460 memset(amcap, 0, sizeof *amcap);
461 amcap->amca_name = name;
462 amcap->amca_amp = amp;
463 if(a_amv_lopts != NULL)
464 amcap->amca_loflags = (a_amv_lopts->as_loflags & a_AMV_LF_CALL_MASK
465 ) >> a_AMV_LF_CALL_TO_SCOPE_SHIFT;
466 if(argc > 0){
467 amcap->amca_pospar.app_count = (ui16_t)argc;
468 amcap->amca_pospar.app_not_heap = TRU1;
469 amcap->amca_pospar.app_dat = argv;
472 (void)a_amv_mac_exec(amcap);
473 rv = n_pstate_ex_no;
475 jleave:
476 NYD_LEAVE;
477 return rv;
480 static bool_t
481 a_amv_mac_exec(struct a_amv_mac_call_args *amcap){
482 struct a_amv_lostack *losp;
483 struct a_amv_mac_line **amlp;
484 char **args_base, **args;
485 struct a_amv_mac *amp;
486 bool_t rv;
487 NYD2_ENTER;
489 amp = amcap->amca_amp;
490 assert(amp != NULL && amp != a_AMV_MACKY_MACK);
491 ++amp->am_refcnt;
492 /* XXX Unfortunately we yet need to dup the macro lines! :( */
493 args_base = args = n_alloc(sizeof(*args) * (amp->am_line_cnt +1));
494 for(amlp = amp->am_line_dat; *amlp != NULL; ++amlp)
495 *(args++) = sbufdup((*amlp)->aml_dat, (*amlp)->aml_len);
496 *args = NULL;
498 losp = n_lofi_alloc(sizeof *losp);
499 losp->as_global_saved = a_amv_lopts;
500 if((losp->as_amcap = amcap)->amca_unroller == NULL){
501 losp->as_up = losp->as_global_saved;
502 losp->as_lopts = NULL;
503 }else{
504 losp->as_up = NULL;
505 losp->as_lopts = *amcap->amca_unroller;
507 losp->as_loflags = amcap->amca_loflags;
509 a_amv_lopts = losp;
510 if(amcap->amca_hook_pre != NULL)
511 n_PS_ROOT_BLOCK((*amcap->amca_hook_pre)(amcap->amca_hook_arg));
512 rv = n_go_macro((n_GO_INPUT_NONE |
513 (amcap->amca_no_xcall ? n_GO_INPUT_NO_XCALL : 0)),
514 amp->am_name, args_base, &a_amv_mac__finalize, losp);
515 NYD2_LEAVE;
516 return rv;
519 static void
520 a_amv_mac__finalize(void *vp){
521 struct a_amv_mac *amp;
522 struct a_amv_mac_call_args *amcap;
523 struct a_amv_lostack *losp;
524 NYD2_ENTER;
526 losp = vp;
527 a_amv_lopts = losp->as_global_saved;
529 amcap = losp->as_amcap;
531 if(!amcap->amca_pospar.app_not_heap && amcap->amca_pospar.app_maxcount > 0){
532 ui16_t i;
534 for(i = amcap->amca_pospar.app_maxcount; i-- != 0;)
535 n_free(n_UNCONST(amcap->amca_pospar.app_dat[i]));
536 n_free(amcap->amca_pospar.app_dat);
539 if(amcap->amca_unroller == NULL){
540 if(losp->as_lopts != NULL)
541 a_amv_lopts_unroll(&losp->as_lopts);
542 }else
543 *amcap->amca_unroller = losp->as_lopts;
545 if(amcap->amca_ps_hook_mask)
546 n_pstate &= ~n_PS_HOOK_MASK;
548 if((amp = amcap->amca_amp) != a_AMV_MACKY_MACK && amp != NULL &&
549 --amp->am_refcnt == 0 && (amp->am_flags & a_AMV_MF_DELETE))
550 a_amv_mac_free(amp);
552 n_lofi_free(losp);
553 n_lofi_free(amcap);
554 NYD2_LEAVE;
557 static bool_t
558 a_amv_mac_show(enum a_amv_mac_flags amf){
559 size_t lc, mc, ti, i;
560 char const *typestr;
561 FILE *fp;
562 bool_t rv;
563 NYD2_ENTER;
565 rv = FAL0;
567 if((fp = Ftmp(NULL, "deflist", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
568 NULL){
569 n_perr(_("`define' or `account' list: cannot create temporary file"), 0);
570 goto jleave;
573 amf &= a_AMV_MF_TYPE_MASK;
574 typestr = (amf & a_AMV_MF_ACCOUNT) ? "account" : "define";
576 for(lc = mc = ti = 0; ti < a_AMV_PRIME; ++ti){
577 struct a_amv_mac *amp;
579 for(amp = a_amv_macs[ti]; amp != NULL; amp = amp->am_next){
580 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
581 struct a_amv_mac_line **amlpp;
583 if(++mc > 1){
584 putc('\n', fp);
585 ++lc;
587 ++lc;
588 fprintf(fp, "%s %s {\n", typestr, amp->am_name);
589 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++lc, ++amlpp){
590 for(i = (*amlpp)->aml_prespc; i > 0; --i)
591 putc(' ', fp);
592 fputs((*amlpp)->aml_dat, fp);
593 putc('\n', fp);
595 fputs("}\n", fp);
596 ++lc;
600 if(mc > 0)
601 page_or_print(fp, lc);
603 rv = (ferror(fp) == 0);
604 Fclose(fp);
605 jleave:
606 NYD2_LEAVE;
607 return rv;
610 static bool_t
611 a_amv_mac_def(char const *name, enum a_amv_mac_flags amf){
612 struct str line;
613 ui32_t line_cnt, maxlen;
614 struct linelist{
615 struct linelist *ll_next;
616 struct a_amv_mac_line *ll_amlp;
617 } *llp, *ll_head, *ll_tail;
618 union {size_t s; int i; ui32_t ui; size_t l;} n;
619 struct a_amv_mac *amp;
620 bool_t rv;
621 NYD2_ENTER;
623 memset(&line, 0, sizeof line);
624 rv = FAL0;
625 amp = NULL;
627 /* TODO We should have our input state machine which emits Line events,
628 * TODO and hook different consumers dependent on our content, as stated
629 * TODO in i think lex_input; */
630 /* Read in the lines which form the macro content */
631 for(ll_tail = ll_head = NULL, line_cnt = maxlen = 0;;){
632 ui32_t leaspc;
633 char *cp;
635 n.i = n_go_input(n_GO_INPUT_CTX_DEFAULT | n_GO_INPUT_NL_ESC, n_empty,
636 &line.s, &line.l, NULL, NULL);
637 if(n.ui == 0)
638 continue;
639 if(n.i < 0){
640 n_err(_("Unterminated %s definition: %s\n"),
641 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
642 goto jerr;
645 /* Trim WS, remember amount of leading spaces for display purposes */
646 for(cp = line.s, leaspc = 0; n.ui > 0; ++cp, --n.ui)
647 if(*cp == '\t')
648 leaspc = (leaspc + 8u) & ~7u;
649 else if(*cp == ' ')
650 ++leaspc;
651 else
652 break;
653 for(; n.ui > 0 && spacechar(cp[n.ui - 1]); --n.ui)
655 if(n.ui == 0)
656 continue;
658 maxlen = n_MAX(maxlen, n.ui);
659 cp[n.ui++] = '\0';
661 /* Is is the closing brace? */
662 if(*cp == '}')
663 break;
665 if(n_LIKELY(++line_cnt < UI32_MAX)){
666 struct a_amv_mac_line *amlp;
668 llp = n_autorec_alloc(sizeof *llp);
669 if(ll_head == NULL)
670 ll_head = llp;
671 else
672 ll_tail->ll_next = llp;
673 ll_tail = llp;
674 llp->ll_next = NULL;
675 llp->ll_amlp = amlp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac_line,
676 aml_dat) + n.ui);
677 amlp->aml_len = n.ui -1;
678 amlp->aml_prespc = leaspc;
679 memcpy(amlp->aml_dat, cp, n.ui);
680 }else{
681 n_err(_("Too much content in %s definition: %s\n"),
682 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
683 goto jerr;
687 /* Create the new macro */
688 n.s = strlen(name) +1;
689 amp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name) + n.s);
690 memset(amp, 0, n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name));
691 amp->am_maxlen = maxlen;
692 amp->am_line_cnt = line_cnt;
693 amp->am_flags = amf;
694 memcpy(amp->am_name, name, n.s);
695 /* C99 */{
696 struct a_amv_mac_line **amlpp;
698 amp->am_line_dat = amlpp = n_alloc(sizeof(*amlpp) * ++line_cnt);
699 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
700 *amlpp++ = llp->ll_amlp;
701 *amlpp = NULL;
704 /* Create entry, replace a yet existing one as necessary */
705 a_amv_mac_lookup(name, amp, amf | a_AMV_MF_UNDEF);
706 rv = TRU1;
707 jleave:
708 if(line.s != NULL)
709 n_free(line.s);
710 NYD2_LEAVE;
711 return rv;
713 jerr:
714 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
715 n_free(llp->ll_amlp);
717 * if(amp != NULL){
718 * n_free(amp->am_line_dat);
719 * n_free(amp);
720 *}*/
721 goto jleave;
724 static bool_t
725 a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf){
726 struct a_amv_mac *amp;
727 bool_t rv;
728 NYD2_ENTER;
730 rv = TRU1;
732 if(n_LIKELY(name[0] != '*' || name[1] != '\0')){
733 if((amp = a_amv_mac_lookup(name, NULL, amf | a_AMV_MF_UNDEF)) == NULL){
734 n_err(_("%s not defined: %s\n"),
735 (amf & a_AMV_MF_ACCOUNT ? "Account" : "Macro"), name);
736 rv = FAL0;
738 }else{
739 struct a_amv_mac **ampp, *lamp;
741 for(ampp = a_amv_macs; PTRCMP(ampp, <, &a_amv_macs[n_NELEM(a_amv_macs)]);
742 ++ampp)
743 for(lamp = NULL, amp = *ampp; amp != NULL;){
744 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
745 /* xxx Expensive but rare: be simple */
746 a_amv_mac_lookup(amp->am_name, NULL, amf | a_AMV_MF_UNDEF);
747 amp = (lamp == NULL) ? *ampp : lamp->am_next;
748 }else{
749 lamp = amp;
750 amp = amp->am_next;
754 NYD2_LEAVE;
755 return rv;
758 static void
759 a_amv_mac_free(struct a_amv_mac *amp){
760 struct a_amv_mac_line **amlpp;
761 NYD2_ENTER;
763 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++amlpp)
764 n_free(*amlpp);
765 n_free(amp->am_line_dat);
766 n_free(amp);
767 NYD2_LEAVE;
770 static void
771 a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
772 struct a_amv_var *oavp){
773 struct a_amv_var *avp;
774 size_t nl, vl;
775 NYD2_ENTER;
777 /* Propagate unrolling up the stack, as necessary */
778 assert(alp != NULL);
779 for(;;){
780 if(alp->as_loflags & a_AMV_LF_SCOPE_MASK)
781 break;
782 if((alp = alp->as_up) == NULL)
783 goto jleave;
786 /* Check whether this variable is handled yet XXX Boost: ID check etc.!! */
787 for(avp = alp->as_lopts; avp != NULL; avp = avp->av_link)
788 if(!strcmp(avp->av_name, name))
789 goto jleave;
791 nl = strlen(name) +1;
792 vl = (oavp != NULL) ? strlen(oavp->av_value) +1 : 0;
793 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + nl + vl);
794 avp->av_link = alp->as_lopts;
795 alp->as_lopts = avp;
796 if(vl != 0){
797 avp->av_value = &avp->av_name[nl];
798 avp->av_flags = oavp->av_flags;
799 memcpy(avp->av_value, oavp->av_value, vl);
800 #ifdef HAVE_PUTENV
801 if(oavp->av_env != NULL)
802 avp->av_env = sstrdup(oavp->av_env);
803 assert(avp->av_env == NULL ||
804 (avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)));
805 #endif
807 memcpy(avp->av_name, name, nl);
808 jleave:
809 NYD2_LEAVE;
812 static void
813 a_amv_lopts_unroll(struct a_amv_var **avpp){
814 struct a_amv_lostack *save_alp;
815 struct a_amv_var *x, *avp;
816 NYD2_ENTER;
818 avp = *avpp;
819 *avpp = NULL;
821 save_alp = a_amv_lopts;
822 a_amv_lopts = NULL;
823 while(avp != NULL){
824 x = avp;
825 avp = avp->av_link;
826 n_PS_ROOT_BLOCK(n_var_vset(x->av_name, (uintptr_t)x->av_value));
827 n_free(x);
829 a_amv_lopts = save_alp;
830 NYD2_LEAVE;
833 static char *
834 a_amv_var_copy(char const *str){
835 char *news;
836 size_t len;
837 NYD2_ENTER;
839 if(*str == '\0')
840 news = n_UNCONST(n_empty);
841 else if(str[1] == '\0'){
842 if(str[0] == '1')
843 news = n_UNCONST(n_1);
844 else if(str[0] == '0')
845 news = n_UNCONST(n_0);
846 else
847 goto jheap;
848 }else if(str[2] == '\0' && str[0] == '-' && str[1] == '1')
849 news = n_UNCONST(n_m1);
850 else{
851 jheap:
852 len = strlen(str) +1;
853 news = n_alloc(len);
854 memcpy(news, str, len);
856 NYD2_LEAVE;
857 return news;
860 static void
861 a_amv_var_free(char *cp){
862 NYD2_ENTER;
863 if(cp[0] != '\0' && cp != n_0 && cp != n_1 && cp != n_m1)
864 n_free(cp);
865 NYD2_LEAVE;
868 static bool_t
869 a_amv_var_check_vips(enum a_amv_var_vip_mode avvm, enum okeys okey,
870 char const *val){
871 bool_t ok;
872 NYD2_ENTER;
874 ok = TRU1;
876 if(avvm == a_AMV_VIP_SET_PRE){
877 switch(okey){
878 default:
879 break;
880 case ok_v_HOME:
881 /* Note this gets called from main.c during initialization, and they
882 * simply set this to pw_dir as a fallback: don't verify _that_ call.
883 * See main.c! */
884 if(!(n_pstate & n_PS_ROOT) && !n_is_dir(val, TRUM1)){
885 n_err(_("$HOME is not a directory or not accessible: %s\n"),
886 n_shexp_quote_cp(val, FAL0));
887 ok = FAL0;
889 break;
890 case ok_v_TMPDIR:
891 if(!n_is_dir(val, TRU1)){
892 n_err(_("$TMPDIR is not a directory or not accessible: %s\n"),
893 n_shexp_quote_cp(val, FAL0));
894 ok = FAL0;
896 break;
897 case ok_v_umask:
898 if(*val != '\0'){
899 ui64_t uib;
901 n_idec_ui64_cp(&uib, val, 0, NULL);
902 if(uib & ~0777u){ /* (is valid _VF_POSNUM) */
903 n_err(_("Invalid *umask* setting: %s\n"), val);
904 ok = FAL0;
907 break;
909 }else if(avvm == a_AMV_VIP_SET_POST){
910 switch(okey){
911 default:
912 break;
913 case ok_b_ask:
914 ok_bset(asksub);
915 break;
916 case ok_b_debug:
917 n_poption |= n_PO_DEBUG;
918 break;
919 case ok_v_HOME:
920 /* Invalidate any resolved folder then, too
921 * FALLTHRU */
922 case ok_v_folder:
923 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
924 break;
925 case ok_v_ifs:{
926 char *x_b, *x, c;
928 x_b = x = n_autorec_alloc(strlen(val) +1);
929 while((c = *val++) != '\0')
930 if(spacechar(c))
931 *x++ = c;
932 *x = '\0';
933 n_PS_ROOT_BLOCK(ok_vset(ifs_ws, x_b));
934 }break;
935 #ifdef HAVE_SETLOCALE
936 case ok_v_LANG:
937 case ok_v_LC_ALL:
938 case ok_v_LC_CTYPE:
939 n_locale_init();
940 break;
941 #endif
942 case ok_b_memdebug:
943 n_poption |= n_PO_MEMDEBUG;
944 break;
945 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
946 if(!(n_pstate & n_PS_ROOT))
947 n_PS_ROOT_BLOCK(ok_bset(posix));
948 break;
949 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
950 if(!(n_pstate & n_PS_ROOT))
951 n_PS_ROOT_BLOCK(ok_bset(POSIXLY_CORRECT));
952 break;
953 case ok_b_skipemptybody:
954 n_poption |= n_PO_E_FLAG;
955 break;
956 case ok_b_typescript_mode:
957 ok_bset(colour_disable);
958 ok_bset(line_editor_disable);
959 if(!(n_psonce & n_PSO_STARTED))
960 ok_bset(termcap_disable);
961 break;
962 case ok_v_umask:
963 if(*val != '\0'){
964 ui64_t uib;
966 n_idec_ui64_cp(&uib, val, 0, NULL);
967 umask((mode_t)uib);
969 break;
970 case ok_b_verbose:
971 n_poption |= (n_poption & n_PO_VERB) ? n_PO_VERBVERB : n_PO_VERB;
972 break;
974 }else{
975 switch(okey){
976 default:
977 break;
978 case ok_b_ask:
979 ok_bclear(asksub);
980 break;
981 case ok_b_debug:
982 n_poption &= ~n_PO_DEBUG;
983 break;
984 case ok_v_HOME:
985 /* Invalidate any resolved folder then, too
986 * FALLTHRU */
987 case ok_v_folder:
988 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
989 break;
990 case ok_b_memdebug:
991 n_poption &= ~n_PO_MEMDEBUG;
992 break;
993 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
994 if(!(n_pstate & n_PS_ROOT))
995 n_PS_ROOT_BLOCK(ok_bclear(posix));
996 break;
997 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
998 if(!(n_pstate & n_PS_ROOT))
999 n_PS_ROOT_BLOCK(ok_bclear(POSIXLY_CORRECT));
1000 break;
1001 case ok_b_skipemptybody:
1002 n_poption &= ~n_PO_E_FLAG;
1003 break;
1004 case ok_b_verbose:
1005 n_poption &= ~(n_PO_VERB | n_PO_VERBVERB);
1006 break;
1009 NYD2_LEAVE;
1010 return ok;
1013 static bool_t
1014 a_amv_var_check_nocntrls(char const *val){
1015 char c;
1016 NYD2_ENTER;
1018 while((c = *val++) != '\0')
1019 if(cntrlchar(c))
1020 break;
1021 NYD2_LEAVE;
1022 return (c == '\0');
1025 static bool_t
1026 a_amv_var_check_num(char const *val, bool_t posnum){
1027 /* TODO The internal/environment variables which are num= or posnum= should
1028 * TODO gain special lookup functions, or the return should be void* and
1029 * TODO castable to integer; i.e. no more strtoX() should be needed.
1030 * TODO I.e., the result of this function should instead be stored */
1031 bool_t rv;
1032 NYD2_ENTER;
1034 rv = TRU1;
1036 if(*val != '\0'){ /* Would be _VF_NOTEMPTY if not allowed */
1037 ui64_t uib;
1038 enum n_idec_state ids;
1040 ids = n_idec_cp(&uib, val, 0,
1041 (posnum ? n_IDEC_MODE_SIGNED_TYPE : n_IDEC_MODE_NONE), NULL);
1042 if((ids & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
1043 ) != n_IDEC_STATE_CONSUMED)
1044 rv = FAL0;
1045 /* TODO Unless we store integers we need to look and forbid, because
1046 * TODO callee may not be able to swallow, e.g., "-1" */
1047 if(posnum && (ids & n_IDEC_STATE_SEEN_MINUS))
1048 rv = FAL0;
1050 NYD2_LEAVE;
1051 return rv;
1054 static char const *
1055 a_amv_var_canonify(char const *vn){
1056 NYD2_ENTER;
1057 if(!upperchar(*vn)){
1058 char const *vp;
1060 for(vp = vn; *vp != '\0' && *vp != '@'; ++vp)
1062 vn = (*vp == '@') ? i_strdup(vn) : vn;
1064 NYD2_LEAVE;
1065 return vn;
1068 static bool_t
1069 a_amv_var_revlookup(struct a_amv_var_carrier *avcp, char const *name){
1070 ui32_t hash, i, j;
1071 struct a_amv_var_map const *avmp;
1072 char c;
1073 NYD2_ENTER;
1075 /* It may be a special a.k.a. macro-local or one-letter parameter */
1076 c = name[0];
1077 if(n_UNLIKELY(digitchar(c))){
1078 /* (Inline dec. atoi, ugh) */
1079 for(j = (ui8_t)c - '0', i = 1;; ++i){
1080 c = name[i];
1081 if(c == '\0')
1082 break;
1083 if(!digitchar(c))
1084 goto jno_special_param;
1085 j = j * 10 + (ui8_t)c - '0';
1087 if(j <= a_AMV_POSPAR_MAX){
1088 avcp->avc_special_cat = a_AMV_VSC_POSPAR;
1089 goto jspecial_param;
1091 }else if(n_UNLIKELY(name[1] == '\0')){
1092 switch(c){
1093 case '?':
1094 case '!':
1095 avcp->avc_special_cat = a_AMV_VSC_GLOBAL;
1096 j = (c == '?') ? a_AMV_VST_QM : a_AMV_VST_EM;
1097 goto jspecial_param;
1098 case '^':
1099 goto jmultiplex;
1100 case '*':
1101 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1102 j = a_AMV_VST_STAR;
1103 goto jspecial_param;
1104 case '@':
1105 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1106 j = a_AMV_VST_AT;
1107 goto jspecial_param;
1108 case '#':
1109 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1110 j = a_AMV_VST_NOSIGN;
1111 goto jspecial_param;
1112 default:
1113 break;
1115 }else if(c == '^'){
1116 jmultiplex:
1117 avcp->avc_special_cat = a_AMV_VSC_MULTIPLEX;
1118 j = a_AMV_VST_CACC;
1119 goto jspecial_param;
1122 /* Normal reverse lookup, walk over the hashtable */
1123 jno_special_param:
1124 avcp->avc_special_cat = a_AMV_VSC_NONE;
1125 avcp->avc_name = name = a_amv_var_canonify(name);
1126 avcp->avc_hash = hash = a_AMV_NAME2HASH(name);
1128 for(i = hash % a_AMV_VAR_REV_PRIME, j = 0; j <= a_AMV_VAR_REV_LONGEST; ++j){
1129 ui32_t x;
1131 if((x = a_amv_var_revmap[i]) == a_AMV_VAR_REV_ILL)
1132 break;
1134 avmp = &a_amv_var_map[x];
1135 if(avmp->avm_hash == hash &&
1136 !strcmp(&a_amv_var_names[avmp->avm_keyoff], name)){
1137 avcp->avc_map = avmp;
1138 avcp->avc_okey = (enum okeys)x;
1139 goto jleave;
1142 if(++i == a_AMV_VAR_REV_PRIME){
1143 #ifdef a_AMV_VAR_REV_WRAPAROUND
1144 i = 0;
1145 #else
1146 break;
1147 #endif
1150 avcp->avc_map = NULL;
1151 avcp = NULL;
1152 jleave:
1153 NYD2_LEAVE;
1154 return (avcp != NULL);
1156 /* All these are mapped to *--special-param* */
1157 jspecial_param:
1158 avcp->avc_name = name;
1159 avcp->avc_special_prop = (ui16_t)j;
1160 avmp = &a_amv_var_map[a_AMV_VAR__SPECIAL_PARAM_MAP_IDX];
1161 avcp->avc_hash = avmp->avm_hash;
1162 avcp->avc_map = avmp;
1163 avcp->avc_okey = ok_v___special_param;
1164 goto jleave;
1167 static bool_t
1168 a_amv_var_lookup(struct a_amv_var_carrier *avcp, bool_t i3val_nonew){
1169 size_t i;
1170 char const *cp;
1171 struct a_amv_var_map const *avmp;
1172 struct a_amv_var *avp;
1173 NYD2_ENTER;
1175 /* C99 */{
1176 struct a_amv_var **avpp, *lavp;
1178 avpp = &a_amv_vars[avcp->avc_prime = a_AMV_HASH2PRIME(avcp->avc_hash)];
1180 for(lavp = NULL, avp = *avpp; avp != NULL; lavp = avp, avp = avp->av_link)
1181 if(!strcmp(avp->av_name, avcp->avc_name)){
1182 /* Relink as head, hope it "sorts on usage" over time.
1183 * The code relies on this behaviour! */
1184 if(lavp != NULL){
1185 lavp->av_link = avp->av_link;
1186 avp->av_link = *avpp;
1187 *avpp = avp;
1190 /* If this setting has been established via -S and we still have
1191 * not reached the _STARTED_CONFIG program state, it may have been
1192 * an explicit "clearance" that is to be treated as unset.
1193 * Because that is a special condition that (has been hacked in
1194 * later and) needs to be encapsulated in lower levels, but not
1195 * of interest if _set() or _clear() called us */
1196 switch(avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK){
1197 case a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET:
1198 if(!i3val_nonew){
1199 avcp->avc_var = avp;
1200 avp = NULL;
1201 goto j_leave;
1203 /* FALLTHRU */
1204 default:
1205 break;
1207 goto jleave;
1211 /* If this is not an assembled variable we need to consider some special
1212 * initialization cases and eventually create the variable anew */
1213 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1214 /* Does it have an import-from-environment flag? */
1215 if(n_UNLIKELY((avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV)) != 0)){
1216 if(n_LIKELY((cp = getenv(avcp->avc_name)) != NULL)){
1217 /* May be better not to use that one, though? */
1218 /* TODO Outsource the tests into a _shared_ test function! */
1219 bool_t isempty, isbltin;
1221 isempty = (*cp == '\0' &&
1222 (avmp->avm_flags & a_AMV_VF_NOTEMPTY) != 0);
1223 isbltin = ((avmp->avm_flags & (a_AMV_VF_I3VAL | a_AMV_VF_DEFVAL)
1224 ) != 0);
1226 if(n_UNLIKELY(isempty)){
1227 n_err(_("Environment variable must not be empty: %s\n"),
1228 avcp->avc_name);
1229 if(!isbltin)
1230 goto jerr;
1231 }else if(n_LIKELY(*cp != '\0')){
1232 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOCNTRLS) &&
1233 !a_amv_var_check_nocntrls(cp))){
1234 n_err(_("Ignoring environment, control characters "
1235 "invalid in variable: %s\n"), avcp->avc_name);
1236 goto jerr;
1238 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1239 !a_amv_var_check_num(cp, FAL0))){
1240 n_err(_("Environment variable value not a number "
1241 "or out of range: %s\n"), avcp->avc_name);
1242 goto jerr;
1244 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1245 !a_amv_var_check_num(cp, TRU1))){
1246 n_err(_("Environment variable value not a number, "
1247 "negative or out of range: %s\n"), avcp->avc_name);
1248 goto jerr;
1250 goto jnewval;
1251 }else
1252 goto jnewval;
1256 /* A first-time init switch is to be handled now and here */
1257 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_I3VAL) != 0)){
1258 static struct a_amv_var_defval const **arr,
1259 *arr_base[a_AMV_VAR_I3VALS_CNT +1];
1261 if(arr == NULL){
1262 arr = &arr_base[0];
1263 arr[i = a_AMV_VAR_I3VALS_CNT] = NULL;
1264 while(i-- > 0)
1265 arr[i] = &a_amv_var_i3vals[i];
1268 for(i = 0; arr[i] != NULL; ++i)
1269 if(arr[i]->avdv_okey == avcp->avc_okey){
1270 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_1
1271 : arr[i]->avdv_value;
1272 /* Remove this entry, hope entire block becomes no-op asap */
1274 arr[i] = arr[i + 1];
1275 while(arr[i++] != NULL);
1277 if(!i3val_nonew)
1278 goto jnewval;
1279 if(i3val_nonew == TRUM1)
1280 avp = (struct a_amv_var*)-1;
1281 goto jleave;
1285 /* */
1286 jdefval:
1287 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0){
1288 for(i = 0; i < a_AMV_VAR_DEFVALS_CNT; ++i)
1289 if(a_amv_var_defvals[i].avdv_okey == avcp->avc_okey){
1290 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_1
1291 : a_amv_var_defvals[i].avdv_value;
1292 goto jnewval;
1296 /* The virtual variables */
1297 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIRT) != 0)){
1298 for(i = 0; i < a_AMV_VAR_VIRTS_CNT; ++i)
1299 if(a_amv_var_virts[i].avv_okey == avcp->avc_okey){
1300 avp = n_UNCONST(a_amv_var_virts[i].avv_var);
1301 goto jleave;
1303 /* Not reached */
1307 jerr:
1308 avp = NULL;
1309 jleave:
1310 avcp->avc_var = avp;
1311 j_leave:
1312 NYD2_LEAVE;
1313 return (avp != NULL);
1315 jnewval:
1316 /* E.g., $TMPDIR may be set to non-existent, so we need to be able to catch
1317 * that and redirect to a possible default value */
1318 if((avmp->avm_flags & a_AMV_VF_VIP) &&
1319 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, cp)){
1320 #ifdef HAVE_SETENV
1321 if(avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV))
1322 unsetenv(avcp->avc_name);
1323 #endif
1324 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)
1325 goto jdefval;
1326 goto jerr;
1327 }else{
1328 struct a_amv_var **avpp;
1329 size_t l;
1331 l = strlen(avcp->avc_name) +1;
1332 avcp->avc_var =
1333 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1334 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1335 *avpp = avp;
1336 avp->av_flags = avmp->avm_flags;
1337 avp->av_value = a_amv_var_copy(cp);
1338 memcpy(avp->av_name, avcp->avc_name, l);
1340 if(avp->av_flags & a_AMV_VF_ENV)
1341 a_amv_var__putenv(avcp, avp);
1342 if(avmp->avm_flags & a_AMV_VF_VIP)
1343 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, cp);
1344 goto jleave;
1348 static char const *
1349 a_amv_var_vsc_global(struct a_amv_var_carrier *avcp){
1350 char itoabuf[32];
1351 char const *rv;
1352 si32_t *ep;
1353 struct a_amv_var_map const *avmp;
1354 NYD2_ENTER;
1356 /* Not function local, TODO but lazy evaluted for now */
1357 if(avcp->avc_special_prop == a_AMV_VST_QM){
1358 avmp = &a_amv_var_map[a_AMV_VAR__QM_MAP_IDX];
1359 avcp->avc_okey = ok_v___qm;
1360 ep = &n_pstate_ex_no;
1361 }else{
1362 avmp = &a_amv_var_map[a_AMV_VAR__EM_MAP_IDX];
1363 avcp->avc_okey = ok_v___em;
1364 ep = &n_pstate_err_no;
1367 /* XXX Oh heaven, we are responsible to ensure that $?/! is up-to-date
1368 * TODO we could num=1 ok_v___[qe]m, but the thing is still a string
1369 * TODO and thus conversion takes places over and over again; also
1370 * TODO for now that would have to occur before we set _that_ value
1371 * TODO so let's special treat it until we store ints as such */
1372 switch(*ep){
1373 case 0: rv = n_0; break;
1374 case 1: rv = n_1; break;
1375 default:
1376 snprintf(itoabuf, sizeof itoabuf, "%d", *ep);
1377 rv = itoabuf;
1378 break;
1380 n_PS_ROOT_BLOCK(n_var_okset(avcp->avc_okey, (uintptr_t)rv));
1382 avcp->avc_hash = avmp->avm_hash;
1383 avcp->avc_map = avmp;
1384 rv = a_amv_var_lookup(avcp, FAL0) ? avcp->avc_var->av_value : NULL;
1385 NYD2_LEAVE;
1386 return rv;
1389 static char const *
1390 a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp){
1391 char itoabuf[32];
1392 si32_t e;
1393 size_t i;
1394 char const *rv;
1395 NYD2_ENTER;
1397 i = strlen(rv = &avcp->avc_name[1]);
1399 /* ERR, ERRDOC, ERRNAME, plus *-NAME variants */
1400 if(rv[0] == 'E' && i >= 3 && rv[1] == 'R' && rv[2] == 'R'){
1401 if(i == 3){
1402 e = n_pstate_err_no;
1403 goto jeno;
1404 }else if(rv[3] == '-'){
1405 e = n_err_from_name(&rv[4]);
1406 jeno:
1407 switch(e){
1408 case 0: rv = n_0; break;
1409 case 1: rv = n_1; break;
1410 default:
1411 snprintf(itoabuf, sizeof itoabuf, "%d", e);
1412 rv = savestr(itoabuf); /* XXX yet, cannot do numbers */
1413 break;
1415 goto jleave;
1416 }else if(i >= 6){
1417 if(!memcmp(&rv[3], "DOC", 3)){
1418 rv += 6;
1419 switch(*rv){
1420 case '\0': e = n_pstate_err_no; break;
1421 case '-': e = n_err_from_name(&rv[1]); break;
1422 default: goto jerr;
1424 rv = n_err_to_doc(e);
1425 goto jleave;
1426 }else if(i >= 7 && !memcmp(&rv[3], "NAME", 4)){
1427 rv += 7;
1428 switch(*rv){
1429 case '\0': e = n_pstate_err_no; break;
1430 case '-': e = n_err_from_name(&rv[1]); break;
1431 default: goto jerr;
1433 rv = n_err_to_name(e);
1434 goto jleave;
1439 jerr:
1440 rv = NULL;
1441 jleave:
1442 NYD2_LEAVE;
1443 return rv;
1446 static char const *
1447 a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp){
1448 size_t i, j;
1449 ui16_t argc;
1450 char const *rv, **argv;
1451 NYD2_ENTER;
1453 rv = NULL;
1455 /* If in a macro/xy.. */
1456 if(a_amv_lopts != NULL){
1457 bool_t ismacky;
1458 struct a_amv_mac_call_args *amcap;
1460 amcap = a_amv_lopts->as_amcap;
1461 argc = amcap->amca_pospar.app_count;
1462 argv = amcap->amca_pospar.app_dat;
1463 argv += amcap->amca_pospar.app_idx;
1465 /* ..in a `call'ed macro only, to be exact. Or in a_AMV_MACKY_MACK */
1466 if(!(ismacky = (amcap->amca_amp == a_AMV_MACKY_MACK)) &&
1467 (amcap->amca_ps_hook_mask ||
1468 (assert(amcap->amca_amp != NULL),
1469 (amcap->amca_amp->am_flags & a_AMV_MF_TYPE_MASK
1470 ) == a_AMV_MF_ACCOUNT)))
1471 goto jleave;
1473 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1474 if(avcp->avc_special_prop > 0){
1475 if(argc >= avcp->avc_special_prop)
1476 rv = argv[avcp->avc_special_prop - 1];
1477 }else if(ismacky)
1478 rv = amcap->amca_name;
1479 else
1480 rv = (a_amv_lopts->as_up != NULL
1481 ? a_amv_lopts->as_up->as_amcap->amca_name : n_empty);
1482 goto jleave;
1484 /* MACKY_MACK doesn't know about [*@#] */
1485 /*else*/ if(ismacky){
1486 if(n_poption & n_PO_D_V)
1487 n_err(_("Cannot use $*/$@/$# in this context: %s\n"),
1488 n_shexp_quote_cp(avcp->avc_name, FAL0));
1489 goto jleave;
1491 }else{
1492 argc = a_amv_pospar.app_count;
1493 argv = a_amv_pospar.app_dat;
1494 argv += a_amv_pospar.app_idx;
1496 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1497 if(avcp->avc_special_prop > 0){
1498 if(argc >= avcp->avc_special_prop)
1499 rv = argv[avcp->avc_special_prop - 1];
1500 }else
1501 rv = n_progname;
1502 goto jleave;
1506 switch(avcp->avc_special_prop){ /* XXX OPTIMIZE */
1507 case a_AMV_VST_STAR:{
1508 char sep;
1510 sep = *ok_vlook(ifs);
1511 if(0){
1512 case a_AMV_VST_AT:
1513 sep = ' ';
1515 for(i = j = 0; i < argc; ++i)
1516 j += strlen(argv[i]) + 1;
1517 if(j == 0)
1518 rv = n_empty;
1519 else{
1520 char *cp;
1522 rv = cp = n_autorec_alloc(j);
1523 for(i = j = 0; i < argc; ++i){
1524 j = strlen(argv[i]);
1525 memcpy(cp, argv[i], j);
1526 cp += j;
1527 if(sep != '\0')
1528 *cp++ = sep;
1530 if(sep != '\0')
1531 --cp;
1532 *cp = '\0';
1534 }break;
1535 case a_AMV_VST_NOSIGN:{
1536 char *cp;
1538 rv = cp = n_autorec_alloc(sizeof("65535"));
1539 snprintf(cp, sizeof("65535"), "%hu", argc);
1540 }break;
1541 default:
1542 rv = n_empty;
1543 break;
1545 jleave:
1546 NYD2_LEAVE;
1547 return rv;
1550 static bool_t
1551 a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
1552 bool_t force_env){
1553 struct a_amv_var *avp;
1554 char *oval;
1555 struct a_amv_var_map const *avmp;
1556 bool_t rv;
1557 NYD2_ENTER;
1559 if(value == NULL){
1560 rv = a_amv_var_clear(avcp, force_env);
1561 goto jleave;
1564 if((avmp = avcp->avc_map) != NULL){
1565 rv = FAL0;
1567 /* Validity checks */
1568 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_RDONLY) != 0 &&
1569 !(n_pstate & n_PS_ROOT))){
1570 value = N_("Variable is read-only: %s\n");
1571 goto jeavmp;
1573 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOTEMPTY) && *value == '\0')){
1574 value = N_("Variable must not be empty: %s\n");
1575 goto jeavmp;
1577 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOCNTRLS) != 0 &&
1578 !a_amv_var_check_nocntrls(value))){
1579 value = N_("Variable forbids control characters: %s\n");
1580 goto jeavmp;
1582 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1583 !a_amv_var_check_num(value, FAL0))){
1584 value = N_("Variable value not a number or out of range: %s\n");
1585 goto jeavmp;
1587 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1588 !a_amv_var_check_num(value, TRU1))){
1589 value = _("Variable value not a number, negative, "
1590 "or out of range: %s\n");
1591 goto jeavmp;
1593 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_IMPORT) != 0 &&
1594 !(n_psonce & n_PSO_STARTED) && !(n_pstate & n_PS_ROOT))){
1595 value = N_("Variable cannot be set in a resource file: %s\n");
1596 goto jeavmp;
1599 /* Any more complicated inter-dependency? */
1600 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1601 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, value))){
1602 value = N_("Assignment of variable aborted: %s\n");
1603 jeavmp:
1604 n_err(V_(value), avcp->avc_name);
1605 goto jleave;
1608 /* Transformations */
1609 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_LOWER)){
1610 char c;
1612 oval = savestr(value);
1613 value = oval;
1614 for(; (c = *oval) != '\0'; ++oval)
1615 *oval = lowerconv(c);
1619 rv = TRU1;
1620 a_amv_var_lookup(avcp, TRU1);
1621 avp = avcp->avc_var;
1623 /* If this setting had been established via -S and we still have not reached
1624 * the _STARTED_CONFIG program state, silently ignore request! */
1625 if(n_UNLIKELY(avp != NULL) &&
1626 n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1627 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1628 if((n_pstate & n_PS_ROOT) ||
1629 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1630 (n_poption & n_PO_S_FLAG_TEMPORARY)))
1631 goto joval_and_go;
1632 if(n_poption & n_PO_D_VV)
1633 n_err(_("Temporarily frozen by -S, not `set'ing: %s\n"),
1634 avcp->avc_name);
1635 goto jleave;
1638 /* Otherwise, if -S freezing was an `unset' request, be very simple and
1639 * avoid tampering with that very special case we are not really prepared
1640 * for just one more line of code: throw the old thing away! */
1641 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET))
1642 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1643 else{
1644 assert(avp->av_value == n_empty);
1645 a_amv_vars[avcp->avc_prime] = avp->av_link;
1646 n_free(avp);
1647 avcp->avc_var = avp = NULL;
1651 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1652 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1653 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1655 if(avp != NULL)
1656 joval_and_go:
1657 oval = avp->av_value;
1658 else{
1659 struct a_amv_var **avpp;
1660 size_t l;
1662 l = strlen(avcp->avc_name) +1;
1663 avcp->avc_var = avp = n_calloc(1,
1664 n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1665 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1666 *avpp = avp;
1667 avp->av_flags = (avmp != NULL) ? avmp->avm_flags : 0;
1668 memcpy(avp->av_name, avcp->avc_name, l);
1669 oval = n_UNCONST(n_empty);
1672 if(avmp == NULL)
1673 avp->av_value = a_amv_var_copy(value);
1674 else{
1675 /* Via `set' etc. the user may give even boolean options non-boolean
1676 * values, ignore that and force boolean */
1677 if(!(avp->av_flags & a_AMV_VF_BOOL))
1678 avp->av_value = a_amv_var_copy(value);
1679 else{
1680 if(!(n_pstate & n_PS_ROOT) && (n_poption & n_PO_D_VV) &&
1681 *value != '\0')
1682 n_err(_("Ignoring value of boolean variable: %s: %s\n"),
1683 avcp->avc_name, value);
1684 avp->av_value = n_UNCONST(n_1);
1688 if(force_env && !(avp->av_flags & a_AMV_VF_ENV))
1689 avp->av_flags |= a_AMV_VF_EXT_LINKED;
1690 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1691 rv = a_amv_var__putenv(avcp, avp);
1692 if(avp->av_flags & a_AMV_VF_VIP)
1693 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, value);
1695 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1696 if(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1697 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0)
1698 avp->av_flags |= a_AMV_VF_EXT_FROZEN;
1700 a_amv_var_free(oval);
1701 jleave:
1702 NYD2_LEAVE;
1703 return rv;
1706 static bool_t
1707 a_amv_var__putenv(struct a_amv_var_carrier *avcp, struct a_amv_var *avp){
1708 #ifndef HAVE_SETENV
1709 char *cp;
1710 #endif
1711 bool_t rv;
1712 NYD2_ENTER;
1714 #ifdef HAVE_SETENV
1715 rv = (setenv(avcp->avc_name, avp->av_value, 1) == 0);
1716 #else
1717 cp = sstrdup(savecatsep(avcp->avc_name, '=', avp->av_value));
1719 if((rv = (putenv(cp) == 0))){
1720 char *ocp;
1722 ocp = avp->av_env;
1723 avp->av_env = cp;
1724 cp = ocp;
1727 if(cp != NULL)
1728 n_free(cp);
1729 #endif
1730 NYD2_LEAVE;
1731 return rv;
1734 static bool_t
1735 a_amv_var_clear(struct a_amv_var_carrier *avcp, bool_t force_env){
1736 struct a_amv_var **avpp, *avp;
1737 struct a_amv_var_map const *avmp;
1738 bool_t rv;
1739 NYD2_ENTER;
1741 rv = FAL0;
1743 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1744 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NODEL) != 0 &&
1745 !(n_pstate & n_PS_ROOT))){
1746 n_err(_("Variable may not be unset: %s\n"), avcp->avc_name);
1747 goto jleave;
1749 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1750 !a_amv_var_check_vips(a_AMV_VIP_CLEAR, avcp->avc_okey, NULL))){
1751 n_err(_("Clearance of variable aborted: %s\n"), avcp->avc_name);
1752 goto jleave;
1756 rv = TRU1;
1758 if(n_UNLIKELY(!a_amv_var_lookup(avcp, TRUM1))){
1759 assert(avcp->avc_var == NULL);
1760 /* This may be a clearance request from the command line, via -S, and we
1761 * need to keep track of that! Unfortunately we are not prepared for
1762 * this, really, so we need to create a fake entry that is known and
1763 * handled correctly by the lowermost variable layer! */
1764 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
1765 (n_poption & n_PO_S_FLAG_TEMPORARY)) Jfreeze:{
1766 size_t l;
1768 l = strlen(avcp->avc_name) +1;
1769 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1770 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1771 *avpp = avp;
1772 avp->av_value = n_UNCONST(n_empty); /* Sth. covered by _var_free()! */
1773 avp->av_flags = (avmp != NULL ? avmp->avm_flags : 0) |
1774 a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET;
1775 memcpy(avp->av_name, avcp->avc_name, l);
1777 if(force_env || (avmp != NULL && (avmp->avm_flags & a_AMV_VF_ENV)))
1778 goto jforce_env;
1779 }else if(force_env){
1780 jforce_env:
1781 rv = a_amv_var__clearenv(avcp->avc_name, NULL);
1782 }else if(!(n_pstate & (n_PS_ROOT | n_PS_ROBOT)) && (n_poption & n_PO_D_V))
1783 n_err(_("Cannot unset undefined variable: %s\n"), avcp->avc_name);
1784 goto jleave;
1785 }else if((avp = avcp->avc_var) == (struct a_amv_var*)-1){
1786 /* Clearance request from command line, via -S? As above.. */
1787 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1788 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0))
1789 goto Jfreeze;
1790 avcp->avc_var = NULL;
1791 if(force_env)
1792 goto jforce_env;
1793 goto jleave;
1796 /* If this setting has been established via -S and we still have not reached
1797 * the _STARTED_CONFIG program state, silently ignore request!
1798 * XXX All this is very complicated for the tenth of a second */
1799 if(avp != NULL &&
1800 n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1801 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1802 if((n_pstate & n_PS_ROOT) ||
1803 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1804 (n_poption & n_PO_S_FLAG_TEMPORARY))){
1805 /* Be aware this may turn a set into an unset! */
1806 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET)){
1807 if(avp->av_flags & a_AMV_VF_DEFVAL)
1808 goto jdefault_path;
1809 a_amv_var_free(avp->av_value);
1810 avp->av_flags |= a_AMV_VF_EXT_FROZEN_UNSET;
1811 avp->av_value = n_UNCONST(n_empty); /* _var_free() covered */
1812 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1813 goto jforce_env;
1815 goto jleave;
1817 if(n_poption & n_PO_D_VV)
1818 n_err(_("Temporarily frozen by -S, not `unset'ting: %s\n"),
1819 avcp->avc_name);
1820 goto jleave;
1822 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1825 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1826 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1827 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1829 jdefault_path:
1830 avp = avcp->avc_var;
1831 avcp->avc_var = NULL;
1832 avpp = &a_amv_vars[avcp->avc_prime];
1833 assert(*avpp == avp); /* (always listhead after lookup()) */
1834 *avpp = (*avpp)->av_link;
1836 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1837 rv = a_amv_var__clearenv(avp->av_name, avp);
1838 a_amv_var_free(avp->av_value);
1839 n_free(avp);
1841 /* XXX Fun part, extremely simple-minded for now: if this variable has
1842 * XXX a default value, immediately reinstantiate it! TODO Heh? */
1843 if(n_UNLIKELY(avmp != NULL && (avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)){
1844 a_amv_var_lookup(avcp, TRU1);
1845 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
1846 (n_poption & n_PO_S_FLAG_TEMPORARY))
1847 avp->av_flags |= a_AMV_VF_EXT_FROZEN;
1849 jleave:
1850 NYD2_LEAVE;
1851 return rv;
1854 static bool_t
1855 a_amv_var__clearenv(char const *name, struct a_amv_var *avp){
1856 #ifndef HAVE_SETENV
1857 extern char **environ;
1858 char **ecpp;
1859 #endif
1860 bool_t rv;
1861 NYD2_ENTER;
1863 #ifdef HAVE_SETENV
1864 n_UNUSED(avp);
1865 unsetenv(name);
1866 rv = TRU1;
1867 #else
1868 rv = FAL0;
1869 ecpp = environ;
1871 if(avp != NULL && avp->av_env != NULL){
1872 for(; *ecpp != NULL; ++ecpp)
1873 if(*ecpp == avp->av_env){
1875 ecpp[0] = ecpp[1];
1876 while(*ecpp++ != NULL);
1877 n_free(avp->av_env);
1878 avp->av_env = NULL;
1879 rv = TRU1;
1880 break;
1882 }else{
1883 size_t l;
1885 if((l = strlen(name)) > 0){
1886 for(; *ecpp != NULL; ++ecpp)
1887 if(!strncmp(*ecpp, name, l) && (*ecpp)[l] == '='){
1889 ecpp[0] = ecpp[1];
1890 while(*ecpp++ != NULL);
1891 rv = TRU1;
1892 break;
1896 #endif /* HAVE_SETENV */
1897 NYD2_LEAVE;
1898 return rv;
1901 static void
1902 a_amv_var_show_all(void){
1903 struct n_string msg, *msgp;
1904 FILE *fp;
1905 size_t no, i;
1906 struct a_amv_var *avp;
1907 char const **vacp, **cap;
1908 NYD2_ENTER;
1910 if((fp = Ftmp(NULL, "setlist", OF_RDWR | OF_UNLINK | OF_REGISTER)) == NULL){
1911 n_perr(_("`set' list: cannot create temporary file"), 0);
1912 goto jleave;
1915 /* We need to instantiate first-time-inits and default values here, so that
1916 * they will be regular members of our _vars[] table */
1917 for(i = a_AMV_VAR_I3VALS_CNT; i-- > 0;)
1918 n_var_oklook(a_amv_var_i3vals[i].avdv_okey);
1919 for(i = a_AMV_VAR_DEFVALS_CNT; i-- > 0;)
1920 n_var_oklook(a_amv_var_defvals[i].avdv_okey);
1922 for(no = i = 0; i < a_AMV_PRIME; ++i)
1923 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
1924 ++no;
1925 no += a_AMV_VAR_VIRTS_CNT;
1927 vacp = n_autorec_alloc(no * sizeof(*vacp));
1929 for(cap = vacp, i = 0; i < a_AMV_PRIME; ++i)
1930 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
1931 *cap++ = avp->av_name;
1932 for(i = a_AMV_VAR_VIRTS_CNT; i-- > 0;)
1933 *cap++ = a_amv_var_virts[i].avv_var->av_name;
1935 if(no > 1)
1936 qsort(vacp, no, sizeof *vacp, &a_amv_var__show_cmp);
1938 msgp = &msg;
1939 msgp = n_string_reserve(n_string_creat(msgp), 80);
1940 for(i = 0, cap = vacp; no != 0; ++cap, --no)
1941 i += a_amv_var_show(*cap, fp, msgp);
1942 n_string_gut(&msg);
1944 page_or_print(fp, i);
1945 Fclose(fp);
1946 jleave:
1947 NYD2_LEAVE;
1950 static int
1951 a_amv_var__show_cmp(void const *s1, void const *s2){
1952 int rv;
1953 NYD2_ENTER;
1955 rv = strcmp(*(char**)n_UNCONST(s1), *(char**)n_UNCONST(s2));
1956 NYD2_LEAVE;
1957 return rv;
1960 static size_t
1961 a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp){
1962 struct a_amv_var_carrier avc;
1963 char const *quote;
1964 struct a_amv_var *avp;
1965 bool_t isset;
1966 size_t i;
1967 NYD2_ENTER;
1969 msgp = n_string_trunc(msgp, 0);
1970 i = 0;
1972 a_amv_var_revlookup(&avc, name);
1973 isset = a_amv_var_lookup(&avc, FAL0);
1974 avp = avc.avc_var;
1976 if(n_poption & n_PO_D_V){
1977 if(avc.avc_map == NULL){
1978 msgp = n_string_push_cp(msgp, "#assembled variable with value");
1979 i = 1;
1980 }else{
1981 struct{
1982 ui16_t flag;
1983 char msg[22];
1984 } const tbase[] = {
1985 {a_AMV_VF_VIRT, "virtual"},
1986 {a_AMV_VF_CHAIN, "variable chain"},
1987 {a_AMV_VF_RDONLY, "read-only"},
1988 {a_AMV_VF_NODEL, "nodelete"},
1989 {a_AMV_VF_I3VAL, "initial-value"},
1990 {a_AMV_VF_DEFVAL, "default-value"},
1991 {a_AMV_VF_IMPORT, "import-environ-first\0"}, /* assert NUL in max */
1992 {a_AMV_VF_ENV, "sync-environ"},
1993 {a_AMV_VF_NOLOPTS, "no-localopts"},
1994 {a_AMV_VF_NOTEMPTY, "notempty"},
1995 {a_AMV_VF_NOCNTRLS, "no-control-chars"},
1996 {a_AMV_VF_NUM, "number"},
1997 {a_AMV_VF_POSNUM, "positive-number"},
1998 }, *tp;
2000 for(tp = tbase; PTRCMP(tp, <, &tbase[n_NELEM(tbase)]); ++tp)
2001 if(isset ? (avp->av_flags & tp->flag)
2002 : (avc.avc_map->avm_flags & tp->flag)){
2003 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2004 msgp = n_string_push_cp(msgp, tp->msg);
2008 if(avp != NULL && (avp->av_flags & a_AMV_VF_EXT_FROZEN)){
2009 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2010 msgp = n_string_push_cp(msgp, "(un)?set via -S");
2013 if(i > 0)
2014 msgp = n_string_push_cp(msgp, "\n ");
2017 /* (Read-only variables are generally shown via comments..) */
2018 if(!isset || (avp->av_flags & a_AMV_VF_RDONLY)){
2019 msgp = n_string_push_c(msgp, n_ns[0]);
2020 if(!isset){
2021 if(avc.avc_map != NULL && (avc.avc_map->avm_flags & a_AMV_VF_BOOL))
2022 msgp = n_string_push_cp(msgp, "boolean; ");
2023 msgp = n_string_push_cp(msgp, "variable not set: ");
2024 msgp = n_string_push_cp(msgp, n_shexp_quote_cp(name, FAL0));
2025 goto jleave;
2029 n_UNINIT(quote, NULL);
2030 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2031 quote = n_shexp_quote_cp(avp->av_value, TRU1);
2032 if(strcmp(quote, avp->av_value))
2033 msgp = n_string_push_cp(msgp, "wysh ");
2034 }else if(n_poption & n_PO_D_V)
2035 msgp = n_string_push_cp(msgp, "wysh "); /* (for shell-style comment) */
2036 if(avp->av_flags & a_AMV_VF_EXT_LINKED)
2037 msgp = n_string_push_cp(msgp, "environ ");
2038 msgp = n_string_push_cp(msgp, "set ");
2039 msgp = n_string_push_cp(msgp, name);
2040 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2041 msgp = n_string_push_c(msgp, '=');
2042 msgp = n_string_push_cp(msgp, quote);
2043 }else if(n_poption & n_PO_D_V)
2044 msgp = n_string_push_cp(msgp, " #boolean");
2046 jleave:
2047 msgp = n_string_push_c(msgp, '\n');
2048 fputs(n_string_cp(msgp), fp);
2049 NYD2_ENTER;
2050 return (i > 0 ? 2 : 1);
2053 static bool_t
2054 a_amv_var_c_set(char **ap, bool_t issetenv){
2055 char *cp, *cp2, *varbuf, c;
2056 size_t errs;
2057 NYD2_ENTER;
2059 errs = 0;
2060 jouter:
2061 while((cp = *ap++) != NULL){
2062 /* Isolate key */
2063 cp2 = varbuf = n_autorec_alloc(strlen(cp) +1);
2065 for(; (c = *cp) != '=' && c != '\0'; ++cp){
2066 if(cntrlchar(c) || spacechar(c)){
2067 n_err(_("Variable name with control or space character ignored: "
2068 "%s\n"), ap[-1]);
2069 ++errs;
2070 goto jouter;
2072 *cp2++ = c;
2074 *cp2 = '\0';
2075 if(c == '\0')
2076 cp = n_UNCONST(n_empty);
2077 else
2078 ++cp;
2080 if(varbuf == cp2){
2081 n_err(_("Empty variable name ignored\n"));
2082 ++errs;
2083 }else{
2084 struct a_amv_var_carrier avc;
2085 bool_t isunset;
2087 if((isunset = (varbuf[0] == 'n' && varbuf[1] == 'o')))
2088 varbuf = &varbuf[2];
2090 a_amv_var_revlookup(&avc, varbuf);
2092 if(isunset)
2093 errs += !a_amv_var_clear(&avc, issetenv);
2094 else
2095 errs += !a_amv_var_set(&avc, cp, issetenv);
2098 NYD2_LEAVE;
2099 return (errs == 0);
2102 FL int
2103 c_define(void *v){
2104 int rv;
2105 char **args;
2106 NYD_ENTER;
2108 rv = 1;
2110 if((args = v)[0] == NULL){
2111 rv = (a_amv_mac_show(a_AMV_MF_NONE) == FAL0);
2112 goto jleave;
2115 if(args[1] == NULL || args[1][0] != '{' || args[1][1] != '\0' ||
2116 args[2] != NULL){
2117 n_err(_("Synopsis: define: <name> {\n"));
2118 goto jleave;
2121 rv = (a_amv_mac_def(args[0], a_AMV_MF_NONE) == FAL0);
2122 jleave:
2123 NYD_LEAVE;
2124 return rv;
2127 FL int
2128 c_undefine(void *v){
2129 int rv;
2130 char **args;
2131 NYD_ENTER;
2133 rv = 0;
2134 args = v;
2136 rv |= !a_amv_mac_undef(*args, a_AMV_MF_NONE);
2137 while(*++args != NULL);
2138 NYD_LEAVE;
2139 return rv;
2142 FL int
2143 c_call(void *vp){
2144 int rv;
2145 NYD_ENTER;
2147 rv = a_amv_mac_call(vp, FAL0);
2148 NYD_LEAVE;
2149 return rv;
2152 FL int
2153 c_call_if(void *vp){
2154 int rv;
2155 NYD_ENTER;
2157 rv = a_amv_mac_call(vp, TRU1);
2158 NYD_LEAVE;
2159 return rv;
2162 FL int
2163 c_account(void *v){
2164 struct a_amv_mac_call_args *amcap;
2165 struct a_amv_mac *amp;
2166 char **args;
2167 int rv, i, oqf, nqf;
2168 NYD_ENTER;
2170 rv = 1;
2172 if((args = v)[0] == NULL){
2173 rv = (a_amv_mac_show(a_AMV_MF_ACCOUNT) == FAL0);
2174 goto jleave;
2177 if(args[1] && args[1][0] == '{' && args[1][1] == '\0'){
2178 if(args[2] != NULL){
2179 n_err(_("Synopsis: account: <name> {\n"));
2180 goto jleave;
2182 if(!asccasecmp(args[0], ACCOUNT_NULL)){
2183 n_err(_("`account': cannot use reserved name: %s\n"),
2184 ACCOUNT_NULL);
2185 goto jleave;
2187 rv = (a_amv_mac_def(args[0], a_AMV_MF_ACCOUNT) == FAL0);
2188 goto jleave;
2191 if(n_pstate & n_PS_HOOK_MASK){
2192 n_err(_("`account': cannot change account from within a hook\n"));
2193 goto jleave;
2196 save_mbox_for_possible_quitstuff();
2198 amp = NULL;
2199 if(asccasecmp(args[0], ACCOUNT_NULL) != 0 &&
2200 (amp = a_amv_mac_lookup(args[0], NULL, a_AMV_MF_ACCOUNT)) == NULL){
2201 n_err(_("`account': account does not exist: %s\n"), args[0]);
2202 goto jleave;
2205 oqf = savequitflags();
2207 /* Shutdown the active account */
2208 if(a_amv_acc_curr != NULL){
2209 char const *cp;
2210 char *var;
2212 /* Is there a cleanup hook? */
2213 var = savecat("on-account-cleanup-", a_amv_acc_curr->am_name);
2214 if((cp = n_var_vlook(var, FAL0)) != NULL ||
2215 (cp = ok_vlook(on_account_cleanup)) != NULL){
2216 struct a_amv_mac *amphook;
2218 if((amphook = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) != NULL){
2219 amcap = n_lofi_alloc(sizeof *amcap);
2220 memset(amcap, 0, sizeof *amcap);
2221 amcap->amca_name = cp;
2222 amcap->amca_amp = amphook;
2223 amcap->amca_unroller = &a_amv_acc_curr->am_lopts;
2224 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2225 amcap->amca_no_xcall = TRU1;
2226 n_pstate |= n_PS_HOOK;
2227 rv = a_amv_mac_exec(amcap);
2228 n_pstate &= ~n_PS_HOOK_MASK;
2229 }else
2230 n_err(_("*on-account-leave* hook %s does not exist\n"),
2231 n_shexp_quote_cp(cp, FAL0));
2234 /* `localopts'? */
2235 if(a_amv_acc_curr->am_lopts != NULL)
2236 a_amv_lopts_unroll(&a_amv_acc_curr->am_lopts);
2238 /* For accounts this lingers */
2239 --a_amv_acc_curr->am_refcnt;
2240 if(a_amv_acc_curr->am_flags & a_AMV_MF_DELETE)
2241 a_amv_mac_free(a_amv_acc_curr);
2244 a_amv_acc_curr = amp;
2246 /* And switch to any non-"null" account */
2247 if(amp != NULL){
2248 assert(amp->am_lopts == NULL);
2249 amcap = n_lofi_alloc(sizeof *amcap);
2250 memset(amcap, 0, sizeof *amcap);
2251 amcap->amca_name = amp->am_name;
2252 amcap->amca_amp = amp;
2253 amcap->amca_unroller = &amp->am_lopts;
2254 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2255 amcap->amca_no_xcall = TRU1;
2256 ++amp->am_refcnt; /* We may not run 0 to avoid being deleted! */
2257 if(!a_amv_mac_exec(amcap)){
2258 /* XXX account switch incomplete, unroll? */
2259 n_err(_("`account': failed to switch to account: %s\n"), amp->am_name);
2260 goto jleave;
2264 n_PS_ROOT_BLOCK((amp != NULL ? ok_vset(account, amp->am_name)
2265 : ok_vclear(account)));
2267 if(n_psonce & n_PSO_STARTED){
2268 assert(!(n_pstate & n_PS_HOOK_MASK));
2269 nqf = savequitflags(); /* TODO obsolete (leave -> void -> new box!) */
2270 restorequitflags(oqf);
2271 if((i = setfile("%", 0)) < 0)
2272 goto jleave;
2273 temporary_folder_hook_check(FAL0);
2274 if(i > 0 && !ok_blook(emptystart))
2275 goto jleave;
2276 n_folder_announce(n_ANNOUNCE_CHANGE);
2277 restorequitflags(nqf);
2279 rv = 0;
2280 jleave:
2281 NYD_LEAVE;
2282 return rv;
2285 FL int
2286 c_unaccount(void *v){
2287 int rv;
2288 char **args;
2289 NYD_ENTER;
2291 rv = 0;
2292 args = v;
2294 rv |= !a_amv_mac_undef(*args, a_AMV_MF_ACCOUNT);
2295 while(*++args != NULL);
2296 NYD_LEAVE;
2297 return rv;
2300 FL int
2301 c_localopts(void *vp){
2302 enum a_amv_loflags alf, alm;
2303 char const **argv;
2304 int rv;
2305 NYD_ENTER;
2307 rv = 1;
2309 if(a_amv_lopts == NULL){
2310 n_err(_("Cannot use `localopts' in this context "
2311 "(not in `define' or `account', nor special hook)\n"));
2312 goto jleave;
2315 if((argv = vp)[1] == NULL || is_asccaseprefix((++argv)[-1], "scope"))
2316 alf = alm = a_AMV_LF_SCOPE;
2317 else if(is_asccaseprefix(argv[-1], "call"))
2318 alf = a_AMV_LF_CALL, alm = a_AMV_LF_CALL_MASK;
2319 else if(is_asccaseprefix(argv[-1], "call-fixate"))
2320 alf = a_AMV_LF_CALL_FIXATE, alm = a_AMV_LF_CALL_MASK;
2321 else{
2322 n_err(_("Synopsis: localopts: [<scope|call|call-fixate>] <boolean>\n"));
2323 goto jleave;
2326 if(alf == a_AMV_LF_SCOPE &&
2327 (a_amv_lopts->as_loflags & a_AMV_LF_SCOPE_FIXATE)){
2328 if(n_poption & n_PO_D_V)
2329 n_err(_("Cannot turn off `localopts', setting is fixated\n"));
2330 goto jleave;
2333 a_amv_lopts->as_loflags &= ~alm;
2334 if(boolify(*argv, UIZ_MAX, FAL0) > 0)
2335 a_amv_lopts->as_loflags |= alf;
2336 rv = 0;
2337 jleave:
2338 NYD_LEAVE;
2339 return rv;
2342 FL int
2343 c_shift(void *vp){
2344 struct a_amv_pospar *appp;
2345 ui16_t i;
2346 int rv;
2347 NYD_ENTER;
2349 rv = 1;
2351 if((vp = *(char**)vp) == NULL)
2352 i = 1;
2353 else{
2354 si16_t sib;
2356 if((n_idec_si16_cp(&sib, vp, 10, NULL
2357 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2358 ) != n_IDEC_STATE_CONSUMED || sib < 0){
2359 n_err(_("`shift': invalid argument: %s\n"), vp);
2360 goto jleave;
2362 i = (ui16_t)sib;
2365 /* If in in a macro/xy */
2366 if(a_amv_lopts != NULL){
2367 struct a_amv_mac const *amp;
2368 struct a_amv_mac_call_args *amcap;
2370 /* Explicitly do allow `vpospar' created things! */
2371 amp = (amcap = a_amv_lopts->as_amcap)->amca_amp;
2372 if((amp == NULL || amcap->amca_ps_hook_mask ||
2373 (amp->am_flags & a_AMV_MF_TYPE_MASK) == a_AMV_MF_ACCOUNT) &&
2374 amcap->amca_pospar.app_not_heap){
2375 n_err(_("Cannot use `shift' in `account's or hook macros etc.\n"));
2376 goto jleave;
2378 appp = &amcap->amca_pospar;
2379 }else
2380 appp = &a_amv_pospar;
2382 if(i > appp->app_count){
2383 n_err(_("`shift': cannot shift %hu of %hu parameters\n"),
2384 i, appp->app_count);
2385 goto jleave;
2386 }else{
2387 appp->app_idx += i;
2388 appp->app_count -= i;
2389 rv = 0;
2391 jleave:
2392 NYD_LEAVE;
2393 return rv;
2396 FL int
2397 c_return(void *vp){ /* TODO the exit status should be m_si64! */
2398 int rv;
2399 NYD_ENTER;
2401 if(a_amv_lopts != NULL){
2402 char const **argv;
2404 n_go_input_force_eof();
2405 n_pstate_err_no = n_ERR_NONE;
2406 rv = 0;
2408 if((argv = vp)[0] != NULL){
2409 si32_t i;
2411 if((n_idec_si32_cp(&i, argv[0], 10, NULL
2412 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2413 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2414 rv = (int)i;
2415 else{
2416 n_err(_("`return': return value argument is invalid: %s\n"),
2417 argv[0]);
2418 n_pstate_err_no = n_ERR_INVAL;
2419 rv = 1;
2422 if(argv[1] != NULL){
2423 if((n_idec_si32_cp(&i, argv[1], 10, NULL
2424 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2425 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2426 n_pstate_err_no = i;
2427 else{
2428 n_err(_("`return': error number argument is invalid: %s\n"),
2429 argv[1]);
2430 n_pstate_err_no = n_ERR_INVAL;
2431 rv = 1;
2435 }else{
2436 n_err(_("Can only use `return' in a macro\n"));
2437 n_pstate_err_no = n_ERR_OPNOTSUPP;
2438 rv = 1;
2440 NYD_LEAVE;
2441 return rv;
2444 FL bool_t
2445 temporary_folder_hook_check(bool_t nmail){ /* TODO temporary, v15: drop */
2446 struct a_amv_mac_call_args *amcap;
2447 struct a_amv_mac *amp;
2448 size_t len;
2449 char const *cp;
2450 char *var;
2451 bool_t rv;
2452 NYD_ENTER;
2454 rv = TRU1;
2455 var = n_autorec_alloc(len = strlen(mailname) + sizeof("folder-hook-") -1 +1);
2457 /* First try the fully resolved path */
2458 snprintf(var, len, "folder-hook-%s", mailname);
2459 if((cp = n_var_vlook(var, FAL0)) != NULL)
2460 goto jmac;
2462 /* If we are under *folder*, try the usual +NAME syntax, too */
2463 if(displayname[0] == '+'){
2464 char *x;
2466 for(x = &mailname[len]; x != mailname; --x)
2467 if(x[-1] == '/'){
2468 snprintf(var, len, "folder-hook-+%s", x);
2469 if((cp = n_var_vlook(var, FAL0)) != NULL)
2470 goto jmac;
2471 break;
2475 /* Plain *folder-hook* is our last try */
2476 if((cp = ok_vlook(folder_hook)) == NULL)
2477 goto jleave;
2479 jmac:
2480 if((amp = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) == NULL){
2481 n_err(_("Cannot call *folder-hook* for %s: macro does not exist: %s\n"),
2482 n_shexp_quote_cp(displayname, FAL0), cp);
2483 rv = FAL0;
2484 goto jleave;
2487 amcap = n_lofi_alloc(sizeof *amcap);
2488 memset(amcap, 0, sizeof *amcap);
2489 amcap->amca_name = cp;
2490 amcap->amca_amp = amp;
2491 n_pstate &= ~n_PS_HOOK_MASK;
2492 if(nmail){
2493 amcap->amca_unroller = NULL;
2494 n_pstate |= n_PS_HOOK_NEWMAIL;
2495 }else{
2496 amcap->amca_unroller = &a_amv_folder_hook_lopts;
2497 n_pstate |= n_PS_HOOK;
2499 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2500 amcap->amca_ps_hook_mask = TRU1;
2501 amcap->amca_no_xcall = TRU1;
2502 rv = a_amv_mac_exec(amcap);
2503 n_pstate &= ~n_PS_HOOK_MASK;
2505 jleave:
2506 NYD_LEAVE;
2507 return rv;
2510 FL void
2511 temporary_folder_hook_unroll(void){ /* XXX intermediate hack */
2512 NYD_ENTER;
2513 if(a_amv_folder_hook_lopts != NULL){
2514 void *save = a_amv_lopts;
2516 a_amv_lopts = NULL;
2517 a_amv_lopts_unroll(&a_amv_folder_hook_lopts);
2518 assert(a_amv_folder_hook_lopts == NULL);
2519 a_amv_lopts = save;
2521 NYD_LEAVE;
2524 FL void
2525 temporary_compose_mode_hook_call(char const *macname,
2526 void (*hook_pre)(void *), void *hook_arg){
2527 /* TODO compose_mode_hook_call() temporary, v15: generalize; see a_GO_SPLICE
2528 * TODO comment in go.c for the right way of doing things! */
2529 static struct a_amv_lostack *cmh_losp;
2530 struct a_amv_mac_call_args *amcap;
2531 struct a_amv_mac *amp;
2532 NYD_ENTER;
2534 amp = NULL;
2536 if(macname == (char*)-1){
2537 a_amv_mac__finalize(cmh_losp);
2538 cmh_losp = NULL;
2539 }else if(macname != NULL &&
2540 (amp = a_amv_mac_lookup(macname, NULL, a_AMV_MF_NONE)) == NULL)
2541 n_err(_("Cannot call *on-compose-**: macro does not exist: %s\n"),
2542 macname);
2543 else{
2544 amcap = n_lofi_alloc(sizeof *amcap);
2545 memset(amcap, 0, sizeof *amcap);
2546 amcap->amca_name = (macname != NULL) ? macname
2547 : "*on-compose-splice(-shell)?*";
2548 amcap->amca_amp = amp;
2549 amcap->amca_unroller = &a_amv_compose_lopts;
2550 amcap->amca_hook_pre = hook_pre;
2551 amcap->amca_hook_arg = hook_arg;
2552 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2553 amcap->amca_ps_hook_mask = TRU1;
2554 amcap->amca_no_xcall = TRU1;
2555 n_pstate &= ~n_PS_HOOK_MASK;
2556 n_pstate |= n_PS_HOOK;
2557 if(macname != NULL)
2558 a_amv_mac_exec(amcap);
2559 else{
2560 cmh_losp = n_lofi_alloc(sizeof *cmh_losp);
2561 memset(cmh_losp, 0, sizeof *cmh_losp);
2562 cmh_losp->as_global_saved = a_amv_lopts;
2563 cmh_losp->as_lopts = *(cmh_losp->as_amcap = amcap)->amca_unroller;
2564 cmh_losp->as_loflags = a_AMV_LF_SCOPE_FIXATE;
2565 a_amv_lopts = cmh_losp;
2568 NYD_LEAVE;
2571 FL void
2572 temporary_compose_mode_hook_unroll(void){ /* XXX intermediate hack */
2573 NYD_ENTER;
2574 if(a_amv_compose_lopts != NULL){
2575 void *save = a_amv_lopts;
2577 a_amv_lopts = NULL;
2578 a_amv_lopts_unroll(&a_amv_compose_lopts);
2579 assert(a_amv_compose_lopts == NULL);
2580 a_amv_lopts = save;
2582 NYD_LEAVE;
2585 FL bool_t
2586 n_var_is_user_writable(char const *name){
2587 struct a_amv_var_carrier avc;
2588 struct a_amv_var_map const *avmp;
2589 bool_t rv;
2590 NYD_ENTER;
2592 a_amv_var_revlookup(&avc, name);
2593 if((avmp = avc.avc_map) == NULL)
2594 rv = TRU1;
2595 else
2596 rv = ((avmp->avm_flags & (a_AMV_VF_BOOL | a_AMV_VF_RDONLY)) == 0);
2597 NYD_LEAVE;
2598 return rv;
2601 FL char *
2602 n_var_oklook(enum okeys okey){
2603 struct a_amv_var_carrier avc;
2604 char *rv;
2605 struct a_amv_var_map const *avmp;
2606 NYD_ENTER;
2608 avc.avc_map = avmp = &a_amv_var_map[okey];
2609 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2610 avc.avc_hash = avmp->avm_hash;
2611 avc.avc_okey = okey;
2613 if(a_amv_var_lookup(&avc, FAL0))
2614 rv = avc.avc_var->av_value;
2615 else
2616 rv = NULL;
2617 NYD_LEAVE;
2618 return rv;
2621 FL bool_t
2622 n_var_okset(enum okeys okey, uintptr_t val){
2623 struct a_amv_var_carrier avc;
2624 bool_t ok;
2625 struct a_amv_var_map const *avmp;
2626 NYD_ENTER;
2628 avc.avc_map = avmp = &a_amv_var_map[okey];
2629 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2630 avc.avc_hash = avmp->avm_hash;
2631 avc.avc_okey = okey;
2633 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val), FAL0);
2634 NYD_LEAVE;
2635 return ok;
2638 FL bool_t
2639 n_var_okclear(enum okeys okey){
2640 struct a_amv_var_carrier avc;
2641 bool_t rv;
2642 struct a_amv_var_map const *avmp;
2643 NYD_ENTER;
2645 avc.avc_map = avmp = &a_amv_var_map[okey];
2646 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2647 avc.avc_hash = avmp->avm_hash;
2648 avc.avc_okey = okey;
2650 rv = a_amv_var_clear(&avc, FAL0);
2651 NYD_LEAVE;
2652 return rv;
2655 FL char const *
2656 n_var_vlook(char const *vokey, bool_t try_getenv){
2657 struct a_amv_var_carrier avc;
2658 char const *rv;
2659 NYD_ENTER;
2661 a_amv_var_revlookup(&avc, vokey);
2663 switch((enum a_amv_var_special_category)avc.avc_special_cat){
2664 default: /* silence CC */
2665 case a_AMV_VSC_NONE:
2666 if(a_amv_var_lookup(&avc, FAL0))
2667 rv = avc.avc_var->av_value;
2668 /* Only check the environment for something that is otherwise unknown */
2669 else if(try_getenv && avc.avc_map == NULL)
2670 rv = getenv(vokey);
2671 else
2672 rv = NULL;
2673 break;
2674 case a_AMV_VSC_GLOBAL:
2675 rv = a_amv_var_vsc_global(&avc);
2676 break;
2677 case a_AMV_VSC_MULTIPLEX:
2678 rv = a_amv_var_vsc_multiplex(&avc);
2679 break;
2680 case a_AMV_VSC_POSPAR:
2681 case a_AMV_VSC_POSPAR_ENV:
2682 rv = a_amv_var_vsc_pospar(&avc);
2683 break;
2685 NYD_LEAVE;
2686 return rv;
2689 FL bool_t
2690 n_var_vexplode(void const **cookie){
2691 struct a_amv_pospar *appp;
2692 NYD_ENTER;
2694 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
2695 : &a_amv_pospar;
2696 *cookie = (appp->app_count > 0) ? &appp->app_dat[appp->app_idx] : NULL;
2697 NYD_LEAVE;
2698 return (*cookie != NULL);
2701 FL bool_t
2702 n_var_vset(char const *vokey, uintptr_t val){
2703 struct a_amv_var_carrier avc;
2704 bool_t ok;
2705 NYD_ENTER;
2707 a_amv_var_revlookup(&avc, vokey);
2709 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val), FAL0);
2710 NYD_LEAVE;
2711 return ok;
2714 FL bool_t
2715 n_var_vclear(char const *vokey){
2716 struct a_amv_var_carrier avc;
2717 bool_t ok;
2718 NYD_ENTER;
2720 a_amv_var_revlookup(&avc, vokey);
2722 ok = a_amv_var_clear(&avc, FAL0);
2723 NYD_LEAVE;
2724 return ok;
2727 #ifdef HAVE_SOCKETS
2728 FL char *
2729 n_var_xoklook(enum okeys okey, struct url const *urlp,
2730 enum okey_xlook_mode oxm){
2731 struct a_amv_var_carrier avc;
2732 struct str const *us;
2733 size_t nlen;
2734 char *nbuf, *rv;
2735 struct a_amv_var_map const *avmp;
2736 NYD_ENTER;
2738 assert(oxm & (OXM_PLAIN | OXM_H_P | OXM_U_H_P));
2740 /* For simplicity: allow this case too */
2741 if(!(oxm & (OXM_H_P | OXM_U_H_P))){
2742 nbuf = NULL;
2743 goto jplain;
2746 avc.avc_map = avmp = &a_amv_var_map[okey];
2747 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2748 avc.avc_okey = okey;
2750 us = (oxm & OXM_U_H_P) ? &urlp->url_u_h_p : &urlp->url_h_p;
2751 nlen = strlen(avc.avc_name);
2752 nbuf = n_lofi_alloc(nlen + 1 + us->l +1);
2753 memcpy(nbuf, avc.avc_name, nlen);
2754 nbuf[nlen++] = '-';
2756 /* One of .url_u_h_p and .url_h_p we test in here */
2757 memcpy(nbuf + nlen, us->s, us->l +1);
2758 avc.avc_name = a_amv_var_canonify(nbuf);
2759 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2760 if(a_amv_var_lookup(&avc, FAL0))
2761 goto jvar;
2763 /* The second */
2764 if(oxm & OXM_H_P){
2765 us = &urlp->url_h_p;
2766 memcpy(nbuf + nlen, us->s, us->l +1);
2767 avc.avc_name = a_amv_var_canonify(nbuf);
2768 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2769 if(a_amv_var_lookup(&avc, FAL0)){
2770 jvar:
2771 rv = avc.avc_var->av_value;
2772 goto jleave;
2776 jplain:
2777 rv = (oxm & OXM_PLAIN) ? n_var_oklook(okey) : NULL;
2778 jleave:
2779 if(nbuf != NULL)
2780 n_lofi_free(nbuf);
2781 NYD_LEAVE;
2782 return rv;
2784 #endif /* HAVE_SOCKETS */
2786 FL int
2787 c_set(void *v){
2788 char **ap;
2789 int err;
2790 NYD_ENTER;
2792 if(*(ap = v) == NULL){
2793 a_amv_var_show_all();
2794 err = 0;
2795 }else
2796 err = !a_amv_var_c_set(ap, FAL0);
2797 NYD_LEAVE;
2798 return err;
2801 FL int
2802 c_unset(void *v){
2803 char **ap;
2804 int err;
2805 NYD_ENTER;
2807 for(err = 0, ap = v; *ap != NULL; ++ap)
2808 err |= !n_var_vclear(*ap);
2809 NYD_LEAVE;
2810 return err;
2813 FL int
2814 c_varshow(void *v){
2815 char **ap;
2816 NYD_ENTER;
2818 if(*(ap = v) == NULL)
2819 v = NULL;
2820 else{
2821 struct n_string msg, *msgp = &msg;
2823 msgp = n_string_creat(msgp);
2824 for(; *ap != NULL; ++ap)
2825 a_amv_var_show(*ap, n_stdout, msgp);
2826 n_string_gut(msgp);
2828 NYD_LEAVE;
2829 return (v == NULL ? !STOP : !OKAY); /* xxx 1:bad 0:good -- do some */
2832 FL int
2833 c_varedit(void *v){
2834 struct a_amv_var_carrier avc;
2835 FILE *of, *nf;
2836 char *val, **argv;
2837 int err;
2838 sighandler_type sigint;
2839 NYD_ENTER;
2841 sigint = safe_signal(SIGINT, SIG_IGN);
2843 for(err = 0, argv = v; *argv != NULL; ++argv){
2844 memset(&avc, 0, sizeof avc);
2846 a_amv_var_revlookup(&avc, *argv);
2848 if(avc.avc_map != NULL){
2849 if(avc.avc_map->avm_flags & a_AMV_VF_BOOL){
2850 n_err(_("`varedit': cannot edit boolean variable: %s\n"),
2851 avc.avc_name);
2852 continue;
2854 if(avc.avc_map->avm_flags & a_AMV_VF_RDONLY){
2855 n_err(_("`varedit': cannot edit read-only variable: %s\n"),
2856 avc.avc_name);
2857 continue;
2861 a_amv_var_lookup(&avc, FAL0);
2863 if((of = Ftmp(NULL, "varedit", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
2864 NULL){
2865 n_perr(_("`varedit': cannot create temporary file"), 0);
2866 err = 1;
2867 break;
2868 }else if(avc.avc_var != NULL && *(val = avc.avc_var->av_value) != '\0' &&
2869 sizeof *val != fwrite(val, strlen(val), sizeof *val, of)){
2870 n_perr(_("`varedit' failed to write old value to temporary file"), 0);
2871 Fclose(of);
2872 err = 1;
2873 continue;
2876 fflush_rewind(of);
2877 nf = run_editor(of, (off_t)-1, 'e', FAL0, NULL, NULL, SEND_MBOX, sigint);
2878 Fclose(of);
2880 if(nf != NULL){
2881 int c;
2882 char const *varres;
2883 off_t l;
2885 l = fsize(nf);
2886 if(UICMP(64, l, >=, UIZ_MAX -42)){
2887 n_err(_("`varedit': not enough memory to store variable: %s\n"),
2888 avc.avc_name);
2889 varres = n_empty;
2890 err = 1;
2891 }else{
2892 varres = val = n_autorec_alloc(l +1);
2893 for(; l > 0 && (c = getc(nf)) != EOF; --l)
2894 *val++ = c;
2895 *val++ = '\0';
2896 if(l != 0){
2897 n_err(_("`varedit': I/O while reading new value of: %s\n"),
2898 avc.avc_name);
2899 err = 1;
2903 if(!a_amv_var_set(&avc, varres, FAL0))
2904 err = 1;
2906 Fclose(nf);
2907 }else{
2908 n_err(_("`varedit': cannot start $EDITOR\n"));
2909 err = 1;
2910 break;
2914 safe_signal(SIGINT, sigint);
2915 NYD_LEAVE;
2916 return err;
2919 FL int
2920 c_environ(void *v){
2921 struct a_amv_var_carrier avc;
2922 int err;
2923 char **ap;
2924 bool_t islnk;
2925 NYD_ENTER;
2927 if((islnk = is_asccaseprefix(*(ap = v), "link")) ||
2928 is_asccaseprefix(*ap, "unlink")){
2929 for(err = 0; *++ap != NULL;){
2930 a_amv_var_revlookup(&avc, *ap);
2932 if(a_amv_var_lookup(&avc, FAL0) && (islnk ||
2933 (avc.avc_var->av_flags & a_AMV_VF_EXT_LINKED))){
2934 if(!islnk){
2935 avc.avc_var->av_flags &= ~a_AMV_VF_EXT_LINKED;
2936 continue;
2937 }else if(avc.avc_var->av_flags &
2938 (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)){
2939 if(n_poption & n_PO_D_V)
2940 n_err(_("`environ': link: already established: %s\n"), *ap);
2941 continue;
2943 avc.avc_var->av_flags |= a_AMV_VF_EXT_LINKED;
2944 if(!(avc.avc_var->av_flags & a_AMV_VF_ENV))
2945 a_amv_var__putenv(&avc, avc.avc_var);
2946 }else if(!islnk){
2947 n_err(_("`environ': unlink: no link established: %s\n"), *ap);
2948 err = 1;
2949 }else{
2950 char const *evp = getenv(*ap);
2952 if(evp != NULL)
2953 err |= !a_amv_var_set(&avc, evp, TRU1);
2954 else{
2955 n_err(_("`environ': link: cannot link to non-existent: %s\n"),
2956 *ap);
2957 err = 1;
2961 }else if(is_asccaseprefix(*ap, "set"))
2962 err = !a_amv_var_c_set(++ap, TRU1);
2963 else if(is_asccaseprefix(*ap, "unset")){
2964 for(err = 0; *++ap != NULL;){
2965 a_amv_var_revlookup(&avc, *ap);
2967 if(!a_amv_var_clear(&avc, TRU1))
2968 err = 1;
2970 }else{
2971 n_err(_("Synopsis: environ: <link|set|unset> <variable>...\n"));
2972 err = 1;
2974 NYD_LEAVE;
2975 return err;
2978 FL int
2979 c_vexpr(void *v){ /* TODO POSIX expr(1) comp. exit status; overly complicat. */
2980 size_t i;
2981 enum n_idec_state ids;
2982 si64_t lhv, rhv;
2983 char op, varbuf[64 + 64 / 8 +1];
2984 char const **argv, *varname, *varres, *cp;
2985 enum{
2986 a_ERR = 1u<<0,
2987 a_SOFTOVERFLOW = 1u<<1,
2988 a_ISNUM = 1u<<2,
2989 a_ISDECIMAL = 1u<<3, /* Print only decimal result */
2990 a_SATURATED = 1u<<4,
2991 a_ICASE = 1u<<5,
2992 a_UNSIGNED = 1u<<6, /* Unsigned right shift (share bit ok) */
2993 a_TMP = 1u<<30
2994 } f;
2995 NYD_ENTER;
2997 f = a_ERR;
2998 argv = v;
2999 varname = (n_pstate & n_PS_ARGMOD_VPUT) ? *argv++ : NULL;
3000 n_UNINIT(varres, n_empty);
3002 if((cp = argv[0])[0] == '\0')
3003 goto jesubcmd;
3005 if(cp[1] == '\0'){
3006 op = cp[0];
3007 jnumop:
3008 f |= a_ISNUM;
3009 switch(op){
3010 case '=':
3011 case '~':
3012 if(argv[1] == NULL || argv[2] != NULL)
3013 goto jesynopsis;
3015 if(*(cp = *++argv) == '\0')
3016 lhv = 0;
3017 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
3018 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3019 ) != n_IDEC_STATE_CONSUMED){
3020 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3021 goto jenum_range;
3022 f |= a_SOFTOVERFLOW;
3023 break;
3025 if(op == '~')
3026 lhv = ~lhv;
3027 break;
3029 case '+':
3030 case '-':
3031 case '*':
3032 case '/':
3033 case '%':
3034 case '|':
3035 case '&':
3036 case '^':
3037 case '<':
3038 case '>':
3039 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3040 goto jesynopsis;
3041 else{
3042 char xop;
3044 if(*(cp = *++argv) == '\0')
3045 lhv = 0;
3046 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
3047 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3048 ) != n_IDEC_STATE_CONSUMED){
3049 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3050 goto jenum_range;
3051 f |= a_SOFTOVERFLOW;
3052 break;
3055 if(*(cp = *++argv) == '\0')
3056 rhv = 0;
3057 else if(((ids = n_idec_si64_cp(&rhv, cp, 0, NULL)
3058 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3059 ) != n_IDEC_STATE_CONSUMED){
3060 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3061 goto jenum_range;
3062 f |= a_SOFTOVERFLOW;
3063 lhv = rhv;
3064 break;
3067 xop = op;
3068 jnumop_again:
3069 switch(xop){
3070 case '+':
3071 if(rhv < 0){
3072 if(rhv != SI64_MIN){
3073 rhv = -rhv;
3074 xop = '-';
3075 goto jnumop_again;
3076 }else if(lhv < 0)
3077 goto jenum_plusminus;
3078 else if(lhv == 0){
3079 lhv = rhv;
3080 break;
3083 if(SI64_MAX - rhv < lhv)
3084 goto jenum_plusminus;
3085 lhv += rhv;
3086 break;
3087 case '-':
3088 if(rhv < 0){
3089 if(rhv != SI64_MIN){
3090 rhv = -rhv;
3091 xop = '+';
3092 goto jnumop_again;
3093 }else if(lhv > 0)
3094 goto jenum_plusminus;
3095 else if(lhv == 0){
3096 lhv = rhv;
3097 break;
3100 if(SI64_MIN + rhv > lhv){
3101 jenum_plusminus:
3102 if(!(f & a_SATURATED))
3103 goto jenum_overflow;
3104 f |= a_SOFTOVERFLOW;
3105 lhv = (lhv < 0 || xop == '-') ? SI64_MIN : SI64_MAX;
3106 }else
3107 lhv -= rhv;
3108 break;
3109 case '*':
3110 /* Will the result be positive? */
3111 if((lhv < 0) == (rhv < 0)){
3112 if(lhv > 0){
3113 lhv = -lhv;
3114 rhv = -rhv;
3116 if(rhv != 0 && lhv != 0 && SI64_MAX / rhv > lhv){
3117 if(!(f & a_SATURATED))
3118 goto jenum_overflow;
3119 f |= a_SOFTOVERFLOW;
3120 lhv = SI64_MAX;
3121 }else
3122 lhv *= rhv;
3123 }else{
3124 if(rhv > 0){
3125 if(lhv != 0 && SI64_MIN / lhv < rhv){
3126 if(!(f & a_SATURATED))
3127 goto jenum_overflow;
3128 f |= a_SOFTOVERFLOW;
3129 lhv = SI64_MIN;
3130 }else
3131 lhv *= rhv;
3132 }else{
3133 if(rhv != 0 && lhv != 0 && SI64_MIN / rhv < lhv){
3134 if(!(f & a_SATURATED))
3135 goto jenum_overflow;
3136 f |= a_SOFTOVERFLOW;
3137 lhv = SI64_MIN;
3138 }else
3139 lhv *= rhv;
3142 break;
3143 case '/':
3144 if(rhv == 0){
3145 if(!(f & a_SATURATED))
3146 goto jenum_range;
3147 f |= a_SOFTOVERFLOW;
3148 lhv = SI64_MAX;
3149 }else
3150 lhv /= rhv;
3151 break;
3152 case '%':
3153 if(rhv == 0){
3154 if(!(f & a_SATURATED))
3155 goto jenum_range;
3156 f |= a_SOFTOVERFLOW;
3157 lhv = SI64_MAX;
3158 }else
3159 lhv %= rhv;
3160 break;
3161 case '|':
3162 lhv |= rhv;
3163 break;
3164 case '&':
3165 lhv &= rhv;
3166 break;
3167 case '^':
3168 lhv ^= rhv;
3169 break;
3170 case '<':
3171 case '>':
3172 if(!(f & a_TMP))
3173 goto jesubcmd;
3174 if(rhv > 63){ /* xxx 63? */
3175 if(!(f & a_SATURATED))
3176 goto jenum_overflow;
3177 rhv = 63;
3179 if(op == '<')
3180 lhv <<= (ui8_t)rhv;
3181 else if(f & a_UNSIGNED)
3182 lhv = (ui64_t)lhv >> (ui8_t)rhv;
3183 else
3184 lhv >>= (ui8_t)rhv;
3185 break;
3188 break;
3189 default:
3190 goto jesubcmd;
3192 }else if(cp[2] == '\0' && cp[1] == '@'){
3193 f |= a_SATURATED;
3194 op = cp[0];
3195 goto jnumop;
3196 }else if(cp[0] == '<'){
3197 if(*++cp != '<')
3198 goto jesubcmd;
3199 if(*++cp == '@'){
3200 f |= a_SATURATED;
3201 ++cp;
3203 if(*cp != '\0')
3204 goto jesubcmd;
3205 f |= a_TMP;
3206 op = '<';
3207 goto jnumop;
3208 }else if(cp[0] == '>'){
3209 if(*++cp != '>')
3210 goto jesubcmd;
3211 if(*++cp == '>'){
3212 f |= a_UNSIGNED;
3213 ++cp;
3215 if(*cp == '@'){
3216 f |= a_SATURATED;
3217 ++cp;
3219 if(*cp != '\0')
3220 goto jesubcmd;
3221 f |= a_TMP;
3222 op = '>';
3223 goto jnumop;
3224 }else if(is_asccaseprefix(cp, "length")){
3225 f |= a_ISNUM | a_ISDECIMAL;
3226 if(argv[1] == NULL || argv[2] != NULL)
3227 goto jesynopsis;
3229 i = strlen(*++argv);
3230 if(UICMP(64, i, >, SI64_MAX))
3231 goto jestr_overflow;
3232 lhv = (si64_t)i;
3233 }else if(is_asccaseprefix(cp, "hash")){
3234 f |= a_ISNUM | a_ISDECIMAL;
3235 if(argv[1] == NULL || argv[2] != NULL)
3236 goto jesynopsis;
3238 i = n_torek_hash(*++argv);
3239 lhv = (si64_t)i;
3240 }else if(is_asccaseprefix(cp, "find")){
3241 f |= a_ISNUM | a_ISDECIMAL;
3242 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3243 goto jesynopsis;
3245 if((cp = strstr(argv[1], argv[2])) == NULL)
3246 goto jestr_nodata;
3247 i = PTR2SIZE(cp - argv[1]);
3248 if(UICMP(64, i, >, SI64_MAX))
3249 goto jestr_overflow;
3250 lhv = (si64_t)i;
3251 }else if(is_asccaseprefix(cp, "ifind")){
3252 f |= a_ISNUM | a_ISDECIMAL;
3253 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3254 goto jesynopsis;
3256 if((cp = asccasestr(argv[1], argv[2])) == NULL)
3257 goto jestr_nodata;
3258 i = PTR2SIZE(cp - argv[1]);
3259 if(UICMP(64, i, >, SI64_MAX))
3260 goto jestr_overflow;
3261 lhv = (si64_t)i;
3262 }else if(is_asccaseprefix(cp, "substring")){
3263 if(argv[1] == NULL || argv[2] == NULL)
3264 goto jesynopsis;
3265 if(argv[3] != NULL && argv[4] != NULL)
3266 goto jesynopsis;
3268 i = strlen(varres = *++argv);
3270 if(*(cp = *++argv) == '\0')
3271 lhv = 0;
3272 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3273 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3274 ) != n_IDEC_STATE_CONSUMED)
3275 goto jestr_numrange;
3276 if(lhv < 0){
3277 if(UICMP(64, i, <, -lhv))
3278 goto jesubstring_off;
3279 lhv = i + lhv;
3281 if(UICMP(64, i, >=, lhv)){
3282 i -= lhv;
3283 varres += lhv;
3284 }else{
3285 jesubstring_off:
3286 if(n_poption & n_PO_D_V)
3287 n_err(_("`vexpr': substring: offset argument too large: %s\n"),
3288 n_shexp_quote_cp(argv[-1], FAL0));
3289 f |= a_SOFTOVERFLOW;
3292 if(argv[1] != NULL){
3293 if(*(cp = *++argv) == '\0')
3294 lhv = 0;
3295 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3296 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3297 ) != n_IDEC_STATE_CONSUMED)
3298 goto jestr_numrange;
3299 if(lhv < 0){
3300 if(UICMP(64, i, <, -lhv))
3301 goto jesubstring_len;
3302 lhv = i + lhv;
3304 if(UICMP(64, i, >=, lhv)){
3305 if(UICMP(64, i, !=, lhv))
3306 varres = savestrbuf(varres, (size_t)lhv);
3307 }else{
3308 jesubstring_len:
3309 if(n_poption & n_PO_D_V)
3310 n_err(_("`vexpr': substring: length argument too large: %s\n"),
3311 n_shexp_quote_cp(argv[-2], FAL0));
3312 f |= a_SOFTOVERFLOW;
3315 }else if(is_asccaseprefix(cp, "trim")) Jtrim: {
3316 struct str trim;
3317 enum n_str_trim_flags stf;
3319 if(argv[1] == NULL || argv[2] != NULL)
3320 goto jesynopsis;
3322 if(is_asccaseprefix(cp, "trim-front"))
3323 stf = n_STR_TRIM_FRONT;
3324 else if(is_asccaseprefix(cp, "trim-end"))
3325 stf = n_STR_TRIM_END;
3326 else
3327 stf = n_STR_TRIM_BOTH;
3329 trim.l = strlen(trim.s = n_UNCONST(argv[1]));
3330 (void)n_str_trim(&trim, stf);
3331 varres = savestrbuf(trim.s, trim.l);
3332 }else if(is_asccaseprefix(cp, "trim-front"))
3333 goto Jtrim;
3334 else if(is_asccaseprefix(cp, "trim-end"))
3335 goto Jtrim;
3336 else if(is_asccaseprefix(cp, "random")){
3337 if(argv[1] == NULL || argv[2] != NULL)
3338 goto jesynopsis;
3340 if((n_idec_si64_cp(&lhv, argv[1], 0, NULL
3341 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3342 ) != n_IDEC_STATE_CONSUMED || lhv < 0 || lhv > PATH_MAX)
3343 goto jestr_numrange;
3344 if(lhv == 0)
3345 lhv = NAME_MAX;
3346 varres = n_random_create_cp((size_t)lhv, NULL);
3347 }else if(is_asccaseprefix(cp, "file-expand")){
3348 if(argv[1] == NULL || argv[2] != NULL)
3349 goto jesynopsis;
3351 if((varres = fexpand(argv[1], FEXP_NVAR | FEXP_NOPROTO)) == NULL)
3352 goto jestr_nodata;
3353 }else if(is_asccaseprefix(cp, "makeprint")){
3354 struct str sin, sout;
3356 if(argv[1] == NULL || argv[2] != NULL)
3357 goto jesynopsis;
3359 /* XXX using strlen for `vexpr makeprint' is wrong for UTF-16 */
3360 sin.l = strlen(sin.s = n_UNCONST(argv[1]));
3361 makeprint(&sin, &sout);
3362 varres = savestrbuf(sout.s, sout.l);
3363 n_free(sout.s);
3364 /* TODO `vexpr': (wide) string length, find, etc!! */
3365 #ifdef HAVE_REGEX
3366 }else if(is_asccaseprefix(cp, "regex")) Jregex:{
3367 regmatch_t rema[1 + n_VEXPR_REGEX_MAX];
3368 regex_t re;
3369 int reflrv;
3371 f |= a_ISNUM | a_ISDECIMAL;
3372 if(argv[1] == NULL || argv[2] == NULL ||
3373 (argv[3] != NULL && argv[4] != NULL))
3374 goto jesynopsis;
3376 reflrv = REG_EXTENDED;
3377 if(f & a_ICASE)
3378 reflrv |= REG_ICASE;
3379 if((reflrv = regcomp(&re, argv[2], reflrv))){
3380 n_err(_("`vexpr': invalid regular expression: %s: %s\n"),
3381 n_shexp_quote_cp(argv[2], FAL0), n_regex_err_to_doc(NULL, reflrv));
3382 assert(f & a_ERR);
3383 n_pstate_err_no = n_ERR_INVAL;
3384 goto jestr;
3386 reflrv = regexec(&re, argv[1], n_NELEM(rema), rema, 0);
3387 regfree(&re);
3388 if(reflrv == REG_NOMATCH)
3389 goto jestr_nodata;
3391 /* Search only? Else replace, which is a bit */
3392 if(argv[3] == NULL){
3393 if(UICMP(64, rema[0].rm_so, >, SI64_MAX))
3394 goto jestr_overflow;
3395 lhv = (si64_t)rema[0].rm_so;
3396 }else{
3397 /* We need to setup some kind of pseudo macro environment for this */
3398 struct a_amv_lostack los;
3399 struct a_amv_mac_call_args amca;
3400 char const **reargv;
3402 memset(&amca, 0, sizeof amca);
3403 amca.amca_name = savestrbuf(&argv[1][rema[0].rm_so],
3404 rema[0].rm_eo - rema[0].rm_so);
3405 amca.amca_amp = a_AMV_MACKY_MACK;
3406 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3408 amca.amca_pospar.app_count = (ui32_t)i - 1;
3409 amca.amca_pospar.app_not_heap = TRU1;
3410 amca.amca_pospar.app_dat =
3411 reargv = n_autorec_alloc(sizeof(char*) * i);
3412 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3413 *reargv++ = savestrbuf(&argv[1][rema[i].rm_so],
3414 rema[i].rm_eo - rema[i].rm_so);
3415 *reargv = NULL;
3417 memset(&los, 0, sizeof los);
3418 hold_all_sigs(); /* TODO DISLIKE! */
3419 los.as_global_saved = a_amv_lopts;
3420 los.as_amcap = &amca;
3421 los.as_up = los.as_global_saved;
3422 a_amv_lopts = &los;
3424 /* C99 */{
3425 struct str templ;
3426 struct n_string s_b;
3427 enum n_shexp_state shs;
3429 templ.s = n_UNCONST(argv[3]);
3430 templ.l = UIZ_MAX;
3431 shs = n_shexp_parse_token((n_SHEXP_PARSE_LOG |
3432 n_SHEXP_PARSE_IGNORE_EMPTY | n_SHEXP_PARSE_QUOTE_AUTO_FIXED |
3433 n_SHEXP_PARSE_QUOTE_AUTO_DSQ),
3434 n_string_creat_auto(&s_b), &templ, NULL);
3435 if((shs & (n_SHEXP_STATE_ERR_MASK | n_SHEXP_STATE_STOP)
3436 ) == n_SHEXP_STATE_STOP){
3437 varres = n_string_cp(&s_b);
3438 n_string_drop_ownership(&s_b);
3439 }else
3440 varres = NULL;
3443 a_amv_lopts = los.as_global_saved;
3444 rele_all_sigs(); /* TODO DISLIKE! */
3446 if(varres == NULL)
3447 goto jestr_nodata;
3448 f &= ~(a_ISNUM | a_ISDECIMAL);
3450 }else if(is_asccaseprefix(argv[0], "iregex")){
3451 f |= a_ICASE;
3452 goto Jregex;
3453 #endif /* HAVE_REGEX */
3454 }else
3455 goto jesubcmd;
3457 n_pstate_err_no = (f & a_SOFTOVERFLOW) ? n_ERR_OVERFLOW : n_ERR_NONE;
3458 f &= ~a_ERR;
3460 /* Generate the variable value content for numerics.
3461 * Anticipate in our handling below! (Don't do needless work) */
3462 jleave:
3463 if((f & a_ISNUM) && ((f & a_ISDECIMAL) || varname != NULL)){
3464 snprintf(varbuf, sizeof varbuf, "%" PRId64 , lhv);
3465 varres = varbuf;
3468 if(varname == NULL){
3469 /* If there was no error and we are printing a numeric result, print some
3470 * more bases for the fun of it */
3471 if((f & (a_ERR | a_ISNUM | a_ISDECIMAL)) == a_ISNUM){
3472 size_t j;
3474 for(j = 1, i = 0; i < 64; ++i){
3475 varbuf[63 + 64 / 8 -j - i] = (lhv & ((ui64_t)1 << i)) ? '1' : '0';
3476 if((i & 7) == 7 && i != 63){
3477 ++j;
3478 varbuf[63 + 64 / 8 -j - i] = ' ';
3481 varbuf[64 + 64 / 8 -1] = '\0';
3482 if(fprintf(n_stdout,
3483 "%s\n0%" PRIo64 " | 0x%" PRIX64 " | %" PRId64 "\n",
3484 varbuf, lhv, lhv, lhv) < 0){
3485 n_pstate_err_no = n_err_no;
3486 f |= a_ERR;
3488 }else if(varres != NULL && fprintf(n_stdout, "%s\n", varres) < 0){
3489 n_pstate_err_no = n_err_no;
3490 f |= a_ERR;
3492 }else if(!n_var_vset(varname, (uintptr_t)varres)){
3493 n_pstate_err_no = n_ERR_NOTSUP;
3494 f |= a_ERR;
3496 NYD_LEAVE;
3497 return (f & a_ERR) ? 1 : 0;
3499 jerr:
3500 f = a_ERR | a_ISNUM;
3501 lhv = -1;
3502 goto jleave;
3503 jesubcmd:
3504 n_err(_("`vexpr': invalid subcommand: %s\n"),
3505 n_shexp_quote_cp(*argv, FAL0));
3506 n_pstate_err_no = n_ERR_INVAL;
3507 goto jerr;
3508 jesynopsis:
3509 n_err(_("Synopsis: vexpr: <operator> <:argument:>\n"));
3510 n_pstate_err_no = n_ERR_INVAL;
3511 goto jerr;
3513 jenum_range:
3514 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3515 n_shexp_quote_cp(*argv, FAL0));
3516 n_pstate_err_no = n_ERR_RANGE;
3517 goto jerr;
3518 jenum_overflow:
3519 n_err(_("`vexpr': expression overflows datatype: %" PRId64 " %c %" PRId64
3520 "\n"), lhv, op, rhv);
3521 n_pstate_err_no = n_ERR_OVERFLOW;
3522 goto jerr;
3524 jestr_numrange:
3525 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3526 n_shexp_quote_cp(*argv, FAL0));
3527 n_pstate_err_no = n_ERR_RANGE;
3528 goto jestr;
3529 jestr_overflow:
3530 n_err(_("`vexpr': string length or offset overflows datatype\n"));
3531 n_pstate_err_no = n_ERR_OVERFLOW;
3532 goto jestr;
3533 jestr_nodata:
3534 n_pstate_err_no = n_ERR_NODATA;
3535 /* FALLTHRU */
3536 jestr:
3537 varres = n_empty;
3538 f &= ~a_ISNUM;
3539 f |= a_ERR;
3540 goto jleave;
3543 FL int
3544 c_vpospar(void *v){
3545 struct n_cmd_arg *cap;
3546 size_t i;
3547 struct a_amv_pospar *appp;
3548 enum{
3549 a_NONE = 0,
3550 a_ERR = 1u<<0,
3551 a_SET = 1u<<1,
3552 a_CLEAR = 1u<<2,
3553 a_QUOTE = 1u<<3
3554 } f;
3555 char const *varres;
3556 struct n_cmd_arg_ctx *cacp;
3557 NYD_ENTER;
3559 n_pstate_err_no = n_ERR_NONE;
3560 n_UNINIT(varres, n_empty);
3561 cacp = v;
3562 cap = cacp->cac_arg;
3564 if(is_asccaseprefix(cap->ca_arg.ca_str.s, "set"))
3565 f = a_SET;
3566 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "clear"))
3567 f = a_CLEAR;
3568 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "quote"))
3569 f = a_QUOTE;
3570 else{
3571 n_err(_("`vpospar': invalid subcommand: %s\n"),
3572 n_shexp_quote_cp(cap->ca_arg.ca_str.s, FAL0));
3573 n_pstate_err_no = n_ERR_INVAL;
3574 f = a_ERR;
3575 goto jleave;
3577 --cacp->cac_no;
3579 if((f & (a_CLEAR | a_QUOTE)) && cap->ca_next != NULL){
3580 n_err(_("`vpospar': `%s': takes no argument\n"), cap->ca_arg.ca_str.s);
3581 n_pstate_err_no = n_ERR_INVAL;
3582 f = a_ERR;
3583 goto jleave;
3586 cap = cap->ca_next;
3588 /* If in a macro, we need to overwrite the local instead of global argv */
3589 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
3590 : &a_amv_pospar;
3592 if(f & (a_SET | a_CLEAR)){
3593 if(cacp->cac_vput != NULL)
3594 n_err(_("`vpospar': `vput' only supported for `quote' subcommand\n"));
3595 if(!appp->app_not_heap && appp->app_maxcount > 0){
3596 for(i = appp->app_maxcount; i-- != 0;)
3597 n_free(n_UNCONST(appp->app_dat[i]));
3598 n_free(appp->app_dat);
3600 memset(appp, 0, sizeof *appp);
3602 if(f & a_SET){
3603 if((i = cacp->cac_no) > a_AMV_POSPAR_MAX){
3604 n_err(_("`vpospar': overflow: %" PRIuZ " arguments!\n"), i);
3605 n_pstate_err_no = n_ERR_OVERFLOW;
3606 f = a_ERR;
3607 goto jleave;
3610 memset(appp, 0, sizeof *appp);
3611 if(i > 0){
3612 appp->app_maxcount = appp->app_count = (ui16_t)i;
3613 /* XXX Optimize: store it all in one chunk! */
3614 ++i;
3615 i *= sizeof *appp->app_dat;
3616 appp->app_dat = n_alloc(i);
3618 for(i = 0; cap != NULL; ++i, cap = cap->ca_next){
3619 appp->app_dat[i] = n_alloc(cap->ca_arg.ca_str.l +1);
3620 memcpy(n_UNCONST(appp->app_dat[i]), cap->ca_arg.ca_str.s,
3621 cap->ca_arg.ca_str.l +1);
3624 appp->app_dat[i] = NULL;
3627 }else{
3628 if(appp->app_count == 0)
3629 varres = n_empty;
3630 else{
3631 struct str in;
3632 struct n_string s, *sp;
3633 char sep1, sep2;
3635 sp = n_string_creat_auto(&s);
3637 sep1 = *ok_vlook(ifs);
3638 sep2 = *ok_vlook(ifs_ws);
3639 if(sep1 == sep2)
3640 sep2 = '\0';
3641 if(sep1 == '\0')
3642 sep1 = ' ';
3644 for(i = 0; i < appp->app_count; ++i){
3645 if(sp->s_len){
3646 if(!n_string_can_swallow(sp, 2))
3647 goto jeover;
3648 sp = n_string_push_c(sp, sep1);
3649 if(sep2 != '\0')
3650 sp = n_string_push_c(sp, sep2);
3652 in.l = strlen(in.s = n_UNCONST(appp->app_dat[i + appp->app_idx]));
3654 if(!n_string_can_swallow(sp, in.l)){
3655 jeover:
3656 n_err(_("`vpospar': overflow: string too long!\n"));
3657 n_pstate_err_no = n_ERR_OVERFLOW;
3658 f = a_ERR;
3659 goto jleave;
3661 sp = n_shexp_quote(sp, &in, TRU1);
3664 varres = n_string_cp(sp);
3667 if(cacp->cac_vput == NULL){
3668 if(fprintf(n_stdout, "%s\n", varres) < 0){
3669 n_pstate_err_no = n_err_no;
3670 f |= a_ERR;
3672 }else if(!n_var_vset(cacp->cac_vput, (uintptr_t)varres)){
3673 n_pstate_err_no = n_ERR_NOTSUP;
3674 f |= a_ERR;
3677 jleave:
3678 NYD_LEAVE;
3679 return (f & a_ERR) ? 1 : 0;
3682 /* s-it-mode */