Introduce *asksend*, by default enabled
[s-mailx.git] / accmacvar.c
blob7c15efdd39e3a1e0761f74e027c8a8e855eeebec
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 . should be recursive environment based.
8 *@ TODO Otherwise, the `localopts' should be an attribute of the go.c
9 *@ TODO command context, so that it belongs to the execution context
10 *@ TODO we are running in, instead of being global data. See, e.g.,
11 *@ TODO the a_GO_SPLICE comment in go.c.
12 *@ TODO . once we can have non-fatal !0 returns for commands, we should
13 *@ TODO return error if "(environ)? unset" goes for non-existent.
15 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
16 * Copyright (c) 2012 - 2017 Steffen (Daode) Nurpmeso <steffen@sdaoden.eu>.
19 * Copyright (c) 1980, 1993
20 * The Regents of the University of California. All rights reserved.
22 * Redistribution and use in source and binary forms, with or without
23 * modification, are permitted provided that the following conditions
24 * are met:
25 * 1. Redistributions of source code must retain the above copyright
26 * notice, this list of conditions and the following disclaimer.
27 * 2. Redistributions in binary form must reproduce the above copyright
28 * notice, this list of conditions and the following disclaimer in the
29 * documentation and/or other materials provided with the distribution.
30 * 3. Neither the name of the University nor the names of its contributors
31 * may be used to endorse or promote products derived from this software
32 * without specific prior written permission.
34 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
35 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
36 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
37 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
38 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
39 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
40 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
41 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
42 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
43 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
44 * SUCH DAMAGE.
46 #undef n_FILE
47 #define n_FILE accmacvar
49 #ifndef HAVE_AMALGAMATION
50 # include "nail.h"
51 #endif
53 #if !defined HAVE_SETENV && !defined HAVE_PUTENV
54 # error Exactly one of HAVE_SETENV and HAVE_PUTENV
55 #endif
57 /* Positional parameter maximum (macro arguments, `vexpr' "splitifs") */
58 #define a_AMV_POSPAR_MAX SI16_MAX
60 /* Special "pseudo macro" that stabs you from the back */
61 #define a_AMV_MACKY_MACK ((struct a_amv_mac*)-1)
63 /* Note: changing the hash function must be reflected in `vexpr' "hash",
64 * because that is used by the hashtable creator scripts! */
65 #define a_AMV_PRIME HSHSIZE
66 #define a_AMV_NAME2HASH(N) n_torek_hash(N)
67 #define a_AMV_HASH2PRIME(H) ((H) % a_AMV_PRIME)
69 enum a_amv_mac_flags{
70 a_AMV_MF_NONE = 0,
71 a_AMV_MF_ACCOUNT = 1u<<0, /* This macro is an `account' */
72 a_AMV_MF_TYPE_MASK = a_AMV_MF_ACCOUNT,
73 a_AMV_MF_UNDEF = 1u<<1, /* Unlink after lookup */
74 a_AMV_MF_DELETE = 1u<<7, /* Delete in progress, free once refcnt==0 */
75 a_AMV_MF__MAX = 0xFFu
78 enum a_amv_loflags{
79 a_AMV_LF_NONE = 0,
80 a_AMV_LF_SCOPE = 1u<<0, /* Current scope `localopts' on */
81 a_AMV_LF_SCOPE_FIXATE = 1u<<1, /* Ditto, but fixated */
82 a_AMV_LF_SCOPE_MASK = a_AMV_LF_SCOPE | a_AMV_LF_SCOPE_FIXATE,
83 a_AMV_LF_CALL = 1u<<2, /* `localopts' on for `call'ed scopes */
84 a_AMV_LF_CALL_FIXATE = 1u<<3, /* Ditto, but fixated */
85 a_AMV_LF_CALL_MASK = a_AMV_LF_CALL | a_AMV_LF_CALL_FIXATE,
86 a_AMV_LF_CALL_TO_SCOPE_SHIFT = 2
89 /* make-okey-map.pl ensures that _VIRT implies _RDONLY and _NODEL, and that
90 * _IMPORT implies _ENV; it doesn't verify anything... */
91 enum a_amv_var_flags{
92 a_AMV_VF_NONE = 0,
93 a_AMV_VF_BOOL = 1u<<0, /* ok_b_* */
94 a_AMV_VF_VIRT = 1u<<1, /* "Stateless" automatic variable */
95 a_AMV_VF_NOLOPTS = 1u<<2, /* May not be tracked by `localopts' */
96 a_AMV_VF_RDONLY = 1u<<3, /* May not be set by user */
97 a_AMV_VF_NODEL = 1u<<4, /* May not be deleted */
98 a_AMV_VF_NOTEMPTY = 1u<<5, /* May not be assigned an empty value */
99 a_AMV_VF_NOCNTRLS = 1u<<6, /* Value may not contain control characters */
100 a_AMV_VF_NUM = 1u<<7, /* Value must be a 32-bit number */
101 a_AMV_VF_POSNUM = 1u<<8, /* Value must be positive 32-bit number */
102 a_AMV_VF_LOWER = 1u<<9, /* Values will be stored in a lowercase version */
103 a_AMV_VF_VIP = 1u<<10, /* Wants _var_check_vips() evaluation */
104 a_AMV_VF_IMPORT = 1u<<11, /* Import ONLY from environ (pre n_PSO_STARTED) */
105 a_AMV_VF_ENV = 1u<<12, /* Update environment on change */
106 a_AMV_VF_I3VAL = 1u<<13, /* Has an initial value */
107 a_AMV_VF_DEFVAL = 1u<<14, /* Has a default value */
108 a_AMV_VF_LINKED = 1u<<15, /* `environ' linked */
109 a_AMV_VF__MASK = (1u<<(15+1)) - 1
112 /* We support some special parameter names for one(+)-letter variable names;
113 * note these have counterparts in the code that manages shell expansion!
114 * All these special variables are solely backed by n_var_vlook(), and besides
115 * there is only a_amv_var_revlookup() which knows about them */
116 enum a_amv_var_special_category{
117 a_AMV_VSC_NONE, /* Normal variable, no special treatment */
118 a_AMV_VSC_GLOBAL, /* ${[?!]} are specially mapped, but global */
119 a_AMV_VSC_MULTIPLEX, /* ${^.*} circumflex accent multiplexer */
120 a_AMV_VSC_POSPAR, /* ${[1-9][0-9]*} positional parameters */
121 a_AMV_VSC_POSPAR_ENV /* ${[*@#]} positional parameter support variables */
124 enum a_amv_var_special_type{
125 /* _VSC_GLOBAL */
126 a_AMV_VST_QM, /* ? */
127 a_AMV_VST_EM, /* ! */
128 /* _VSC_MULTIPLEX */
129 /* This is special in that it is a multiplex indicator, the ^ is followed by
130 * a normal variable */
131 a_AMV_VST_CACC, /* ^ (circumflex accent) */
132 /* _VSC_POSPAR_ENV */
133 a_AMV_VST_STAR, /* * */
134 a_AMV_VST_AT, /* @ */
135 a_AMV_VST_NOSIGN /* # */
138 enum a_amv_var_vip_mode{
139 a_AMV_VIP_SET_PRE,
140 a_AMV_VIP_SET_POST,
141 a_AMV_VIP_CLEAR
144 struct a_amv_pospar{
145 ui16_t app_maxcount; /* == slots in .app_dat */
146 ui16_t app_count; /* Maximum a_AMV_POSPAR_MAX */
147 ui16_t app_idx; /* `shift' moves this one, decs .app_count */
148 bool_t app_not_heap; /* .app_dat stuff not on dynamically allocated */
149 ui8_t app__dummy[1];
150 char const **app_dat; /* NULL terminated (for "$@" explosion support) */
152 n_CTA(a_AMV_POSPAR_MAX <= SI16_MAX, "Limit exceeds datatype capabilities");
154 struct a_amv_mac{
155 struct a_amv_mac *am_next;
156 ui32_t am_maxlen; /* of any line in .am_line_dat */
157 ui32_t am_line_cnt; /* of *.am_line_dat (but NULL terminated) */
158 struct a_amv_mac_line **am_line_dat; /* TODO use deque? */
159 struct a_amv_var *am_lopts; /* `localopts' unroll list */
160 ui32_t am_refcnt; /* 0-based for `un{account,define}' purposes */
161 ui8_t am_flags; /* enum a_amv_mac_flags */
162 char am_name[n_VFIELD_SIZE(3)]; /* of this macro */
164 n_CTA(a_AMV_MF__MAX <= UI8_MAX, "Enumeration excesses storage datatype");
166 struct a_amv_mac_line{
167 ui32_t aml_len;
168 ui32_t aml_prespc; /* Number of leading SPC, for display purposes */
169 char aml_dat[n_VFIELD_SIZE(0)];
172 struct a_amv_mac_call_args{
173 char const *amca_name; /* For MACKY_MACK, this is *0*! */
174 struct a_amv_mac *amca_amp; /* "const", but for am_refcnt */
175 struct a_amv_var **amca_unroller;
176 void (*amca_hook_pre)(void *);
177 void *amca_hook_arg;
178 ui8_t amca_loflags;
179 bool_t amca_ps_hook_mask;
180 bool_t amca_no_xcall; /* We want n_GO_INPUT_NO_XCALL for this */
181 ui8_t amca__pad[5];
182 struct a_amv_pospar amca_pospar;
185 struct a_amv_lostack{
186 struct a_amv_lostack *as_global_saved; /* Saved global XXX due to jump */
187 struct a_amv_mac_call_args *as_amcap;
188 struct a_amv_lostack *as_up; /* Outer context */
189 struct a_amv_var *as_lopts;
190 ui8_t as_loflags; /* enum a_amv_mac_loflags */
191 ui8_t avs__pad[7];
194 struct a_amv_var{
195 struct a_amv_var *av_link;
196 char *av_value;
197 #ifdef HAVE_PUTENV
198 char *av_env; /* Actively managed putenv(3) memory */
199 #endif
200 ui16_t av_flags; /* enum a_amv_var_flags */
201 char av_name[n_VFIELD_SIZE(6)];
203 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
205 struct a_amv_var_map{
206 ui32_t avm_hash;
207 ui16_t avm_keyoff;
208 ui16_t avm_flags; /* enum a_amv_var_flags */
210 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
212 struct a_amv_var_virt{
213 ui32_t avv_okey;
214 ui8_t avv__dummy[4];
215 struct a_amv_var const *avv_var;
218 struct a_amv_var_defval{
219 ui32_t avdv_okey;
220 ui8_t avdv__pad[4];
221 char const *avdv_value; /* Only for !BOOL (otherwise plain existence) */
224 struct a_amv_var_carrier{
225 char const *avc_name;
226 ui32_t avc_hash;
227 ui32_t avc_prime;
228 struct a_amv_var *avc_var;
229 struct a_amv_var_map const *avc_map;
230 enum okeys avc_okey;
231 ui8_t avc__pad[1];
232 ui8_t avc_special_cat;
233 /* Numerical parameter name if .avc_special_cat=a_AMV_VSC_POSPAR,
234 * otherwise a enum a_amv_var_special_type */
235 ui16_t avc_special_prop;
238 /* Include constant make-okey-map.pl output, and the generated version data */
239 #include "gen-version.h"
240 #include "gen-okeys.h"
242 /* The currently active account */
243 static struct a_amv_mac *a_amv_acc_curr;
245 static struct a_amv_mac *a_amv_macs[a_AMV_PRIME]; /* TODO dynamically spaced */
247 /* Unroll list of currently running macro stack */
248 static struct a_amv_lostack *a_amv_lopts;
250 static struct a_amv_var *a_amv_vars[a_AMV_PRIME]; /* TODO dynamically spaced */
252 /* Global (i.e., non-local) a_AMV_VSC_POSPAR stack */
253 static struct a_amv_pospar a_amv_pospar;
255 /* TODO We really deserve localopts support for *folder-hook*s, so hack it in
256 * TODO today via a static lostack, it should be a field in mailbox, once that
257 * TODO is a real multi-instance object */
258 static struct a_amv_var *a_amv_folder_hook_lopts;
260 /* TODO Rather ditto (except for storage -> cmd_ctx), compose hooks */
261 static struct a_amv_var *a_amv_compose_lopts;
263 /* Lookup for macros/accounts: if newamp is not NULL it will be linked in the
264 * map, if _MF_UNDEF is set a possibly existing entry will be removed (first).
265 * Returns NULL if a lookup failed, or if newamp was set, the found entry in
266 * plain lookup cases or when _UNDEF was performed on a currently active entry
267 * (the entry will have been unlinked, and the _MF_DELETE will be honoured once
268 * the reference count reaches 0), and (*)-1 if an _UNDEF was performed */
269 static struct a_amv_mac *a_amv_mac_lookup(char const *name,
270 struct a_amv_mac *newamp, enum a_amv_mac_flags amf);
272 /* `call', `call_if' (and `xcall' via go.c -> c_call()) */
273 static int a_amv_mac_call(void *v, bool_t silent_nexist);
275 /* Execute a macro; amcap must reside in LOFI memory */
276 static bool_t a_amv_mac_exec(struct a_amv_mac_call_args *amcap);
278 static void a_amv_mac__finalize(void *vp);
280 /* User display helpers */
281 static bool_t a_amv_mac_show(enum a_amv_mac_flags amf);
283 /* _def() returns error for faulty definitions and already existing * names,
284 * _undef() returns error if a named thing doesn't exist */
285 static bool_t a_amv_mac_def(char const *name, enum a_amv_mac_flags amf);
286 static bool_t a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf);
288 /* */
289 static void a_amv_mac_free(struct a_amv_mac *amp);
291 /* Update replay-log */
292 static void a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
293 struct a_amv_var *oavp);
294 static void a_amv_lopts_unroll(struct a_amv_var **avpp);
296 /* Special cased value string allocation */
297 static char *a_amv_var_copy(char const *str);
298 static void a_amv_var_free(char *cp);
300 /* Check for special housekeeping. _VIP_SET_POST and _VIP_CLEAR do not fail
301 * (or propagate errors), _VIP_SET_PRE may and should case abortion */
302 static bool_t a_amv_var_check_vips(enum a_amv_var_vip_mode avvm,
303 enum okeys okey, char const *val);
305 /* _VF_NOCNTRLS, _VF_NUM / _VF_POSNUM */
306 static bool_t a_amv_var_check_nocntrls(char const *val);
307 static bool_t a_amv_var_check_num(char const *val, bool_t posnum);
309 /* If a variable name begins with a lowercase-character and contains at
310 * least one '@', it is converted to all-lowercase. This is necessary
311 * for lookups of names based on email addresses.
312 * Following the standard, only the part following the last '@' should
313 * be lower-cased, but practice has established otherwise here */
314 static char const *a_amv_var_canonify(char const *vn);
316 /* Try to reverse lookup an option name to an enum okeys mapping.
317 * Updates .avc_name and .avc_hash; .avc_map is NULL if none found */
318 static bool_t a_amv_var_revlookup(struct a_amv_var_carrier *avcp,
319 char const *name);
321 /* Lookup a variable from .avc_(map|name|hash), return whether it was found.
322 * Sets .avc_prime; .avc_var is NULL if not found.
323 * Here it is where we care for _I3VAL and _DEFVAL, too.
324 * An _I3VAL will be "consumed" as necessary anyway, but it won't be used to
325 * create a new variable if i3val_nonew is true; if i3val_nonew is TRUM1 then
326 * we set .avc_var to -1 and return true if that was the case, otherwise we'll
327 * return FAL0, then! */
328 static bool_t a_amv_var_lookup(struct a_amv_var_carrier *avcp,
329 bool_t i3val_nonew);
331 /* Lookup functions for special category variables, _pospar drives all
332 * positional parameter etc. special categories */
333 static char const *a_amv_var_vsc_global(struct a_amv_var_carrier *avcp);
334 static char const *a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp);
335 static char const *a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp);
337 /* Set var from .avc_(map|name|hash), return success */
338 static bool_t a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
339 bool_t force_env);
341 static bool_t a_amv_var__putenv(struct a_amv_var_carrier *avcp,
342 struct a_amv_var *avp);
344 /* Clear var from .avc_(map|name|hash); sets .avc_var=NULL, return success */
345 static bool_t a_amv_var_clear(struct a_amv_var_carrier *avcp, bool_t force_env);
347 static bool_t a_amv_var__clearenv(char const *name, char *value);
349 /* List all variables */
350 static void a_amv_var_show_all(void);
352 static int a_amv_var__show_cmp(void const *s1, void const *s2);
354 /* Actually do print one, return number of lines written */
355 static size_t a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp);
357 /* Shared c_set() and c_environ():set impl, return success */
358 static bool_t a_amv_var_c_set(char **ap, bool_t issetenv);
360 static struct a_amv_mac *
361 a_amv_mac_lookup(char const *name, struct a_amv_mac *newamp,
362 enum a_amv_mac_flags amf){
363 struct a_amv_mac *amp, **ampp;
364 ui32_t h;
365 enum a_amv_mac_flags save_amf;
366 NYD2_ENTER;
368 save_amf = amf;
369 amf &= a_AMV_MF_TYPE_MASK;
370 h = a_AMV_NAME2HASH(name);
371 h = a_AMV_HASH2PRIME(h);
372 ampp = &a_amv_macs[h];
374 for(amp = *ampp; amp != NULL; ampp = &(*ampp)->am_next, amp = amp->am_next){
375 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf &&
376 !strcmp(amp->am_name, name)){
377 if(n_LIKELY((save_amf & a_AMV_MF_UNDEF) == 0))
378 goto jleave;
380 *ampp = amp->am_next;
382 if(amp->am_refcnt > 0){
383 amp->am_flags |= a_AMV_MF_DELETE;
384 if(n_poption & n_PO_D_V)
385 n_err(_("Delayed deletion of currently active %s: %s\n"),
386 (amp->am_flags & a_AMV_MF_ACCOUNT ? "account" : "define"),
387 name);
388 }else{
389 a_amv_mac_free(amp);
390 amp = (struct a_amv_mac*)-1;
392 break;
396 if(newamp != NULL){
397 ampp = &a_amv_macs[h];
398 newamp->am_next = *ampp;
399 *ampp = newamp;
400 amp = NULL;
402 jleave:
403 NYD2_LEAVE;
404 return amp;
407 static int
408 a_amv_mac_call(void *v, bool_t silent_nexist){
409 struct a_amv_mac *amp;
410 int rv;
411 char const *name;
412 struct n_cmd_arg_ctx *cacp;
413 NYD_ENTER;
415 cacp = v;
417 if(cacp->cac_no == 0){
418 n_err(_("Synopsis: call(_if)?: name [:<arguments>:]\n"));
419 n_pstate_err_no = n_ERR_INVAL;
420 rv = 1;
421 goto jleave;
424 name = cacp->cac_arg->ca_arg.ca_str.s;
426 if(n_UNLIKELY(cacp->cac_no > a_AMV_POSPAR_MAX)){
427 n_err(_("Too many arguments to macro `call': %s\n"), name);
428 n_pstate_err_no = n_ERR_OVERFLOW;
429 rv = 1;
430 }else if(n_UNLIKELY((amp = a_amv_mac_lookup(name, NULL, a_AMV_MF_NONE)
431 ) == NULL)){
432 if(!silent_nexist)
433 n_err(_("Undefined macro called: %s\n"), n_shexp_quote_cp(name, FAL0));
434 n_pstate_err_no = n_ERR_NOENT;
435 rv = 1;
436 }else{
437 char const **argv;
438 struct a_amv_mac_call_args *amcap;
439 size_t argc;
441 argc = cacp->cac_no + 1;
442 amcap = n_lofi_alloc(sizeof *amcap + (argc * sizeof *argv));
443 argv = (void*)&amcap[1];
445 for(argc = 0; (cacp->cac_arg = cacp->cac_arg->ca_next) != NULL; ++argc)
446 argv[argc] = cacp->cac_arg->ca_arg.ca_str.s;
447 argv[argc] = NULL;
449 memset(amcap, 0, sizeof *amcap);
450 amcap->amca_name = name;
451 amcap->amca_amp = amp;
452 if(a_amv_lopts != NULL)
453 amcap->amca_loflags = (a_amv_lopts->as_loflags & a_AMV_LF_CALL_MASK
454 ) >> a_AMV_LF_CALL_TO_SCOPE_SHIFT;
455 if(argc > 0){
456 amcap->amca_pospar.app_count = (ui16_t)argc;
457 amcap->amca_pospar.app_not_heap = TRU1;
458 amcap->amca_pospar.app_dat = argv;
461 (void)a_amv_mac_exec(amcap);
462 rv = n_pstate_ex_no;
464 jleave:
465 NYD_LEAVE;
466 return rv;
469 static bool_t
470 a_amv_mac_exec(struct a_amv_mac_call_args *amcap){
471 struct a_amv_lostack *losp;
472 struct a_amv_mac_line **amlp;
473 char **args_base, **args;
474 struct a_amv_mac *amp;
475 bool_t rv;
476 NYD2_ENTER;
478 amp = amcap->amca_amp;
479 assert(amp != NULL && amp != a_AMV_MACKY_MACK);
480 ++amp->am_refcnt;
481 /* XXX Unfortunately we yet need to dup the macro lines! :( */
482 args_base = args = n_alloc(sizeof(*args) * (amp->am_line_cnt +1));
483 for(amlp = amp->am_line_dat; *amlp != NULL; ++amlp)
484 *(args++) = sbufdup((*amlp)->aml_dat, (*amlp)->aml_len);
485 *args = NULL;
487 losp = n_lofi_alloc(sizeof *losp);
488 losp->as_global_saved = a_amv_lopts;
489 if((losp->as_amcap = amcap)->amca_unroller == NULL){
490 losp->as_up = losp->as_global_saved;
491 losp->as_lopts = NULL;
492 }else{
493 losp->as_up = NULL;
494 losp->as_lopts = *amcap->amca_unroller;
496 losp->as_loflags = amcap->amca_loflags;
498 a_amv_lopts = losp;
499 if(amcap->amca_hook_pre != NULL)
500 n_PS_ROOT_BLOCK((*amcap->amca_hook_pre)(amcap->amca_hook_arg));
501 rv = n_go_macro((n_GO_INPUT_NONE |
502 (amcap->amca_no_xcall ? n_GO_INPUT_NO_XCALL : 0)),
503 amp->am_name, args_base, &a_amv_mac__finalize, losp);
504 NYD2_LEAVE;
505 return rv;
508 static void
509 a_amv_mac__finalize(void *vp){
510 struct a_amv_mac *amp;
511 struct a_amv_mac_call_args *amcap;
512 struct a_amv_lostack *losp;
513 NYD2_ENTER;
515 losp = vp;
516 a_amv_lopts = losp->as_global_saved;
518 amcap = losp->as_amcap;
520 if(!amcap->amca_pospar.app_not_heap && amcap->amca_pospar.app_maxcount > 0){
521 ui16_t i;
523 for(i = amcap->amca_pospar.app_maxcount; i-- != 0;)
524 n_free(n_UNCONST(amcap->amca_pospar.app_dat[i]));
525 n_free(amcap->amca_pospar.app_dat);
528 if(amcap->amca_unroller == NULL){
529 if(losp->as_lopts != NULL)
530 a_amv_lopts_unroll(&losp->as_lopts);
531 }else
532 *amcap->amca_unroller = losp->as_lopts;
534 if(amcap->amca_ps_hook_mask)
535 n_pstate &= ~n_PS_HOOK_MASK;
537 if((amp = amcap->amca_amp) != a_AMV_MACKY_MACK && amp != NULL &&
538 --amp->am_refcnt == 0 && (amp->am_flags & a_AMV_MF_DELETE))
539 a_amv_mac_free(amp);
541 n_lofi_free(losp);
542 n_lofi_free(amcap);
543 NYD2_LEAVE;
546 static bool_t
547 a_amv_mac_show(enum a_amv_mac_flags amf){
548 size_t lc, mc, ti, i;
549 char const *typestr;
550 FILE *fp;
551 bool_t rv;
552 NYD2_ENTER;
554 rv = FAL0;
556 if((fp = Ftmp(NULL, "deflist", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
557 NULL){
558 n_perr(_("Can't create temporary file for `define' or `account' listing"),
560 goto jleave;
563 amf &= a_AMV_MF_TYPE_MASK;
564 typestr = (amf & a_AMV_MF_ACCOUNT) ? "account" : "define";
566 for(lc = mc = ti = 0; ti < a_AMV_PRIME; ++ti){
567 struct a_amv_mac *amp;
569 for(amp = a_amv_macs[ti]; amp != NULL; amp = amp->am_next){
570 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
571 struct a_amv_mac_line **amlpp;
573 if(++mc > 1){
574 putc('\n', fp);
575 ++lc;
577 ++lc;
578 fprintf(fp, "%s %s {\n", typestr, amp->am_name);
579 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++lc, ++amlpp){
580 for(i = (*amlpp)->aml_prespc; i > 0; --i)
581 putc(' ', fp);
582 fputs((*amlpp)->aml_dat, fp);
583 putc('\n', fp);
585 fputs("}\n", fp);
586 ++lc;
590 if(mc > 0)
591 page_or_print(fp, lc);
593 rv = (ferror(fp) == 0);
594 Fclose(fp);
595 jleave:
596 NYD2_LEAVE;
597 return rv;
600 static bool_t
601 a_amv_mac_def(char const *name, enum a_amv_mac_flags amf){
602 struct str line;
603 ui32_t line_cnt, maxlen;
604 struct linelist{
605 struct linelist *ll_next;
606 struct a_amv_mac_line *ll_amlp;
607 } *llp, *ll_head, *ll_tail;
608 union {size_t s; int i; ui32_t ui; size_t l;} n;
609 struct a_amv_mac *amp;
610 bool_t rv;
611 NYD2_ENTER;
613 memset(&line, 0, sizeof line);
614 rv = FAL0;
615 amp = NULL;
617 /* TODO We should have our input state machine which emits Line events,
618 * TODO and hook different consumers dependent on our content, as stated
619 * TODO in i think lex_input; */
620 /* Read in the lines which form the macro content */
621 for(ll_tail = ll_head = NULL, line_cnt = maxlen = 0;;){
622 ui32_t leaspc;
623 char *cp;
625 n.i = n_go_input(n_GO_INPUT_CTX_DEFAULT | n_GO_INPUT_NL_ESC, n_empty,
626 &line.s, &line.l, NULL, NULL);
627 if(n.ui == 0)
628 continue;
629 if(n.i < 0){
630 n_err(_("Unterminated %s definition: %s\n"),
631 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
632 goto jerr;
635 /* Trim WS, remember amount of leading spaces for display purposes */
636 for(cp = line.s, leaspc = 0; n.ui > 0; ++cp, --n.ui)
637 if(*cp == '\t')
638 leaspc = (leaspc + 8u) & ~7u;
639 else if(*cp == ' ')
640 ++leaspc;
641 else
642 break;
643 for(; n.ui > 0 && spacechar(cp[n.ui - 1]); --n.ui)
645 if(n.ui == 0)
646 continue;
648 maxlen = n_MAX(maxlen, n.ui);
649 cp[n.ui++] = '\0';
651 /* Is is the closing brace? */
652 if(*cp == '}')
653 break;
655 if(n_LIKELY(++line_cnt < UI32_MAX)){
656 struct a_amv_mac_line *amlp;
658 llp = n_autorec_alloc(sizeof *llp);
659 if(ll_head == NULL)
660 ll_head = llp;
661 else
662 ll_tail->ll_next = llp;
663 ll_tail = llp;
664 llp->ll_next = NULL;
665 llp->ll_amlp = amlp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac_line,
666 aml_dat) + n.ui);
667 amlp->aml_len = n.ui -1;
668 amlp->aml_prespc = leaspc;
669 memcpy(amlp->aml_dat, cp, n.ui);
670 }else{
671 n_err(_("Too much content in %s definition: %s\n"),
672 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
673 goto jerr;
677 /* Create the new macro */
678 n.s = strlen(name) +1;
679 amp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name) + n.s);
680 memset(amp, 0, n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name));
681 amp->am_maxlen = maxlen;
682 amp->am_line_cnt = line_cnt;
683 amp->am_flags = amf;
684 memcpy(amp->am_name, name, n.s);
685 /* C99 */{
686 struct a_amv_mac_line **amlpp;
688 amp->am_line_dat = amlpp = n_alloc(sizeof(*amlpp) * ++line_cnt);
689 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
690 *amlpp++ = llp->ll_amlp;
691 *amlpp = NULL;
694 /* Create entry, replace a yet existing one as necessary */
695 a_amv_mac_lookup(name, amp, amf | a_AMV_MF_UNDEF);
696 rv = TRU1;
697 jleave:
698 if(line.s != NULL)
699 free(line.s);
700 NYD2_LEAVE;
701 return rv;
703 jerr:
704 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
705 free(llp->ll_amlp);
707 * if(amp != NULL){
708 * free(amp->am_line_dat);
709 * free(amp);
710 *}*/
711 goto jleave;
714 static bool_t
715 a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf){
716 struct a_amv_mac *amp;
717 bool_t rv;
718 NYD2_ENTER;
720 rv = TRU1;
722 if(n_LIKELY(name[0] != '*' || name[1] != '\0')){
723 if((amp = a_amv_mac_lookup(name, NULL, amf | a_AMV_MF_UNDEF)) == NULL){
724 n_err(_("%s not defined: %s\n"),
725 (amf & a_AMV_MF_ACCOUNT ? "Account" : "Macro"), name);
726 rv = FAL0;
728 }else{
729 struct a_amv_mac **ampp, *lamp;
731 for(ampp = a_amv_macs; PTRCMP(ampp, <, &a_amv_macs[n_NELEM(a_amv_macs)]);
732 ++ampp)
733 for(lamp = NULL, amp = *ampp; amp != NULL;){
734 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
735 /* xxx Expensive but rare: be simple */
736 a_amv_mac_lookup(amp->am_name, NULL, amf | a_AMV_MF_UNDEF);
737 amp = (lamp == NULL) ? *ampp : lamp->am_next;
738 }else{
739 lamp = amp;
740 amp = amp->am_next;
744 NYD2_LEAVE;
745 return rv;
748 static void
749 a_amv_mac_free(struct a_amv_mac *amp){
750 struct a_amv_mac_line **amlpp;
751 NYD2_ENTER;
753 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++amlpp)
754 free(*amlpp);
755 free(amp->am_line_dat);
756 free(amp);
757 NYD2_LEAVE;
760 static void
761 a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
762 struct a_amv_var *oavp){
763 struct a_amv_var *avp;
764 size_t nl, vl;
765 NYD2_ENTER;
767 /* Propagate unrolling up the stack, as necessary */
768 assert(alp != NULL);
769 for(;;){
770 if(alp->as_loflags & a_AMV_LF_SCOPE_MASK)
771 break;
772 if((alp = alp->as_up) == NULL)
773 goto jleave;
776 /* Check whether this variable is handled yet XXX Boost: ID check etc.!! */
777 for(avp = alp->as_lopts; avp != NULL; avp = avp->av_link)
778 if(!strcmp(avp->av_name, name))
779 goto jleave;
781 nl = strlen(name) +1;
782 vl = (oavp != NULL) ? strlen(oavp->av_value) +1 : 0;
783 avp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + nl + vl);
784 avp->av_link = alp->as_lopts;
785 alp->as_lopts = avp;
786 memcpy(avp->av_name, name, nl);
787 if(vl == 0){
788 avp->av_value = NULL;
789 avp->av_flags = 0;
790 #ifdef HAVE_PUTENV
791 avp->av_env = NULL;
792 #endif
793 }else{
794 avp->av_value = &avp->av_name[nl];
795 avp->av_flags = oavp->av_flags;
796 memcpy(avp->av_value, oavp->av_value, vl);
797 #ifdef HAVE_PUTENV
798 avp->av_env = (oavp->av_env == NULL) ? NULL : sstrdup(oavp->av_env);
799 #endif
801 jleave:
802 NYD2_LEAVE;
805 static void
806 a_amv_lopts_unroll(struct a_amv_var **avpp){
807 struct a_amv_lostack *save_alp;
808 struct a_amv_var *x, *avp;
809 NYD2_ENTER;
811 avp = *avpp;
812 *avpp = NULL;
814 save_alp = a_amv_lopts;
815 a_amv_lopts = NULL;
816 while(avp != NULL){
817 x = avp;
818 avp = avp->av_link;
819 n_PS_ROOT_BLOCK(n_var_vset(x->av_name, (uintptr_t)x->av_value));
820 free(x);
822 a_amv_lopts = save_alp;
823 NYD2_LEAVE;
826 static char *
827 a_amv_var_copy(char const *str){
828 char *news;
829 size_t len;
830 NYD2_ENTER;
832 if(*str == '\0')
833 news = n_UNCONST(n_empty);
834 else if(str[1] == '\0'){
835 if(str[0] == '1')
836 news = n_UNCONST(n_1);
837 else if(str[0] == '0')
838 news = n_UNCONST(n_0);
839 else
840 goto jheap;
841 }else if(str[2] == '\0' && str[0] == '-' && str[1] == '1')
842 news = n_UNCONST(n_m1);
843 else{
844 jheap:
845 len = strlen(str) +1;
846 news = n_alloc(len);
847 memcpy(news, str, len);
849 NYD2_LEAVE;
850 return news;
853 static void
854 a_amv_var_free(char *cp){
855 NYD2_ENTER;
856 if(cp[0] != '\0' && cp != n_0 && cp != n_1 && cp != n_m1)
857 free(cp);
858 NYD2_LEAVE;
861 static bool_t
862 a_amv_var_check_vips(enum a_amv_var_vip_mode avvm, enum okeys okey,
863 char const *val){
864 bool_t ok;
865 NYD2_ENTER;
867 ok = TRU1;
869 if(avvm == a_AMV_VIP_SET_PRE){
870 switch(okey){
871 default:
872 break;
873 case ok_v_HOME:
874 /* Note this gets called from main.c during initialization, and they
875 * simply set this to pw_dir as a fallback: don't verify _that_ call.
876 * See main.c! */
877 if(!(n_pstate & n_PS_ROOT) && !n_is_dir(val, TRUM1)){
878 n_err(_("$HOME is not a directory or not accessible: %s\n"),
879 n_shexp_quote_cp(val, FAL0));
880 ok = FAL0;
882 break;
883 case ok_v_TMPDIR:
884 if(!n_is_dir(val, TRU1)){
885 n_err(_("$TMPDIR 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_umask:
891 if(*val != '\0'){
892 ui64_t uib;
894 n_idec_ui64_cp(&uib, val, 0, NULL);
895 if(uib & ~0777u){ /* (is valid _VF_POSNUM) */
896 n_err(_("Invalid *umask* setting: %s\n"), val);
897 ok = FAL0;
900 break;
902 }else if(avvm == a_AMV_VIP_SET_POST){
903 switch(okey){
904 default:
905 break;
906 case ok_b_ask:
907 ok_bset(asksub);
908 break;
909 case ok_b_debug:
910 n_poption |= n_PO_DEBUG;
911 break;
912 case ok_v_HOME:
913 /* Invalidate any resolved folder then, too
914 * FALLTHRU */
915 case ok_v_folder:
916 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
917 break;
918 case ok_v_ifs:{
919 char *x_b, *x, c;
921 x_b = x = n_autorec_alloc(strlen(val) +1);
922 while((c = *val++) != '\0')
923 if(spacechar(c))
924 *x++ = c;
925 *x = '\0';
926 n_PS_ROOT_BLOCK(ok_vset(ifs_ws, x_b));
927 } break;
928 #ifdef HAVE_SETLOCALE
929 case ok_v_LANG:
930 case ok_v_LC_ALL:
931 case ok_v_LC_CTYPE:
932 n_locale_init();
933 break;
934 #endif
935 case ok_b_memdebug:
936 n_poption |= n_PO_MEMDEBUG;
937 break;
938 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
939 if(!(n_pstate & n_PS_ROOT))
940 n_PS_ROOT_BLOCK(ok_bset(posix));
941 break;
942 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
943 if(!(n_pstate & n_PS_ROOT))
944 n_PS_ROOT_BLOCK(ok_bset(POSIXLY_CORRECT));
945 break;
946 case ok_b_skipemptybody:
947 n_poption |= n_PO_E_FLAG;
948 break;
949 case ok_b_typescript_mode:
950 ok_bset(colour_disable);
951 ok_bset(line_editor_disable);
952 if(!(n_psonce & n_PSO_STARTED))
953 ok_bset(termcap_disable);
954 break;
955 case ok_v_umask:
956 if(*val != '\0'){
957 ui64_t uib;
959 n_idec_ui64_cp(&uib, val, 0, NULL);
960 umask((mode_t)uib);
962 break;
963 case ok_b_verbose:
964 n_poption |= (n_poption & n_PO_VERB) ? n_PO_VERBVERB : n_PO_VERB;
965 break;
967 }else{
968 switch(okey){
969 default:
970 break;
971 case ok_b_ask:
972 ok_bclear(asksub);
973 break;
974 case ok_b_debug:
975 n_poption &= ~n_PO_DEBUG;
976 break;
977 case ok_v_HOME:
978 /* Invalidate any resolved folder then, too
979 * FALLTHRU */
980 case ok_v_folder:
981 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
982 break;
983 case ok_b_memdebug:
984 n_poption &= ~n_PO_MEMDEBUG;
985 break;
986 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
987 if(!(n_pstate & n_PS_ROOT))
988 n_PS_ROOT_BLOCK(ok_bclear(posix));
989 break;
990 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
991 if(!(n_pstate & n_PS_ROOT))
992 n_PS_ROOT_BLOCK(ok_bclear(POSIXLY_CORRECT));
993 break;
994 case ok_b_skipemptybody:
995 n_poption &= ~n_PO_E_FLAG;
996 break;
997 case ok_b_verbose:
998 n_poption &= ~(n_PO_VERB | n_PO_VERBVERB);
999 break;
1002 NYD2_LEAVE;
1003 return ok;
1006 static bool_t
1007 a_amv_var_check_nocntrls(char const *val){
1008 char c;
1009 NYD2_ENTER;
1011 while((c = *val++) != '\0')
1012 if(cntrlchar(c))
1013 break;
1014 NYD2_LEAVE;
1015 return (c == '\0');
1018 static bool_t
1019 a_amv_var_check_num(char const *val, bool_t posnum){
1020 /* TODO The internal/environment variables which are num= or posnum= should
1021 * TODO gain special lookup functions, or the return should be void* and
1022 * TODO castable to integer; i.e. no more strtoX() should be needed.
1023 * TODO I.e., the result of this function should instead be stored */
1024 bool_t rv;
1025 NYD2_ENTER;
1027 rv = TRU1;
1029 if(*val != '\0'){ /* Would be _VF_NOTEMPTY if not allowed */
1030 ui64_t uib;
1031 enum n_idec_state ids;
1033 ids = n_idec_cp(&uib, val, 0,
1034 (posnum ? n_IDEC_MODE_SIGNED_TYPE : n_IDEC_MODE_NONE), NULL);
1035 if((ids & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
1036 ) != n_IDEC_STATE_CONSUMED)
1037 rv = FAL0;
1038 /* TODO Unless we store integers we need to look and forbid, because
1039 * TODO callee may not be able to swallow, e.g., "-1" */
1040 if(posnum && (ids & n_IDEC_STATE_SEEN_MINUS))
1041 rv = FAL0;
1043 NYD2_LEAVE;
1044 return rv;
1047 static char const *
1048 a_amv_var_canonify(char const *vn){
1049 NYD2_ENTER;
1050 if(!upperchar(*vn)){
1051 char const *vp;
1053 for(vp = vn; *vp != '\0' && *vp != '@'; ++vp)
1055 vn = (*vp == '@') ? i_strdup(vn) : vn;
1057 NYD2_LEAVE;
1058 return vn;
1061 static bool_t
1062 a_amv_var_revlookup(struct a_amv_var_carrier *avcp, char const *name){
1063 ui32_t hash, i, j;
1064 struct a_amv_var_map const *avmp;
1065 char c;
1066 NYD2_ENTER;
1068 /* It may be a special a.k.a. macro-local or one-letter parameter */
1069 c = name[0];
1070 if(n_UNLIKELY(digitchar(c))){
1071 /* (Inline dec. atoi, ugh) */
1072 for(j = (ui8_t)c - '0', i = 1;; ++i){
1073 c = name[i];
1074 if(c == '\0')
1075 break;
1076 if(!digitchar(c))
1077 goto jno_special_param;
1078 j = j * 10 + (ui8_t)c - '0';
1080 if(j <= a_AMV_POSPAR_MAX){
1081 avcp->avc_special_cat = a_AMV_VSC_POSPAR;
1082 goto jspecial_param;
1084 }else if(n_UNLIKELY(name[1] == '\0')){
1085 switch(c){
1086 case '?':
1087 case '!':
1088 avcp->avc_special_cat = a_AMV_VSC_GLOBAL;
1089 j = (c == '?') ? a_AMV_VST_QM : a_AMV_VST_EM;
1090 goto jspecial_param;
1091 case '^':
1092 goto jmultiplex;
1093 case '*':
1094 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1095 j = a_AMV_VST_STAR;
1096 goto jspecial_param;
1097 case '@':
1098 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1099 j = a_AMV_VST_AT;
1100 goto jspecial_param;
1101 case '#':
1102 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1103 j = a_AMV_VST_NOSIGN;
1104 goto jspecial_param;
1105 default:
1106 break;
1108 }else if(c == '^'){
1109 jmultiplex:
1110 avcp->avc_special_cat = a_AMV_VSC_MULTIPLEX;
1111 j = a_AMV_VST_CACC;
1112 goto jspecial_param;
1115 /* Normal reverse lookup, walk over the hashtable */
1116 jno_special_param:
1117 avcp->avc_special_cat = a_AMV_VSC_NONE;
1118 avcp->avc_name = name = a_amv_var_canonify(name);
1119 avcp->avc_hash = hash = a_AMV_NAME2HASH(name);
1121 for(i = hash % a_AMV_VAR_REV_PRIME, j = 0; j <= a_AMV_VAR_REV_LONGEST; ++j){
1122 ui32_t x;
1124 if((x = a_amv_var_revmap[i]) == a_AMV_VAR_REV_ILL)
1125 break;
1127 avmp = &a_amv_var_map[x];
1128 if(avmp->avm_hash == hash &&
1129 !strcmp(&a_amv_var_names[avmp->avm_keyoff], name)){
1130 avcp->avc_map = avmp;
1131 avcp->avc_okey = (enum okeys)x;
1132 goto jleave;
1135 if(++i == a_AMV_VAR_REV_PRIME){
1136 #ifdef a_AMV_VAR_REV_WRAPAROUND
1137 i = 0;
1138 #else
1139 break;
1140 #endif
1143 avcp->avc_map = NULL;
1144 avcp = NULL;
1145 jleave:
1146 NYD2_LEAVE;
1147 return (avcp != NULL);
1149 /* All these are mapped to *--special-param* */
1150 jspecial_param:
1151 avcp->avc_name = name;
1152 avcp->avc_special_prop = (ui16_t)j;
1153 avmp = &a_amv_var_map[a_AMV_VAR__SPECIAL_PARAM_MAP_IDX];
1154 avcp->avc_hash = avmp->avm_hash;
1155 avcp->avc_map = avmp;
1156 avcp->avc_okey = ok_v___special_param;
1157 goto jleave;
1160 static bool_t
1161 a_amv_var_lookup(struct a_amv_var_carrier *avcp, bool_t i3val_nonew){
1162 size_t i;
1163 char const *cp;
1164 struct a_amv_var_map const *avmp;
1165 struct a_amv_var *avp;
1166 NYD2_ENTER;
1168 /* C99 */{
1169 struct a_amv_var **avpp, *lavp;
1171 avpp = &a_amv_vars[avcp->avc_prime = a_AMV_HASH2PRIME(avcp->avc_hash)];
1173 for(lavp = NULL, avp = *avpp; avp != NULL; lavp = avp, avp = avp->av_link)
1174 if(!strcmp(avp->av_name, avcp->avc_name)){
1175 /* Relink as head, hope it "sorts on usage" over time.
1176 * _clear() relies on this behaviour */
1177 if(lavp != NULL){
1178 lavp->av_link = avp->av_link;
1179 avp->av_link = *avpp;
1180 *avpp = avp;
1182 goto jleave;
1186 /* If this is not an assembled variable we need to consider some special
1187 * initialization cases and eventually create the variable anew */
1188 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1189 /* Does it have an import-from-environment flag? */
1190 if(n_UNLIKELY((avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV)) != 0)){
1191 if(n_LIKELY((cp = getenv(avcp->avc_name)) != NULL)){
1192 /* May be better not to use that one, though? */
1193 /* TODO Outsource the tests into a _shared_ test function! */
1194 bool_t isempty, isbltin;
1196 isempty = (*cp == '\0' &&
1197 (avmp->avm_flags & a_AMV_VF_NOTEMPTY) != 0);
1198 isbltin = ((avmp->avm_flags & (a_AMV_VF_I3VAL | a_AMV_VF_DEFVAL)
1199 ) != 0);
1201 if(n_UNLIKELY(isempty)){
1202 n_err(_("Environment variable must not be empty: %s\n"),
1203 avcp->avc_name);
1204 if(!isbltin)
1205 goto jerr;
1206 }else if(n_LIKELY(*cp != '\0')){
1207 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOCNTRLS) &&
1208 !a_amv_var_check_nocntrls(cp))){
1209 n_err(_("Ignoring environment, control characters "
1210 "invalid in variable: %s\n"), avcp->avc_name);
1211 goto jerr;
1213 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1214 !a_amv_var_check_num(cp, FAL0))){
1215 n_err(_("Environment variable value not a number "
1216 "or out of range: %s\n"), avcp->avc_name);
1217 goto jerr;
1219 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1220 !a_amv_var_check_num(cp, TRU1))){
1221 n_err(_("Environment variable value not a number, "
1222 "negative or out of range: %s\n"), avcp->avc_name);
1223 goto jerr;
1225 goto jnewval;
1226 }else
1227 goto jnewval;
1231 /* A first-time init switch is to be handled now and here */
1232 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_I3VAL) != 0)){
1233 static struct a_amv_var_defval const **arr,
1234 *arr_base[a_AMV_VAR_I3VALS_CNT +1];
1236 if(arr == NULL){
1237 arr = &arr_base[0];
1238 arr[i = a_AMV_VAR_I3VALS_CNT] = NULL;
1239 while(i-- > 0)
1240 arr[i] = &a_amv_var_i3vals[i];
1243 for(i = 0; arr[i] != NULL; ++i)
1244 if(arr[i]->avdv_okey == avcp->avc_okey){
1245 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_empty
1246 : arr[i]->avdv_value;
1247 /* Remove this entry, hope entire block becomes no-op asap */
1249 arr[i] = arr[i + 1];
1250 while(arr[i++] != NULL);
1252 if(!i3val_nonew)
1253 goto jnewval;
1254 if(i3val_nonew == TRUM1)
1255 avp = (struct a_amv_var*)-1;
1256 goto jleave;
1260 /* The virtual variables */
1261 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIRT) != 0)){
1262 for(i = 0; i < a_AMV_VAR_VIRTS_CNT; ++i)
1263 if(a_amv_var_virts[i].avv_okey == avcp->avc_okey){
1264 avp = n_UNCONST(a_amv_var_virts[i].avv_var);
1265 goto jleave;
1267 /* Not reached */
1270 /* Place this last because once it is set first the variable will never
1271 * be removed again and thus match in the first block above */
1272 jdefval:
1273 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0){
1274 for(i = 0; i < a_AMV_VAR_DEFVALS_CNT; ++i)
1275 if(a_amv_var_defvals[i].avdv_okey == avcp->avc_okey){
1276 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_empty
1277 : a_amv_var_defvals[i].avdv_value;
1278 goto jnewval;
1283 jerr:
1284 avp = NULL;
1285 jleave:
1286 avcp->avc_var = avp;
1287 NYD2_LEAVE;
1288 return (avp != NULL);
1290 jnewval:
1291 /* E.g., $TMPDIR may be set to non-existent, so we need to be able to catch
1292 * that and redirect to a possible default value */
1293 if((avmp->avm_flags & a_AMV_VF_VIP) &&
1294 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, cp)){
1295 #ifdef HAVE_SETENV
1296 if(avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV))
1297 unsetenv(avcp->avc_name);
1298 #endif
1299 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)
1300 goto jdefval;
1301 goto jerr;
1302 }else{
1303 struct a_amv_var **avpp;
1304 size_t l;
1306 l = strlen(avcp->avc_name) +1;
1307 avcp->avc_var =
1308 avp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1309 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1310 *avpp = avp;
1311 memcpy(avp->av_name, avcp->avc_name, l);
1312 #ifdef HAVE_PUTENV
1313 avp->av_env = NULL;
1314 #endif
1315 avp->av_flags = avmp->avm_flags;
1316 avp->av_value = a_amv_var_copy(cp);
1318 if(avp->av_flags & a_AMV_VF_ENV)
1319 a_amv_var__putenv(avcp, avp);
1320 if(avmp->avm_flags & a_AMV_VF_VIP)
1321 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, cp);
1322 goto jleave;
1326 static char const *
1327 a_amv_var_vsc_global(struct a_amv_var_carrier *avcp){
1328 char itoabuf[32];
1329 char const *rv;
1330 si32_t *ep;
1331 struct a_amv_var_map const *avmp;
1332 NYD2_ENTER;
1334 /* Not function local, TODO but lazy evaluted for now */
1335 if(avcp->avc_special_prop == a_AMV_VST_QM){
1336 avmp = &a_amv_var_map[a_AMV_VAR__QM_MAP_IDX];
1337 avcp->avc_okey = ok_v___qm;
1338 ep = &n_pstate_ex_no;
1339 }else{
1340 avmp = &a_amv_var_map[a_AMV_VAR__EM_MAP_IDX];
1341 avcp->avc_okey = ok_v___em;
1342 ep = &n_pstate_err_no;
1345 /* XXX Oh heaven, we are responsible to ensure that $?/! is up-to-date
1346 * TODO we could num=1 ok_v___[qe]m, but the thing is still a string
1347 * TODO and thus conversion takes places over and over again; also
1348 * TODO for now that would have to occur before we set _that_ value
1349 * TODO so let's special treat it until we store ints as such */
1350 switch(*ep){
1351 case 0: rv = n_0; break;
1352 case 1: rv = n_1; break;
1353 default:
1354 snprintf(itoabuf, sizeof itoabuf, "%d", *ep);
1355 rv = itoabuf;
1356 break;
1358 n_PS_ROOT_BLOCK(n_var_okset(avcp->avc_okey, (uintptr_t)rv));
1360 avcp->avc_hash = avmp->avm_hash;
1361 avcp->avc_map = avmp;
1362 rv = a_amv_var_lookup(avcp, FAL0) ? avcp->avc_var->av_value : NULL;
1363 NYD2_LEAVE;
1364 return rv;
1367 static char const *
1368 a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp){
1369 char itoabuf[32];
1370 si32_t e;
1371 size_t i;
1372 char const *rv;
1373 NYD2_ENTER;
1375 i = strlen(rv = &avcp->avc_name[1]);
1377 /* ERR, ERRDOC, ERRNAME, plus *-NAME variants */
1378 if(rv[0] == 'E' && i >= 3 && rv[1] == 'R' && rv[2] == 'R'){
1379 if(i == 3){
1380 e = n_pstate_err_no;
1381 goto jeno;
1382 }else if(rv[3] == '-'){
1383 e = n_err_from_name(&rv[4]);
1384 jeno:
1385 switch(e){
1386 case 0: rv = n_0; break;
1387 case 1: rv = n_1; break;
1388 default:
1389 snprintf(itoabuf, sizeof itoabuf, "%d", e);
1390 rv = savestr(itoabuf); /* XXX yet, cannot do numbers */
1391 break;
1393 goto jleave;
1394 }else if(i >= 6){
1395 if(!memcmp(&rv[3], "DOC", 3)){
1396 rv += 6;
1397 switch(*rv){
1398 case '\0': e = n_pstate_err_no; break;
1399 case '-': e = n_err_from_name(&rv[1]); break;
1400 default: goto jerr;
1402 rv = n_err_to_doc(e);
1403 goto jleave;
1404 }else if(i >= 7 && !memcmp(&rv[3], "NAME", 4)){
1405 rv += 7;
1406 switch(*rv){
1407 case '\0': e = n_pstate_err_no; break;
1408 case '-': e = n_err_from_name(&rv[1]); break;
1409 default: goto jerr;
1411 rv = n_err_to_name(e);
1412 goto jleave;
1417 jerr:
1418 rv = NULL;
1419 jleave:
1420 NYD2_LEAVE;
1421 return rv;
1424 static char const *
1425 a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp){
1426 size_t i, j;
1427 ui16_t argc;
1428 char const *rv, **argv;
1429 NYD2_ENTER;
1431 rv = NULL;
1433 /* If in a macro/xy.. */
1434 if(a_amv_lopts != NULL){
1435 bool_t ismacky;
1436 struct a_amv_mac_call_args *amcap;
1438 amcap = a_amv_lopts->as_amcap;
1439 argc = amcap->amca_pospar.app_count;
1440 argv = amcap->amca_pospar.app_dat;
1441 argv += amcap->amca_pospar.app_idx;
1443 /* ..in a `call'ed macro only, to be exact. Or in a_AMV_MACKY_MACK */
1444 if(!(ismacky = (amcap->amca_amp == a_AMV_MACKY_MACK)) &&
1445 (amcap->amca_ps_hook_mask ||
1446 (assert(amcap->amca_amp != NULL),
1447 (amcap->amca_amp->am_flags & a_AMV_MF_TYPE_MASK
1448 ) == a_AMV_MF_ACCOUNT)))
1449 goto jleave;
1451 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1452 if(avcp->avc_special_prop > 0){
1453 if(argc >= avcp->avc_special_prop)
1454 rv = argv[avcp->avc_special_prop - 1];
1455 }else if(ismacky)
1456 rv = amcap->amca_name;
1457 else
1458 rv = (a_amv_lopts->as_up != NULL
1459 ? a_amv_lopts->as_up->as_amcap->amca_name : n_empty);
1460 goto jleave;
1462 /* MACKY_MACK doesn't know about [*@#] */
1463 /*else*/ if(ismacky){
1464 if(n_poption & n_PO_D_V)
1465 n_err(_("Cannot use $*/$@/$# in this context: %s\n"),
1466 n_shexp_quote_cp(avcp->avc_name, FAL0));
1467 goto jleave;
1469 }else{
1470 argc = a_amv_pospar.app_count;
1471 argv = a_amv_pospar.app_dat;
1472 argv += a_amv_pospar.app_idx;
1474 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1475 if(avcp->avc_special_prop > 0){
1476 if(argc >= avcp->avc_special_prop)
1477 rv = argv[avcp->avc_special_prop - 1];
1478 }else
1479 rv = n_progname;
1480 goto jleave;
1484 switch(avcp->avc_special_prop){ /* XXX OPTIMIZE */
1485 case a_AMV_VST_STAR:
1486 case a_AMV_VST_AT:
1487 for(i = j = 0; i < argc; ++i)
1488 j += strlen(argv[i]) + 1;
1489 if(j == 0)
1490 rv = n_empty;
1491 else{
1492 char *cp;
1494 rv = cp = n_autorec_alloc(j);
1495 for(i = j = 0; i < argc; ++i){
1496 j = strlen(argv[i]);
1497 memcpy(cp, argv[i], j);
1498 cp += j;
1499 *cp++ = ' ';
1501 *--cp = '\0';
1503 break;
1504 case a_AMV_VST_NOSIGN:{
1505 char *cp;
1507 rv = cp = n_autorec_alloc(sizeof("65535"));
1508 snprintf(cp, sizeof("65535"), "%hu", argc);
1509 } break;
1510 default:
1511 rv = n_empty;
1512 break;
1514 jleave:
1515 NYD2_LEAVE;
1516 return rv;
1519 static bool_t
1520 a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
1521 bool_t force_env){
1522 struct a_amv_var *avp;
1523 char *oval;
1524 struct a_amv_var_map const *avmp;
1525 bool_t rv;
1526 NYD2_ENTER;
1528 if(value == NULL){
1529 rv = a_amv_var_clear(avcp, force_env);
1530 goto jleave;
1533 if((avmp = avcp->avc_map) != NULL){
1534 rv = FAL0;
1536 /* Validity checks */
1537 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_RDONLY) != 0 &&
1538 !(n_pstate & n_PS_ROOT))){
1539 value = N_("Variable is read-only: %s\n");
1540 goto jeavmp;
1542 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOTEMPTY) && *value == '\0')){
1543 value = N_("Variable must not be empty: %s\n");
1544 goto jeavmp;
1546 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOCNTRLS) != 0 &&
1547 !a_amv_var_check_nocntrls(value))){
1548 value = N_("Variable forbids control characters: %s\n");
1549 goto jeavmp;
1551 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1552 !a_amv_var_check_num(value, FAL0))){
1553 value = N_("Variable value not a number or out of range: %s\n");
1554 goto jeavmp;
1556 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1557 !a_amv_var_check_num(value, TRU1))){
1558 value = _("Variable value not a number, negative, "
1559 "or out of range: %s\n");
1560 goto jeavmp;
1562 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_IMPORT) != 0 &&
1563 !(n_psonce & n_PSO_STARTED) && !(n_pstate & n_PS_ROOT))){
1564 value = N_("Variable cannot be set in a resource file: %s\n");
1565 goto jeavmp;
1568 /* Any more complicated inter-dependency? */
1569 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1570 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, value))){
1571 value = N_("Assignment of variable aborted: %s\n");
1572 jeavmp:
1573 n_err(V_(value), avcp->avc_name);
1574 goto jleave;
1577 /* Transformations */
1578 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_LOWER)){
1579 char c;
1581 oval = savestr(value);
1582 value = oval;
1583 for(; (c = *oval) != '\0'; ++oval)
1584 *oval = lowerconv(c);
1588 rv = TRU1;
1589 a_amv_var_lookup(avcp, TRU1);
1591 /* Don't care what happens later on, store this in the unroll list */
1592 if(a_amv_lopts != NULL &&
1593 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1594 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1596 if((avp = avcp->avc_var) == NULL){
1597 struct a_amv_var **avpp;
1598 size_t l;
1600 l = strlen(avcp->avc_name) +1;
1601 avcp->avc_var = avp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_var, av_name
1602 ) + l);
1603 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1604 *avpp = avp;
1605 #ifdef HAVE_PUTENV
1606 avp->av_env = NULL;
1607 #endif
1608 memcpy(avp->av_name, avcp->avc_name, l);
1609 avp->av_flags = (avmp != NULL) ? avmp->avm_flags : 0;
1610 oval = n_UNCONST(n_empty);
1611 }else
1612 oval = avp->av_value;
1614 if(avmp == NULL)
1615 avp->av_value = a_amv_var_copy(value);
1616 else{
1617 /* Via `set' etc. the user may give even boolean options non-boolean
1618 * values, ignore that and force boolean */
1619 if(avp->av_flags & a_AMV_VF_BOOL){
1620 if(!(n_pstate & n_PS_ROOT) && (n_poption & n_PO_D_VV) &&
1621 *value != '\0')
1622 n_err(_("Ignoring value of boolean variable: %s: %s\n"),
1623 avcp->avc_name, value);
1624 avp->av_value = n_UNCONST(n_1);
1625 }else
1626 avp->av_value = a_amv_var_copy(value);
1629 if(force_env && !(avp->av_flags & a_AMV_VF_ENV))
1630 avp->av_flags |= a_AMV_VF_LINKED;
1631 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_LINKED))
1632 rv = a_amv_var__putenv(avcp, avp);
1633 if(avp->av_flags & a_AMV_VF_VIP)
1634 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, value);
1635 a_amv_var_free(oval);
1636 jleave:
1637 NYD2_LEAVE;
1638 return rv;
1641 static bool_t
1642 a_amv_var__putenv(struct a_amv_var_carrier *avcp, struct a_amv_var *avp){
1643 #ifndef HAVE_SETENV
1644 char *cp;
1645 #endif
1646 bool_t rv;
1647 NYD2_ENTER;
1649 #ifdef HAVE_SETENV
1650 rv = (setenv(avcp->avc_name, avp->av_value, 1) == 0);
1651 #else
1652 cp = sstrdup(savecatsep(avcp->avc_name, '=', avp->av_value));
1654 if((rv = (putenv(cp) == 0))){
1655 char *ocp;
1657 if((ocp = avp->av_env) != NULL)
1658 free(ocp);
1659 avp->av_env = cp;
1660 }else
1661 free(cp);
1662 #endif
1663 NYD2_LEAVE;
1664 return rv;
1667 static bool_t
1668 a_amv_var_clear(struct a_amv_var_carrier *avcp, bool_t force_env){
1669 struct a_amv_var **avpp, *avp;
1670 struct a_amv_var_map const *avmp;
1671 bool_t rv;
1672 NYD2_ENTER;
1674 rv = FAL0;
1676 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1677 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NODEL) != 0 &&
1678 !(n_pstate & n_PS_ROOT))){
1679 n_err(_("Variable may not be unset: %s\n"), avcp->avc_name);
1680 goto jleave;
1682 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1683 !a_amv_var_check_vips(a_AMV_VIP_CLEAR, avcp->avc_okey, NULL))){
1684 n_err(_("Clearance of variable aborted: %s\n"), avcp->avc_name);
1685 goto jleave;
1689 rv = TRU1;
1691 if(n_UNLIKELY(!a_amv_var_lookup(avcp, TRUM1))){
1692 if(force_env){
1693 jforce_env:
1694 rv = a_amv_var__clearenv(avcp->avc_name, NULL);
1695 }else if(!(n_pstate & (n_PS_ROOT | n_PS_ROBOT)) && (n_poption & n_PO_D_V))
1696 n_err(_("Can't unset undefined variable: %s\n"), avcp->avc_name);
1697 goto jleave;
1698 }else if(avcp->avc_var == (struct a_amv_var*)-1){
1699 avcp->avc_var = NULL;
1700 if(force_env)
1701 goto jforce_env;
1702 goto jleave;
1705 if(a_amv_lopts != NULL &&
1706 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1707 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1709 avp = avcp->avc_var;
1710 avcp->avc_var = NULL;
1711 avpp = &a_amv_vars[avcp->avc_prime];
1712 assert(*avpp == avp); /* (always listhead after lookup()) */
1713 *avpp = (*avpp)->av_link;
1715 /* C99 */{
1716 char *envval;
1718 #ifdef HAVE_SETENV
1719 envval = NULL;
1720 #else
1721 envval = avp->av_env;
1722 #endif
1723 if((avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_LINKED)) || envval != NULL)
1724 rv = a_amv_var__clearenv(avp->av_name, envval);
1726 a_amv_var_free(avp->av_value);
1727 free(avp);
1729 /* XXX Fun part, extremely simple-minded for now: if this variable has
1730 * XXX a default value, immediately reinstantiate it! TODO Heh? */
1731 if(n_UNLIKELY(avmp != NULL && (avmp->avm_flags & a_AMV_VF_DEFVAL) != 0))
1732 a_amv_var_lookup(avcp, TRU1);
1733 jleave:
1734 NYD2_LEAVE;
1735 return rv;
1738 static bool_t
1739 a_amv_var__clearenv(char const *name, char *value){
1740 #ifndef HAVE_SETENV
1741 extern char **environ;
1742 char **ecpp;
1743 #endif
1744 bool_t rv;
1745 NYD2_ENTER;
1746 n_UNUSED(value);
1748 #ifdef HAVE_SETENV
1749 unsetenv(name);
1750 rv = TRU1;
1751 #else
1752 if(value != NULL)
1753 for(ecpp = environ; *ecpp != NULL; ++ecpp)
1754 if(*ecpp == value){
1755 free(value);
1757 ecpp[0] = ecpp[1];
1758 while(*ecpp++ != NULL);
1759 break;
1761 rv = TRU1;
1762 #endif
1763 NYD2_LEAVE;
1764 return rv;
1767 static void
1768 a_amv_var_show_all(void){
1769 struct n_string msg, *msgp;
1770 FILE *fp;
1771 size_t no, i;
1772 struct a_amv_var *avp;
1773 char const **vacp, **cap;
1774 NYD2_ENTER;
1776 if((fp = Ftmp(NULL, "setlist", OF_RDWR | OF_UNLINK | OF_REGISTER)) == NULL){
1777 n_perr(_("Can't create temporary file for `set' listing"), 0);
1778 goto jleave;
1781 /* We need to instantiate first-time-inits and default values here, so that
1782 * they will be regular members of our _vars[] table */
1783 for(i = a_AMV_VAR_I3VALS_CNT; i-- > 0;)
1784 n_var_oklook(a_amv_var_i3vals[i].avdv_okey);
1785 for(i = a_AMV_VAR_DEFVALS_CNT; i-- > 0;)
1786 n_var_oklook(a_amv_var_defvals[i].avdv_okey);
1788 for(no = i = 0; i < a_AMV_PRIME; ++i)
1789 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
1790 ++no;
1791 no += a_AMV_VAR_VIRTS_CNT;
1793 vacp = n_autorec_alloc(no * sizeof(*vacp));
1795 for(cap = vacp, i = 0; i < a_AMV_PRIME; ++i)
1796 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
1797 *cap++ = avp->av_name;
1798 for(i = a_AMV_VAR_VIRTS_CNT; i-- > 0;)
1799 *cap++ = a_amv_var_virts[i].avv_var->av_name;
1801 if(no > 1)
1802 qsort(vacp, no, sizeof *vacp, &a_amv_var__show_cmp);
1804 msgp = &msg;
1805 msgp = n_string_reserve(n_string_creat(msgp), 80);
1806 for(i = 0, cap = vacp; no != 0; ++cap, --no)
1807 i += a_amv_var_show(*cap, fp, msgp);
1808 n_string_gut(&msg);
1810 page_or_print(fp, i);
1811 Fclose(fp);
1812 jleave:
1813 NYD2_LEAVE;
1816 static int
1817 a_amv_var__show_cmp(void const *s1, void const *s2){
1818 int rv;
1819 NYD2_ENTER;
1821 rv = strcmp(*(char**)n_UNCONST(s1), *(char**)n_UNCONST(s2));
1822 NYD2_LEAVE;
1823 return rv;
1826 static size_t
1827 a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp){
1828 struct a_amv_var_carrier avc;
1829 char const *quote;
1830 bool_t isset;
1831 size_t i;
1832 NYD2_ENTER;
1834 msgp = n_string_trunc(msgp, 0);
1835 i = 0;
1837 a_amv_var_revlookup(&avc, name);
1838 isset = a_amv_var_lookup(&avc, FAL0);
1840 if(n_poption & n_PO_D_V){
1841 if(avc.avc_map == NULL){
1842 msgp = n_string_push_cp(msgp, "#assembled variable with value");
1843 i = 1;
1844 }else{
1845 struct{
1846 ui16_t flag;
1847 char msg[22];
1848 } const tbase[] = {
1849 {a_AMV_VF_VIRT, "virtual"},
1850 {a_AMV_VF_RDONLY, "read-only"},
1851 {a_AMV_VF_NODEL, "nodelete"},
1852 {a_AMV_VF_NOTEMPTY, "notempty"},
1853 {a_AMV_VF_NOCNTRLS, "no-control-chars"},
1854 {a_AMV_VF_NUM, "number"},
1855 {a_AMV_VF_POSNUM, "positive-number"},
1856 {a_AMV_VF_IMPORT, "import-environ-first\0"},
1857 {a_AMV_VF_ENV, "sync-environ"},
1858 {a_AMV_VF_I3VAL, "initial-value"},
1859 {a_AMV_VF_DEFVAL, "default-value"},
1860 {a_AMV_VF_LINKED, "`environ' link"}
1861 }, *tp;
1863 for(tp = tbase; PTRCMP(tp, <, &tbase[n_NELEM(tbase)]); ++tp)
1864 if(isset ? (avc.avc_var->av_flags & tp->flag)
1865 : (avc.avc_map->avm_flags & tp->flag)){
1866 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
1867 msgp = n_string_push_cp(msgp, tp->msg);
1870 if(i > 0)
1871 msgp = n_string_push_cp(msgp, "\n ");
1874 if(!isset || (avc.avc_var->av_flags & a_AMV_VF_RDONLY)){
1875 msgp = n_string_push_c(msgp, *n_ns);
1876 if(!isset){
1877 if(avc.avc_map != NULL && (avc.avc_map->avm_flags & a_AMV_VF_BOOL))
1878 msgp = n_string_push_cp(msgp, "boolean; ");
1879 msgp = n_string_push_cp(msgp, "variable not set: ");
1880 msgp = n_string_push_cp(msgp, n_shexp_quote_cp(name, FAL0));
1881 goto jleave;
1885 n_UNINIT(quote, NULL);
1886 if(!(avc.avc_var->av_flags & a_AMV_VF_BOOL)){
1887 quote = n_shexp_quote_cp(avc.avc_var->av_value, TRU1);
1888 if(strcmp(quote, avc.avc_var->av_value))
1889 msgp = n_string_push_cp(msgp, "wysh ");
1890 }else if(n_poption & n_PO_D_V)
1891 msgp = n_string_push_cp(msgp, "wysh ");
1892 if(avc.avc_var->av_flags & a_AMV_VF_LINKED)
1893 msgp = n_string_push_cp(msgp, "environ ");
1894 msgp = n_string_push_cp(msgp, "set ");
1895 msgp = n_string_push_cp(msgp, name);
1896 if(!(avc.avc_var->av_flags & a_AMV_VF_BOOL)){
1897 msgp = n_string_push_c(msgp, '=');
1898 msgp = n_string_push_cp(msgp, quote);
1899 }else if(n_poption & n_PO_D_V)
1900 msgp = n_string_push_cp(msgp, " #boolean");
1902 jleave:
1903 msgp = n_string_push_c(msgp, '\n');
1904 fputs(n_string_cp(msgp), fp);
1905 NYD2_ENTER;
1906 return (i > 0 ? 2 : 1);
1909 static bool_t
1910 a_amv_var_c_set(char **ap, bool_t issetenv){
1911 char *cp, *cp2, *varbuf, c;
1912 size_t errs;
1913 NYD2_ENTER;
1915 errs = 0;
1916 jouter:
1917 while((cp = *ap++) != NULL){
1918 /* Isolate key */
1919 cp2 = varbuf = n_autorec_alloc(strlen(cp) +1);
1921 for(; (c = *cp) != '=' && c != '\0'; ++cp){
1922 if(cntrlchar(c) || spacechar(c)){
1923 n_err(_("Variable name with control character ignored: %s\n"),
1924 ap[-1]);
1925 ++errs;
1926 goto jouter;
1928 *cp2++ = c;
1930 *cp2 = '\0';
1931 if(c == '\0')
1932 cp = n_UNCONST(n_empty);
1933 else
1934 ++cp;
1936 if(varbuf == cp2){
1937 n_err(_("Empty variable name ignored\n"));
1938 ++errs;
1939 }else{
1940 struct a_amv_var_carrier avc;
1941 bool_t isunset;
1943 if((isunset = (varbuf[0] == 'n' && varbuf[1] == 'o')))
1944 varbuf = &varbuf[2];
1946 a_amv_var_revlookup(&avc, varbuf);
1948 if(isunset)
1949 errs += !a_amv_var_clear(&avc, issetenv);
1950 else
1951 errs += !a_amv_var_set(&avc, cp, issetenv);
1954 NYD2_LEAVE;
1955 return (errs == 0);
1958 FL int
1959 c_define(void *v){
1960 int rv;
1961 char **args;
1962 NYD_ENTER;
1964 rv = 1;
1966 if((args = v)[0] == NULL){
1967 rv = (a_amv_mac_show(a_AMV_MF_NONE) == FAL0);
1968 goto jleave;
1971 if(args[1] == NULL || args[1][0] != '{' || args[1][1] != '\0' ||
1972 args[2] != NULL){
1973 n_err(_("Synopsis: define: <name> {\n"));
1974 goto jleave;
1977 rv = (a_amv_mac_def(args[0], a_AMV_MF_NONE) == FAL0);
1978 jleave:
1979 NYD_LEAVE;
1980 return rv;
1983 FL int
1984 c_undefine(void *v){
1985 int rv;
1986 char **args;
1987 NYD_ENTER;
1989 rv = 0;
1990 args = v;
1992 rv |= !a_amv_mac_undef(*args, a_AMV_MF_NONE);
1993 while(*++args != NULL);
1994 NYD_LEAVE;
1995 return rv;
1998 FL int
1999 c_call(void *vp){
2000 int rv;
2001 NYD_ENTER;
2003 rv = a_amv_mac_call(vp, FAL0);
2004 NYD_LEAVE;
2005 return rv;
2008 FL int
2009 c_call_if(void *vp){
2010 int rv;
2011 NYD_ENTER;
2013 rv = a_amv_mac_call(vp, TRU1);
2014 NYD_LEAVE;
2015 return rv;
2018 FL int
2019 c_account(void *v){
2020 struct a_amv_mac_call_args *amcap;
2021 struct a_amv_mac *amp;
2022 char **args;
2023 int rv, i, oqf, nqf;
2024 NYD_ENTER;
2026 rv = 1;
2028 if((args = v)[0] == NULL){
2029 rv = (a_amv_mac_show(a_AMV_MF_ACCOUNT) == FAL0);
2030 goto jleave;
2033 if(args[1] && args[1][0] == '{' && args[1][1] == '\0'){
2034 if(args[2] != NULL){
2035 n_err(_("Synopsis: account: <name> {\n"));
2036 goto jleave;
2038 if(!asccasecmp(args[0], ACCOUNT_NULL)){
2039 n_err(_("`account': cannot use reserved name: %s\n"),
2040 ACCOUNT_NULL);
2041 goto jleave;
2043 rv = (a_amv_mac_def(args[0], a_AMV_MF_ACCOUNT) == FAL0);
2044 goto jleave;
2047 if(n_pstate & n_PS_HOOK_MASK){
2048 n_err(_("`account': can't change account from within a hook\n"));
2049 goto jleave;
2052 save_mbox_for_possible_quitstuff();
2054 amp = NULL;
2055 if(asccasecmp(args[0], ACCOUNT_NULL) != 0 &&
2056 (amp = a_amv_mac_lookup(args[0], NULL, a_AMV_MF_ACCOUNT)) == NULL){
2057 n_err(_("`account': account does not exist: %s\n"), args[0]);
2058 goto jleave;
2061 oqf = savequitflags();
2063 /* Shutdown the active account */
2064 if(a_amv_acc_curr != NULL){
2065 char const *cp;
2066 char *var;
2068 /* Is there a cleanup hook? */
2069 var = savecat("on-account-cleanup-", a_amv_acc_curr->am_name);
2070 if((cp = n_var_vlook(var, FAL0)) != NULL ||
2071 (cp = ok_vlook(on_account_cleanup)) != NULL){
2072 struct a_amv_mac *amphook;
2074 if((amphook = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) != NULL){
2075 amcap = n_lofi_alloc(sizeof *amcap);
2076 memset(amcap, 0, sizeof *amcap);
2077 amcap->amca_name = cp;
2078 amcap->amca_amp = amphook;
2079 amcap->amca_unroller = &a_amv_acc_curr->am_lopts;
2080 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2081 amcap->amca_no_xcall = TRU1;
2082 n_pstate |= n_PS_HOOK;
2083 rv = a_amv_mac_exec(amcap);
2084 n_pstate &= ~n_PS_HOOK_MASK;
2085 }else
2086 n_err(_("*on-account-leave* hook %s does not exist\n"),
2087 n_shexp_quote_cp(cp, FAL0));
2090 /* `localopts'? */
2091 if(a_amv_acc_curr->am_lopts != NULL)
2092 a_amv_lopts_unroll(&a_amv_acc_curr->am_lopts);
2094 /* For accounts this lingers */
2095 --a_amv_acc_curr->am_refcnt;
2096 if(a_amv_acc_curr->am_flags & a_AMV_MF_DELETE)
2097 a_amv_mac_free(a_amv_acc_curr);
2100 a_amv_acc_curr = amp;
2102 /* And switch to any non-"null" account */
2103 if(amp != NULL){
2104 assert(amp->am_lopts == NULL);
2105 amcap = n_lofi_alloc(sizeof *amcap);
2106 memset(amcap, 0, sizeof *amcap);
2107 amcap->amca_name = amp->am_name;
2108 amcap->amca_amp = amp;
2109 amcap->amca_unroller = &amp->am_lopts;
2110 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2111 amcap->amca_no_xcall = TRU1;
2112 ++amp->am_refcnt; /* We may not run 0 to avoid being deleted! */
2113 if(!a_amv_mac_exec(amcap)){
2114 /* XXX account switch incomplete, unroll? */
2115 n_err(_("`account': failed to switch to account: %s\n"), amp->am_name);
2116 goto jleave;
2120 n_PS_ROOT_BLOCK((amp != NULL ? ok_vset(account, amp->am_name)
2121 : ok_vclear(account)));
2123 if(n_psonce & n_PSO_STARTED){
2124 assert(!(n_pstate & n_PS_HOOK_MASK));
2125 nqf = savequitflags(); /* TODO obsolete (leave -> void -> new box!) */
2126 restorequitflags(oqf);
2127 if((i = setfile("%", 0)) < 0)
2128 goto jleave;
2129 temporary_folder_hook_check(FAL0);
2130 if(i > 0 && !ok_blook(emptystart))
2131 goto jleave;
2132 n_folder_announce(n_ANNOUNCE_CHANGE);
2133 restorequitflags(nqf);
2135 rv = 0;
2136 jleave:
2137 NYD_LEAVE;
2138 return rv;
2141 FL int
2142 c_unaccount(void *v){
2143 int rv;
2144 char **args;
2145 NYD_ENTER;
2147 rv = 0;
2148 args = v;
2150 rv |= !a_amv_mac_undef(*args, a_AMV_MF_ACCOUNT);
2151 while(*++args != NULL);
2152 NYD_LEAVE;
2153 return rv;
2156 FL int
2157 c_localopts(void *vp){
2158 enum a_amv_loflags alf, alm;
2159 char const **argv;
2160 int rv;
2161 NYD_ENTER;
2163 rv = 1;
2165 if(a_amv_lopts == NULL){
2166 n_err(_("Cannot use `localopts' in this context "
2167 "(not in `define' or `account', nor special hook)\n"));
2168 goto jleave;
2171 if((argv = vp)[1] == NULL || is_asccaseprefix((++argv)[-1], "scope"))
2172 alf = alm = a_AMV_LF_SCOPE;
2173 else if(is_asccaseprefix(argv[-1], "call"))
2174 alf = a_AMV_LF_CALL, alm = a_AMV_LF_CALL_MASK;
2175 else if(is_asccaseprefix(argv[-1], "call-fixate"))
2176 alf = a_AMV_LF_CALL_FIXATE, alm = a_AMV_LF_CALL_MASK;
2177 else{
2178 n_err(_("Synopsis: localopts: [<scope|call|call-fixate>] <boolean>\n"));
2179 goto jleave;
2182 if(alf == a_AMV_LF_SCOPE &&
2183 (a_amv_lopts->as_loflags & a_AMV_LF_SCOPE_FIXATE)){
2184 if(n_poption & n_PO_D_V)
2185 n_err(_("Cannot turn off `localopts', setting is fixated\n"));
2186 goto jleave;
2189 a_amv_lopts->as_loflags &= ~alm;
2190 if(boolify(*argv, UIZ_MAX, FAL0) > 0)
2191 a_amv_lopts->as_loflags |= alf;
2192 rv = 0;
2193 jleave:
2194 NYD_LEAVE;
2195 return rv;
2198 FL int
2199 c_shift(void *vp){
2200 struct a_amv_pospar *appp;
2201 ui16_t i;
2202 int rv;
2203 NYD_ENTER;
2205 rv = 1;
2207 if((vp = *(char**)vp) == NULL)
2208 i = 1;
2209 else{
2210 si16_t sib;
2212 if((n_idec_si16_cp(&sib, vp, 10, NULL
2213 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2214 ) != n_IDEC_STATE_CONSUMED || sib < 0){
2215 n_err(_("`shift': invalid argument: %s\n"), vp);
2216 goto jleave;
2218 i = (ui16_t)sib;
2221 /* If in in a macro/xy */
2222 if(a_amv_lopts != NULL){
2223 struct a_amv_mac const *amp;
2224 struct a_amv_mac_call_args *amcap;
2226 /* Explicitly do allow `vpospar' created things! */
2227 amp = (amcap = a_amv_lopts->as_amcap)->amca_amp;
2228 if((amp == NULL || amcap->amca_ps_hook_mask ||
2229 (amp->am_flags & a_AMV_MF_TYPE_MASK) == a_AMV_MF_ACCOUNT) &&
2230 amcap->amca_pospar.app_not_heap){
2231 n_err(_("Cannot use `shift' in `account's or hook macros etc.\n"));
2232 goto jleave;
2234 appp = &amcap->amca_pospar;
2235 }else
2236 appp = &a_amv_pospar;
2238 if(i > appp->app_count){
2239 n_err(_("`shift': cannot shift %hu of %hu parameters\n"),
2240 i, appp->app_count);
2241 goto jleave;
2242 }else{
2243 appp->app_idx += i;
2244 appp->app_count -= i;
2245 rv = 0;
2247 jleave:
2248 NYD_LEAVE;
2249 return rv;
2252 FL int
2253 c_return(void *vp){ /* TODO the exit status should be m_si64! */
2254 int rv;
2255 NYD_ENTER;
2257 if(a_amv_lopts != NULL){
2258 char const **argv;
2260 n_go_input_force_eof();
2261 n_pstate_err_no = n_ERR_NONE;
2262 rv = 0;
2264 if((argv = vp)[0] != NULL){
2265 si32_t i;
2267 if((n_idec_si32_cp(&i, argv[0], 10, NULL
2268 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2269 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2270 rv = (int)i;
2271 else{
2272 n_err(_("`return': return value argument is invalid: %s\n"),
2273 argv[0]);
2274 n_pstate_err_no = n_ERR_INVAL;
2275 rv = 1;
2278 if(argv[1] != NULL){
2279 if((n_idec_si32_cp(&i, argv[1], 10, NULL
2280 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2281 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2282 n_pstate_err_no = i;
2283 else{
2284 n_err(_("`return': error number argument is invalid: %s\n"),
2285 argv[1]);
2286 n_pstate_err_no = n_ERR_INVAL;
2287 rv = 1;
2291 }else{
2292 n_err(_("Can only use `return' in a macro\n"));
2293 n_pstate_err_no = n_ERR_OPNOTSUPP;
2294 rv = 1;
2296 NYD_LEAVE;
2297 return rv;
2300 FL bool_t
2301 temporary_folder_hook_check(bool_t nmail){ /* TODO temporary, v15: drop */
2302 struct a_amv_mac_call_args *amcap;
2303 struct a_amv_mac *amp;
2304 size_t len;
2305 char const *cp;
2306 char *var;
2307 bool_t rv;
2308 NYD_ENTER;
2310 rv = TRU1;
2311 var = n_autorec_alloc(len = strlen(mailname) + sizeof("folder-hook-") -1 +1);
2313 /* First try the fully resolved path */
2314 snprintf(var, len, "folder-hook-%s", mailname);
2315 if((cp = n_var_vlook(var, FAL0)) != NULL)
2316 goto jmac;
2318 /* If we are under *folder*, try the usual +NAME syntax, too */
2319 if(displayname[0] == '+'){
2320 char *x;
2322 for(x = &mailname[len]; x != mailname; --x)
2323 if(x[-1] == '/'){
2324 snprintf(var, len, "folder-hook-+%s", x);
2325 if((cp = n_var_vlook(var, FAL0)) != NULL)
2326 goto jmac;
2327 break;
2331 /* Plain *folder-hook* is our last try */
2332 if((cp = ok_vlook(folder_hook)) == NULL)
2333 goto jleave;
2335 jmac:
2336 if((amp = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) == NULL){
2337 n_err(_("Cannot call *folder-hook* for %s: macro does not exist: %s\n"),
2338 n_shexp_quote_cp(displayname, FAL0), cp);
2339 rv = FAL0;
2340 goto jleave;
2343 amcap = n_lofi_alloc(sizeof *amcap);
2344 memset(amcap, 0, sizeof *amcap);
2345 amcap->amca_name = cp;
2346 amcap->amca_amp = amp;
2347 n_pstate &= ~n_PS_HOOK_MASK;
2348 if(nmail){
2349 amcap->amca_unroller = NULL;
2350 n_pstate |= n_PS_HOOK_NEWMAIL;
2351 }else{
2352 amcap->amca_unroller = &a_amv_folder_hook_lopts;
2353 n_pstate |= n_PS_HOOK;
2355 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2356 amcap->amca_ps_hook_mask = TRU1;
2357 amcap->amca_no_xcall = TRU1;
2358 rv = a_amv_mac_exec(amcap);
2359 n_pstate &= ~n_PS_HOOK_MASK;
2361 jleave:
2362 NYD_LEAVE;
2363 return rv;
2366 FL void
2367 temporary_folder_hook_unroll(void){ /* XXX intermediate hack */
2368 NYD_ENTER;
2369 if(a_amv_folder_hook_lopts != NULL){
2370 void *save = a_amv_lopts;
2372 a_amv_lopts = NULL;
2373 a_amv_lopts_unroll(&a_amv_folder_hook_lopts);
2374 assert(a_amv_folder_hook_lopts == NULL);
2375 a_amv_lopts = save;
2377 NYD_LEAVE;
2380 FL void
2381 temporary_compose_mode_hook_call(char const *macname,
2382 void (*hook_pre)(void *), void *hook_arg){
2383 /* TODO compose_mode_hook_call() temporary, v15: generalize; see a_GO_SPLICE
2384 * TODO comment in go.c for the right way of doing things! */
2385 static struct a_amv_lostack *cmh_losp;
2386 struct a_amv_mac_call_args *amcap;
2387 struct a_amv_mac *amp;
2388 NYD_ENTER;
2390 amp = NULL;
2392 if(macname == (char*)-1){
2393 a_amv_mac__finalize(cmh_losp);
2394 cmh_losp = NULL;
2395 }else if(macname != NULL &&
2396 (amp = a_amv_mac_lookup(macname, NULL, a_AMV_MF_NONE)) == NULL)
2397 n_err(_("Cannot call *on-compose-**: macro does not exist: %s\n"),
2398 macname);
2399 else{
2400 amcap = n_lofi_alloc(sizeof *amcap);
2401 memset(amcap, 0, sizeof *amcap);
2402 amcap->amca_name = (macname != NULL) ? macname
2403 : "*on-compose-splice(-shell)?*";
2404 amcap->amca_amp = amp;
2405 amcap->amca_unroller = &a_amv_compose_lopts;
2406 amcap->amca_hook_pre = hook_pre;
2407 amcap->amca_hook_arg = hook_arg;
2408 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2409 amcap->amca_ps_hook_mask = TRU1;
2410 amcap->amca_no_xcall = TRU1;
2411 n_pstate &= ~n_PS_HOOK_MASK;
2412 n_pstate |= n_PS_HOOK;
2413 if(macname != NULL)
2414 a_amv_mac_exec(amcap);
2415 else{
2416 cmh_losp = n_lofi_alloc(sizeof *cmh_losp);
2417 memset(cmh_losp, 0, sizeof *cmh_losp);
2418 cmh_losp->as_global_saved = a_amv_lopts;
2419 cmh_losp->as_lopts = *(cmh_losp->as_amcap = amcap)->amca_unroller;
2420 cmh_losp->as_loflags = a_AMV_LF_SCOPE_FIXATE;
2421 a_amv_lopts = cmh_losp;
2424 NYD_LEAVE;
2427 FL void
2428 temporary_compose_mode_hook_unroll(void){ /* XXX intermediate hack */
2429 NYD_ENTER;
2430 if(a_amv_compose_lopts != NULL){
2431 void *save = a_amv_lopts;
2433 a_amv_lopts = NULL;
2434 a_amv_lopts_unroll(&a_amv_compose_lopts);
2435 assert(a_amv_compose_lopts == NULL);
2436 a_amv_lopts = save;
2438 NYD_LEAVE;
2441 FL bool_t
2442 n_var_is_user_writable(char const *name){
2443 struct a_amv_var_carrier avc;
2444 struct a_amv_var_map const *avmp;
2445 bool_t rv;
2446 NYD_ENTER;
2448 a_amv_var_revlookup(&avc, name);
2449 if((avmp = avc.avc_map) == NULL)
2450 rv = TRU1;
2451 else
2452 rv = ((avmp->avm_flags & (a_AMV_VF_BOOL | a_AMV_VF_RDONLY)) == 0);
2453 NYD_LEAVE;
2454 return rv;
2457 FL char *
2458 n_var_oklook(enum okeys okey){
2459 struct a_amv_var_carrier avc;
2460 char *rv;
2461 struct a_amv_var_map const *avmp;
2462 NYD_ENTER;
2464 avc.avc_map = avmp = &a_amv_var_map[okey];
2465 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2466 avc.avc_hash = avmp->avm_hash;
2467 avc.avc_okey = okey;
2469 if(a_amv_var_lookup(&avc, FAL0))
2470 rv = avc.avc_var->av_value;
2471 else
2472 rv = NULL;
2473 NYD_LEAVE;
2474 return rv;
2477 FL bool_t
2478 n_var_okset(enum okeys okey, uintptr_t val){
2479 struct a_amv_var_carrier avc;
2480 bool_t ok;
2481 struct a_amv_var_map const *avmp;
2482 NYD_ENTER;
2484 avc.avc_map = avmp = &a_amv_var_map[okey];
2485 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2486 avc.avc_hash = avmp->avm_hash;
2487 avc.avc_okey = okey;
2489 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val), FAL0);
2490 NYD_LEAVE;
2491 return ok;
2494 FL bool_t
2495 n_var_okclear(enum okeys okey){
2496 struct a_amv_var_carrier avc;
2497 bool_t rv;
2498 struct a_amv_var_map const *avmp;
2499 NYD_ENTER;
2501 avc.avc_map = avmp = &a_amv_var_map[okey];
2502 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2503 avc.avc_hash = avmp->avm_hash;
2504 avc.avc_okey = okey;
2506 rv = a_amv_var_clear(&avc, FAL0);
2507 NYD_LEAVE;
2508 return rv;
2511 FL char const *
2512 n_var_vlook(char const *vokey, bool_t try_getenv){
2513 struct a_amv_var_carrier avc;
2514 char const *rv;
2515 NYD_ENTER;
2517 a_amv_var_revlookup(&avc, vokey);
2519 switch((enum a_amv_var_special_category)avc.avc_special_cat){
2520 default: /* silence CC */
2521 case a_AMV_VSC_NONE:
2522 if(a_amv_var_lookup(&avc, FAL0))
2523 rv = avc.avc_var->av_value;
2524 /* Only check the environment for something that is otherwise unknown */
2525 else if(try_getenv && avc.avc_map == NULL)
2526 rv = getenv(vokey);
2527 else
2528 rv = NULL;
2529 break;
2530 case a_AMV_VSC_GLOBAL:
2531 rv = a_amv_var_vsc_global(&avc);
2532 break;
2533 case a_AMV_VSC_MULTIPLEX:
2534 rv = a_amv_var_vsc_multiplex(&avc);
2535 break;
2536 case a_AMV_VSC_POSPAR:
2537 case a_AMV_VSC_POSPAR_ENV:
2538 rv = a_amv_var_vsc_pospar(&avc);
2539 break;
2541 NYD_LEAVE;
2542 return rv;
2545 FL bool_t
2546 n_var_vexplode(void const **cookie){
2547 struct a_amv_pospar *appp;
2548 NYD_ENTER;
2550 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
2551 : &a_amv_pospar;
2552 *cookie = (appp->app_count > 0) ? &appp->app_dat[appp->app_idx] : NULL;
2553 NYD_LEAVE;
2554 return (*cookie != NULL);
2557 FL bool_t
2558 n_var_vset(char const *vokey, uintptr_t val){
2559 struct a_amv_var_carrier avc;
2560 bool_t ok;
2561 NYD_ENTER;
2563 a_amv_var_revlookup(&avc, vokey);
2565 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val), FAL0);
2566 NYD_LEAVE;
2567 return ok;
2570 FL bool_t
2571 n_var_vclear(char const *vokey){
2572 struct a_amv_var_carrier avc;
2573 bool_t ok;
2574 NYD_ENTER;
2576 a_amv_var_revlookup(&avc, vokey);
2578 ok = a_amv_var_clear(&avc, FAL0);
2579 NYD_LEAVE;
2580 return ok;
2583 #ifdef HAVE_SOCKETS
2584 FL char *
2585 n_var_xoklook(enum okeys okey, struct url const *urlp,
2586 enum okey_xlook_mode oxm){
2587 struct a_amv_var_carrier avc;
2588 struct str const *us;
2589 size_t nlen;
2590 char *nbuf, *rv;
2591 struct a_amv_var_map const *avmp;
2592 NYD_ENTER;
2594 assert(oxm & (OXM_PLAIN | OXM_H_P | OXM_U_H_P));
2596 /* For simplicity: allow this case too */
2597 if(!(oxm & (OXM_H_P | OXM_U_H_P))){
2598 nbuf = NULL;
2599 goto jplain;
2602 avc.avc_map = avmp = &a_amv_var_map[okey];
2603 avc.avc_name = a_amv_var_names + avmp->avm_keyoff;
2604 avc.avc_okey = okey;
2606 us = (oxm & OXM_U_H_P) ? &urlp->url_u_h_p : &urlp->url_h_p;
2607 nlen = strlen(avc.avc_name);
2608 nbuf = n_lofi_alloc(nlen + 1 + us->l +1);
2609 memcpy(nbuf, avc.avc_name, nlen);
2610 nbuf[nlen++] = '-';
2612 /* One of .url_u_h_p and .url_h_p we test in here */
2613 memcpy(nbuf + nlen, us->s, us->l +1);
2614 avc.avc_name = a_amv_var_canonify(nbuf);
2615 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2616 if(a_amv_var_lookup(&avc, FAL0))
2617 goto jvar;
2619 /* The second */
2620 if(oxm & OXM_H_P){
2621 us = &urlp->url_h_p;
2622 memcpy(nbuf + nlen, us->s, us->l +1);
2623 avc.avc_name = a_amv_var_canonify(nbuf);
2624 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2625 if(a_amv_var_lookup(&avc, FAL0)){
2626 jvar:
2627 rv = avc.avc_var->av_value;
2628 goto jleave;
2632 jplain:
2633 rv = (oxm & OXM_PLAIN) ? n_var_oklook(okey) : NULL;
2634 jleave:
2635 if(nbuf != NULL)
2636 n_lofi_free(nbuf);
2637 NYD_LEAVE;
2638 return rv;
2640 #endif /* HAVE_SOCKETS */
2642 FL int
2643 c_set(void *v){
2644 char **ap;
2645 int err;
2646 NYD_ENTER;
2648 if(*(ap = v) == NULL){
2649 a_amv_var_show_all();
2650 err = 0;
2651 }else
2652 err = !a_amv_var_c_set(ap, FAL0);
2653 NYD_LEAVE;
2654 return err;
2657 FL int
2658 c_unset(void *v){
2659 char **ap;
2660 int err;
2661 NYD_ENTER;
2663 for(err = 0, ap = v; *ap != NULL; ++ap)
2664 err |= !n_var_vclear(*ap);
2665 NYD_LEAVE;
2666 return err;
2669 FL int
2670 c_varshow(void *v){
2671 char **ap;
2672 NYD_ENTER;
2674 if(*(ap = v) == NULL)
2675 v = NULL;
2676 else{
2677 struct n_string msg, *msgp = &msg;
2679 msgp = n_string_creat(msgp);
2680 for(; *ap != NULL; ++ap)
2681 a_amv_var_show(*ap, n_stdout, msgp);
2682 n_string_gut(msgp);
2684 NYD_LEAVE;
2685 return (v == NULL ? !STOP : !OKAY); /* xxx 1:bad 0:good -- do some */
2688 FL int
2689 c_varedit(void *v){
2690 struct a_amv_var_carrier avc;
2691 FILE *of, *nf;
2692 char *val, **argv;
2693 int err;
2694 sighandler_type sigint;
2695 NYD_ENTER;
2697 sigint = safe_signal(SIGINT, SIG_IGN);
2699 for(err = 0, argv = v; *argv != NULL; ++argv){
2700 memset(&avc, 0, sizeof avc);
2702 a_amv_var_revlookup(&avc, *argv);
2704 if(avc.avc_map != NULL){
2705 if(avc.avc_map->avm_flags & a_AMV_VF_BOOL){
2706 n_err(_("`varedit': cannot edit boolean variable: %s\n"),
2707 avc.avc_name);
2708 continue;
2710 if(avc.avc_map->avm_flags & a_AMV_VF_RDONLY){
2711 n_err(_("`varedit': cannot edit read-only variable: %s\n"),
2712 avc.avc_name);
2713 continue;
2717 a_amv_var_lookup(&avc, FAL0);
2719 if((of = Ftmp(NULL, "varedit", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
2720 NULL){
2721 n_perr(_("`varedit': can't create temporary file, bailing out"), 0);
2722 err = 1;
2723 break;
2724 }else if(avc.avc_var != NULL && *(val = avc.avc_var->av_value) != '\0' &&
2725 sizeof *val != fwrite(val, strlen(val), sizeof *val, of)){
2726 n_perr(_("`varedit' failed to write old value to temporary file"), 0);
2727 Fclose(of);
2728 err = 1;
2729 continue;
2732 fflush_rewind(of);
2733 nf = run_editor(of, (off_t)-1, 'e', FAL0, NULL, NULL, SEND_MBOX, sigint);
2734 Fclose(of);
2736 if(nf != NULL){
2737 int c;
2738 char const *varres;
2739 off_t l;
2741 l = fsize(nf);
2742 if(UICMP(64, l, >=, UIZ_MAX -42)){
2743 n_err(_("`varedit': not enough memory to store variable: %s\n"),
2744 avc.avc_name);
2745 varres = n_empty;
2746 err = 1;
2747 }else{
2748 varres = val = n_autorec_alloc(l +1);
2749 for(; l > 0 && (c = getc(nf)) != EOF; --l)
2750 *val++ = c;
2751 *val++ = '\0';
2752 if(l != 0){
2753 n_err(_("`varedit': I/O while reading new value of: %s\n"),
2754 avc.avc_name);
2755 err = 1;
2759 if(!a_amv_var_set(&avc, varres, FAL0))
2760 err = 1;
2762 Fclose(nf);
2763 }else{
2764 n_err(_("`varedit': can't start $EDITOR, bailing out\n"));
2765 err = 1;
2766 break;
2770 safe_signal(SIGINT, sigint);
2771 NYD_LEAVE;
2772 return err;
2775 FL int
2776 c_environ(void *v){
2777 struct a_amv_var_carrier avc;
2778 int err;
2779 char **ap;
2780 bool_t islnk;
2781 NYD_ENTER;
2783 if((islnk = is_asccaseprefix(*(ap = v), "link")) ||
2784 is_asccaseprefix(*ap, "unlink")){
2785 for(err = 0; *++ap != NULL;){
2786 a_amv_var_revlookup(&avc, *ap);
2788 if(a_amv_var_lookup(&avc, FAL0) && (islnk ||
2789 (avc.avc_var->av_flags & a_AMV_VF_LINKED))){
2790 if(!islnk){
2791 avc.avc_var->av_flags &= ~a_AMV_VF_LINKED;
2792 continue;
2793 }else if(avc.avc_var->av_flags & (a_AMV_VF_ENV | a_AMV_VF_LINKED)){
2794 if(n_poption & n_PO_D_V)
2795 n_err(_("`environ': link: already established: %s\n"), *ap);
2796 continue;
2798 avc.avc_var->av_flags |= a_AMV_VF_LINKED;
2799 if(!(avc.avc_var->av_flags & a_AMV_VF_ENV))
2800 a_amv_var__putenv(&avc, avc.avc_var);
2801 }else if(!islnk){
2802 n_err(_("`environ': unlink: no link established: %s\n"), *ap);
2803 err = 1;
2804 }else{
2805 char const *evp = getenv(*ap);
2807 if(evp != NULL)
2808 err |= !a_amv_var_set(&avc, evp, TRU1);
2809 else{
2810 n_err(_("`environ': link: cannot link to non-existent: %s\n"),
2811 *ap);
2812 err = 1;
2816 }else if(is_asccaseprefix(*ap, "set"))
2817 err = !a_amv_var_c_set(++ap, TRU1);
2818 else if(is_asccaseprefix(*ap, "unset")){
2819 for(err = 0; *++ap != NULL;){
2820 a_amv_var_revlookup(&avc, *ap);
2822 if(!a_amv_var_clear(&avc, TRU1))
2823 err = 1;
2825 }else{
2826 n_err(_("Synopsis: environ: <link|set|unset> <variable>...\n"));
2827 err = 1;
2829 NYD_LEAVE;
2830 return err;
2833 FL int
2834 c_vexpr(void *v){ /* TODO POSIX expr(1) comp. exit status; overly complicat. */
2835 size_t i;
2836 enum n_idec_state ids;
2837 si64_t lhv, rhv;
2838 char op, varbuf[64 + 64 / 8 +1];
2839 char const **argv, *varname, *varres, *cp;
2840 enum{
2841 a_ERR = 1u<<0,
2842 a_SOFTOVERFLOW = 1u<<1,
2843 a_ISNUM = 1u<<2,
2844 a_ISDECIMAL = 1u<<3, /* Print only decimal result */
2845 a_SATURATED = 1u<<4,
2846 a_ICASE = 1u<<5,
2847 a_UNSIGNED = 1u<<6, /* Unsigned right shift (share bit ok) */
2848 a_TMP = 1u<<30
2849 } f;
2850 NYD_ENTER;
2852 f = a_ERR;
2853 argv = v;
2854 varname = (n_pstate & n_PS_ARGMOD_VPUT) ? *argv++ : NULL;
2855 n_UNINIT(varres, n_empty);
2857 if((cp = argv[0])[0] == '\0')
2858 goto jesubcmd;
2860 if(cp[1] == '\0'){
2861 op = cp[0];
2862 jnumop:
2863 f |= a_ISNUM;
2864 switch(op){
2865 case '=':
2866 case '~':
2867 if(argv[1] == NULL || argv[2] != NULL)
2868 goto jesynopsis;
2870 if(*(cp = *++argv) == '\0')
2871 lhv = 0;
2872 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
2873 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2874 ) != n_IDEC_STATE_CONSUMED){
2875 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
2876 goto jenum_range;
2877 f |= a_SOFTOVERFLOW;
2878 break;
2880 if(op == '~')
2881 lhv = ~lhv;
2882 break;
2884 case '+':
2885 case '-':
2886 case '*':
2887 case '/':
2888 case '%':
2889 case '|':
2890 case '&':
2891 case '^':
2892 case '<':
2893 case '>':
2894 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
2895 goto jesynopsis;
2896 else{
2897 char xop;
2899 if(*(cp = *++argv) == '\0')
2900 lhv = 0;
2901 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
2902 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2903 ) != n_IDEC_STATE_CONSUMED){
2904 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
2905 goto jenum_range;
2906 f |= a_SOFTOVERFLOW;
2907 break;
2910 if(*(cp = *++argv) == '\0')
2911 rhv = 0;
2912 else if(((ids = n_idec_si64_cp(&rhv, cp, 0, NULL)
2913 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2914 ) != n_IDEC_STATE_CONSUMED){
2915 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
2916 goto jenum_range;
2917 f |= a_SOFTOVERFLOW;
2918 lhv = rhv;
2919 break;
2922 xop = op;
2923 jnumop_again:
2924 switch(xop){
2925 case '+':
2926 if(rhv < 0){
2927 if(rhv != SI64_MIN){
2928 rhv = -rhv;
2929 xop = '-';
2930 goto jnumop_again;
2931 }else if(lhv < 0)
2932 goto jenum_plusminus;
2933 else if(lhv == 0){
2934 lhv = rhv;
2935 break;
2938 if(SI64_MAX - rhv < lhv)
2939 goto jenum_plusminus;
2940 lhv += rhv;
2941 break;
2942 case '-':
2943 if(rhv < 0){
2944 if(rhv != SI64_MIN){
2945 rhv = -rhv;
2946 xop = '+';
2947 goto jnumop_again;
2948 }else if(lhv > 0)
2949 goto jenum_plusminus;
2950 else if(lhv == 0){
2951 lhv = rhv;
2952 break;
2955 if(SI64_MIN + rhv > lhv){
2956 jenum_plusminus:
2957 if(!(f & a_SATURATED))
2958 goto jenum_overflow;
2959 f |= a_SOFTOVERFLOW;
2960 lhv = (lhv < 0 || xop == '-') ? SI64_MIN : SI64_MAX;
2961 }else
2962 lhv -= rhv;
2963 break;
2964 case '*':
2965 /* Will the result be positive? */
2966 if((lhv < 0) == (rhv < 0)){
2967 if(lhv > 0){
2968 lhv = -lhv;
2969 rhv = -rhv;
2971 if(rhv != 0 && lhv != 0 && SI64_MAX / rhv > lhv){
2972 if(!(f & a_SATURATED))
2973 goto jenum_overflow;
2974 f |= a_SOFTOVERFLOW;
2975 lhv = SI64_MAX;
2976 }else
2977 lhv *= rhv;
2978 }else{
2979 if(rhv > 0){
2980 if(lhv != 0 && SI64_MIN / lhv < rhv){
2981 if(!(f & a_SATURATED))
2982 goto jenum_overflow;
2983 f |= a_SOFTOVERFLOW;
2984 lhv = SI64_MIN;
2985 }else
2986 lhv *= rhv;
2987 }else{
2988 if(rhv != 0 && lhv != 0 && SI64_MIN / rhv < lhv){
2989 if(!(f & a_SATURATED))
2990 goto jenum_overflow;
2991 f |= a_SOFTOVERFLOW;
2992 lhv = SI64_MIN;
2993 }else
2994 lhv *= rhv;
2997 break;
2998 case '/':
2999 if(rhv == 0){
3000 if(!(f & a_SATURATED))
3001 goto jenum_range;
3002 f |= a_SOFTOVERFLOW;
3003 lhv = SI64_MAX;
3004 }else
3005 lhv /= rhv;
3006 break;
3007 case '%':
3008 if(rhv == 0){
3009 if(!(f & a_SATURATED))
3010 goto jenum_range;
3011 f |= a_SOFTOVERFLOW;
3012 lhv = SI64_MAX;
3013 }else
3014 lhv %= rhv;
3015 break;
3016 case '|':
3017 lhv |= rhv;
3018 break;
3019 case '&':
3020 lhv &= rhv;
3021 break;
3022 case '^':
3023 lhv ^= rhv;
3024 break;
3025 case '<':
3026 case '>':
3027 if(!(f & a_TMP))
3028 goto jesubcmd;
3029 if(rhv > 63){ /* xxx 63? */
3030 if(!(f & a_SATURATED))
3031 goto jenum_overflow;
3032 rhv = 63;
3034 if(op == '<')
3035 lhv <<= (ui8_t)rhv;
3036 else if(f & a_UNSIGNED)
3037 lhv = (ui64_t)lhv >> (ui8_t)rhv;
3038 else
3039 lhv >>= (ui8_t)rhv;
3040 break;
3043 break;
3044 default:
3045 goto jesubcmd;
3047 }else if(cp[2] == '\0' && cp[1] == '@'){
3048 f |= a_SATURATED;
3049 op = cp[0];
3050 goto jnumop;
3051 }else if(cp[0] == '<'){
3052 if(*++cp != '<')
3053 goto jesubcmd;
3054 if(*++cp == '@'){
3055 f |= a_SATURATED;
3056 ++cp;
3058 if(*cp != '\0')
3059 goto jesubcmd;
3060 f |= a_TMP;
3061 op = '<';
3062 goto jnumop;
3063 }else if(cp[0] == '>'){
3064 if(*++cp != '>')
3065 goto jesubcmd;
3066 if(*++cp == '>'){
3067 f |= a_UNSIGNED;
3068 ++cp;
3070 if(*cp == '@'){
3071 f |= a_SATURATED;
3072 ++cp;
3074 if(*cp != '\0')
3075 goto jesubcmd;
3076 f |= a_TMP;
3077 op = '>';
3078 goto jnumop;
3079 }else if(is_asccaseprefix(cp, "length")){
3080 f |= a_ISNUM | a_ISDECIMAL;
3081 if(argv[1] == NULL || argv[2] != NULL)
3082 goto jesynopsis;
3084 i = strlen(*++argv);
3085 if(UICMP(64, i, >, SI64_MAX))
3086 goto jestr_overflow;
3087 lhv = (si64_t)i;
3088 }else if(is_asccaseprefix(cp, "hash")){
3089 f |= a_ISNUM | a_ISDECIMAL;
3090 if(argv[1] == NULL || argv[2] != NULL)
3091 goto jesynopsis;
3093 i = n_torek_hash(*++argv);
3094 lhv = (si64_t)i;
3095 }else if(is_asccaseprefix(cp, "find")){
3096 f |= a_ISNUM | a_ISDECIMAL;
3097 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3098 goto jesynopsis;
3100 if((cp = strstr(argv[1], argv[2])) == NULL)
3101 goto jestr_nodata;
3102 i = PTR2SIZE(cp - argv[1]);
3103 if(UICMP(64, i, >, SI64_MAX))
3104 goto jestr_overflow;
3105 lhv = (si64_t)i;
3106 }else if(is_asccaseprefix(cp, "ifind")){
3107 f |= a_ISNUM | a_ISDECIMAL;
3108 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3109 goto jesynopsis;
3111 if((cp = asccasestr(argv[1], argv[2])) == NULL)
3112 goto jestr_nodata;
3113 i = PTR2SIZE(cp - argv[1]);
3114 if(UICMP(64, i, >, SI64_MAX))
3115 goto jestr_overflow;
3116 lhv = (si64_t)i;
3117 }else if(is_asccaseprefix(cp, "substring")){
3118 if(argv[1] == NULL || argv[2] == NULL)
3119 goto jesynopsis;
3120 if(argv[3] != NULL && argv[4] != NULL)
3121 goto jesynopsis;
3123 i = strlen(varres = *++argv);
3125 if(*(cp = *++argv) == '\0')
3126 lhv = 0;
3127 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3128 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3129 ) != n_IDEC_STATE_CONSUMED)
3130 goto jestr_numrange;
3131 if(lhv < 0){
3132 if(UICMP(64, i, <, -lhv))
3133 goto jesubstring_off;
3134 lhv = i + lhv;
3136 if(UICMP(64, i, >=, lhv)){
3137 i -= lhv;
3138 varres += lhv;
3139 }else{
3140 jesubstring_off:
3141 if(n_poption & n_PO_D_V)
3142 n_err(_("`vexpr': substring: offset argument too large: %s\n"),
3143 n_shexp_quote_cp(argv[-1], FAL0));
3144 f |= a_SOFTOVERFLOW;
3147 if(argv[1] != NULL){
3148 if(*(cp = *++argv) == '\0')
3149 lhv = 0;
3150 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3151 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3152 ) != n_IDEC_STATE_CONSUMED)
3153 goto jestr_numrange;
3154 if(lhv < 0){
3155 if(UICMP(64, i, <, -lhv))
3156 goto jesubstring_len;
3157 lhv = i + lhv;
3159 if(UICMP(64, i, >=, lhv)){
3160 if(UICMP(64, i, !=, lhv))
3161 varres = savestrbuf(varres, (size_t)lhv);
3162 }else{
3163 jesubstring_len:
3164 if(n_poption & n_PO_D_V)
3165 n_err(_("`vexpr': substring: length argument too large: %s\n"),
3166 n_shexp_quote_cp(argv[-2], FAL0));
3167 f |= a_SOFTOVERFLOW;
3170 }else if(is_asccaseprefix(cp, "trim")) Jtrim: {
3171 struct str trim;
3172 enum n_str_trim_flags stf;
3174 if(argv[1] == NULL || argv[2] != NULL)
3175 goto jesynopsis;
3177 if(is_asccaseprefix(cp, "trim-front"))
3178 stf = n_STR_TRIM_FRONT;
3179 else if(is_asccaseprefix(cp, "trim-end"))
3180 stf = n_STR_TRIM_END;
3181 else
3182 stf = n_STR_TRIM_BOTH;
3184 trim.l = strlen(trim.s = n_UNCONST(argv[1]));
3185 (void)n_str_trim(&trim, stf);
3186 varres = savestrbuf(trim.s, trim.l);
3187 }else if(is_asccaseprefix(cp, "trim-front"))
3188 goto Jtrim;
3189 else if(is_asccaseprefix(cp, "trim-end"))
3190 goto Jtrim;
3191 else if(is_asccaseprefix(cp, "random")){
3192 if(argv[1] == NULL || argv[2] != NULL)
3193 goto jesynopsis;
3195 if((n_idec_si64_cp(&lhv, argv[1], 0, NULL
3196 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3197 ) != n_IDEC_STATE_CONSUMED || lhv < 0 || lhv > PATH_MAX)
3198 goto jestr_numrange;
3199 if(lhv == 0)
3200 lhv = NAME_MAX;
3201 varres = n_random_create_cp((size_t)lhv, NULL);
3202 }else if(is_asccaseprefix(cp, "file-expand")){
3203 if(argv[1] == NULL || argv[2] != NULL)
3204 goto jesynopsis;
3206 if((varres = fexpand(argv[1], FEXP_NVAR | FEXP_NOPROTO)) == NULL)
3207 goto jestr_nodata;
3208 }else if(is_asccaseprefix(cp, "makeprint")){
3209 struct str sin, sout;
3211 if(argv[1] == NULL || argv[2] != NULL)
3212 goto jesynopsis;
3214 /* XXX using strlen for `vexpr makeprint' is wrong for UTF-16 */
3215 sin.l = strlen(sin.s = n_UNCONST(argv[1]));
3216 makeprint(&sin, &sout);
3217 varres = savestrbuf(sout.s, sout.l);
3218 n_free(sout.s);
3219 /* TODO `vexpr': (wide) string length, find, etc!! */
3220 #ifdef HAVE_REGEX
3221 }else if(is_asccaseprefix(cp, "regex")) Jregex:{
3222 regmatch_t rema[1 + n_VEXPR_REGEX_MAX];
3223 regex_t re;
3224 int reflrv;
3226 f |= a_ISNUM | a_ISDECIMAL;
3227 if(argv[1] == NULL || argv[2] == NULL ||
3228 (argv[3] != NULL && argv[4] != NULL))
3229 goto jesynopsis;
3231 reflrv = REG_EXTENDED;
3232 if(f & a_ICASE)
3233 reflrv |= REG_ICASE;
3234 if((reflrv = regcomp(&re, argv[2], reflrv))){
3235 n_err(_("`vexpr': invalid regular expression: %s: %s\n"),
3236 n_shexp_quote_cp(argv[2], FAL0), n_regex_err_to_doc(&re, reflrv));
3237 assert(f & a_ERR);
3238 n_pstate_err_no = n_ERR_INVAL;
3239 goto jestr;
3241 reflrv = regexec(&re, argv[1], n_NELEM(rema), rema, 0);
3242 regfree(&re);
3243 if(reflrv == REG_NOMATCH)
3244 goto jestr_nodata;
3246 /* Search only? Else replace, which is a bit */
3247 if(argv[3] == NULL){
3248 if(UICMP(64, rema[0].rm_so, >, SI64_MAX))
3249 goto jestr_overflow;
3250 lhv = (si64_t)rema[0].rm_so;
3251 }else{
3252 /* We need to setup some kind of pseudo macro environment for this */
3253 struct a_amv_lostack los;
3254 struct a_amv_mac_call_args amca;
3255 char const **reargv;
3257 memset(&amca, 0, sizeof amca);
3258 amca.amca_name = savestrbuf(&argv[1][rema[0].rm_so],
3259 rema[0].rm_eo - rema[0].rm_so);
3260 amca.amca_amp = a_AMV_MACKY_MACK;
3261 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3263 amca.amca_pospar.app_count = (ui32_t)i - 1;
3264 amca.amca_pospar.app_not_heap = TRU1;
3265 amca.amca_pospar.app_dat =
3266 reargv = n_autorec_alloc(sizeof(char*) * i);
3267 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3268 *reargv++ = savestrbuf(&argv[1][rema[i].rm_so],
3269 rema[i].rm_eo - rema[i].rm_so);
3270 *reargv = NULL;
3272 memset(&los, 0, sizeof los);
3273 hold_all_sigs(); /* TODO DISLIKE! */
3274 los.as_global_saved = a_amv_lopts;
3275 los.as_amcap = &amca;
3276 los.as_up = los.as_global_saved;
3277 a_amv_lopts = &los;
3279 /* C99 */{
3280 struct str templ;
3281 struct n_string s_b;
3282 enum n_shexp_state shs;
3284 templ.s = n_UNCONST(argv[3]);
3285 templ.l = UIZ_MAX;
3286 shs = n_shexp_parse_token((n_SHEXP_PARSE_LOG |
3287 n_SHEXP_PARSE_IGNORE_EMPTY | n_SHEXP_PARSE_QUOTE_AUTO_FIXED |
3288 n_SHEXP_PARSE_QUOTE_AUTO_DSQ),
3289 n_string_creat_auto(&s_b), &templ, NULL);
3290 if((shs & (n_SHEXP_STATE_ERR_MASK | n_SHEXP_STATE_STOP)
3291 ) == n_SHEXP_STATE_STOP){
3292 varres = n_string_cp(&s_b);
3293 n_string_drop_ownership(&s_b);
3294 }else
3295 varres = NULL;
3298 a_amv_lopts = los.as_global_saved;
3299 rele_all_sigs(); /* TODO DISLIKE! */
3301 if(varres == NULL)
3302 goto jestr_nodata;
3303 f &= ~(a_ISNUM | a_ISDECIMAL);
3305 }else if(is_asccaseprefix(argv[0], "iregex")){
3306 f |= a_ICASE;
3307 goto Jregex;
3308 #endif /* HAVE_REGEX */
3309 }else
3310 goto jesubcmd;
3312 n_pstate_err_no = (f & a_SOFTOVERFLOW) ? n_ERR_OVERFLOW : n_ERR_NONE;
3313 f &= ~a_ERR;
3315 /* Generate the variable value content for numerics.
3316 * Anticipate in our handling below! (Don't do needless work) */
3317 jleave:
3318 if((f & a_ISNUM) && ((f & a_ISDECIMAL) || varname != NULL)){
3319 snprintf(varbuf, sizeof varbuf, "%" PRId64 , lhv);
3320 varres = varbuf;
3323 if(varname == NULL){
3324 /* If there was no error and we are printing a numeric result, print some
3325 * more bases for the fun of it */
3326 if((f & (a_ERR | a_ISNUM | a_ISDECIMAL)) == a_ISNUM){
3327 size_t j;
3329 for(j = 1, i = 0; i < 64; ++i){
3330 varbuf[63 + 64 / 8 -j - i] = (lhv & ((ui64_t)1 << i)) ? '1' : '0';
3331 if((i & 7) == 7 && i != 63){
3332 ++j;
3333 varbuf[63 + 64 / 8 -j - i] = ' ';
3336 varbuf[64 + 64 / 8 -1] = '\0';
3337 if(fprintf(n_stdout,
3338 "%s\n0%" PRIo64 " | 0x%" PRIX64 " | %" PRId64 "\n",
3339 varbuf, lhv, lhv, lhv) < 0){
3340 n_pstate_err_no = n_err_no;
3341 f |= a_ERR;
3343 }else if(varres != NULL && fprintf(n_stdout, "%s\n", varres) < 0){
3344 n_pstate_err_no = n_err_no;
3345 f |= a_ERR;
3347 }else if(!n_var_vset(varname, (uintptr_t)varres)){
3348 n_pstate_err_no = n_ERR_NOTSUP;
3349 f |= a_ERR;
3351 NYD_LEAVE;
3352 return (f & a_ERR) ? 1 : 0;
3354 jerr:
3355 f = a_ERR | a_ISNUM;
3356 lhv = -1;
3357 goto jleave;
3358 jesubcmd:
3359 n_err(_("`vexpr': invalid subcommand: %s\n"),
3360 n_shexp_quote_cp(*argv, FAL0));
3361 n_pstate_err_no = n_ERR_INVAL;
3362 goto jerr;
3363 jesynopsis:
3364 n_err(_("Synopsis: vexpr: <operator> <:argument:>\n"));
3365 n_pstate_err_no = n_ERR_INVAL;
3366 goto jerr;
3368 jenum_range:
3369 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3370 n_shexp_quote_cp(*argv, FAL0));
3371 n_pstate_err_no = n_ERR_RANGE;
3372 goto jerr;
3373 jenum_overflow:
3374 n_err(_("`vexpr': expression overflows datatype: %" PRId64 " %c %" PRId64
3375 "\n"), lhv, op, rhv);
3376 n_pstate_err_no = n_ERR_OVERFLOW;
3377 goto jerr;
3379 jestr_numrange:
3380 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3381 n_shexp_quote_cp(*argv, FAL0));
3382 n_pstate_err_no = n_ERR_RANGE;
3383 goto jestr;
3384 jestr_overflow:
3385 n_err(_("`vexpr': string length or offset overflows datatype\n"));
3386 n_pstate_err_no = n_ERR_OVERFLOW;
3387 goto jestr;
3388 jestr_nodata:
3389 n_pstate_err_no = n_ERR_NODATA;
3390 /* FALLTHRU */
3391 jestr:
3392 varres = n_empty;
3393 f &= ~a_ISNUM;
3394 f |= a_ERR;
3395 goto jleave;
3398 FL int
3399 c_vpospar(void *v){
3400 struct n_cmd_arg *cap;
3401 size_t i;
3402 struct a_amv_pospar *appp;
3403 enum{
3404 a_NONE = 0,
3405 a_ERR = 1u<<0,
3406 a_SET = 1u<<1,
3407 a_CLEAR = 1u<<2,
3408 a_QUOTE = 1u<<3
3409 } f;
3410 char const *varres;
3411 struct n_cmd_arg_ctx *cacp;
3412 NYD_ENTER;
3414 n_pstate_err_no = n_ERR_NONE;
3415 n_UNINIT(varres, n_empty);
3416 cacp = v;
3417 cap = cacp->cac_arg;
3419 if(is_asccaseprefix(cap->ca_arg.ca_str.s, "set"))
3420 f = a_SET;
3421 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "clear"))
3422 f = a_CLEAR;
3423 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "quote"))
3424 f = a_QUOTE;
3425 else{
3426 n_err(_("`vpospar': invalid subcommand: %s\n"),
3427 n_shexp_quote_cp(cap->ca_arg.ca_str.s, FAL0));
3428 n_pstate_err_no = n_ERR_INVAL;
3429 f = a_ERR;
3430 goto jleave;
3432 --cacp->cac_no;
3434 if((f & (a_CLEAR | a_QUOTE)) && cap->ca_next != NULL){
3435 n_err(_("`vpospar': `%s': takes no argument\n"), cap->ca_arg.ca_str.s);
3436 n_pstate_err_no = n_ERR_INVAL;
3437 f = a_ERR;
3438 goto jleave;
3441 cap = cap->ca_next;
3443 /* If in a macro, we need to overwrite the local instead of global argv */
3444 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
3445 : &a_amv_pospar;
3447 if(f & (a_SET | a_CLEAR)){
3448 if(cacp->cac_vput != NULL)
3449 n_err(_("`vpospar': `vput' only supported for `quote' subcommand\n"));
3450 if(!appp->app_not_heap && appp->app_maxcount > 0){
3451 for(i = appp->app_maxcount; i-- != 0;)
3452 n_free(n_UNCONST(appp->app_dat[i]));
3453 n_free(appp->app_dat);
3455 memset(appp, 0, sizeof *appp);
3457 if(f & a_SET){
3458 if((i = cacp->cac_no) > a_AMV_POSPAR_MAX){
3459 n_err(_("`vpospar': overflow: %" PRIuZ " arguments!\n"), i);
3460 n_pstate_err_no = n_ERR_OVERFLOW;
3461 f = a_ERR;
3462 goto jleave;
3465 memset(appp, 0, sizeof *appp);
3466 if(i > 0){
3467 appp->app_maxcount = appp->app_count = (ui16_t)i;
3468 /* XXX Optimize: store it all in one chunk! */
3469 ++i;
3470 i *= sizeof *appp->app_dat;
3471 appp->app_dat = n_alloc(i);
3473 for(i = 0; cap != NULL; ++i, cap = cap->ca_next){
3474 appp->app_dat[i] = n_alloc(cap->ca_arg.ca_str.l +1);
3475 memcpy(n_UNCONST(appp->app_dat[i]), cap->ca_arg.ca_str.s,
3476 cap->ca_arg.ca_str.l +1);
3479 appp->app_dat[i] = NULL;
3482 }else{
3483 if(appp->app_count == 0)
3484 varres = n_empty;
3485 else{
3486 struct str in;
3487 struct n_string s, *sp;
3488 char sep1, sep2;
3490 sp = n_string_creat_auto(&s);
3492 sep1 = *ok_vlook(ifs);
3493 sep2 = *ok_vlook(ifs_ws);
3494 if(sep1 == sep2)
3495 sep2 = '\0';
3496 if(sep1 == '\0')
3497 sep1 = ' ';
3499 for(i = 0; i < appp->app_count; ++i){
3500 if(sp->s_len){
3501 if(!n_string_can_swallow(sp, 2))
3502 goto jeover;
3503 sp = n_string_push_c(sp, sep1);
3504 if(sep2 != '\0')
3505 sp = n_string_push_c(sp, sep2);
3507 in.l = strlen(in.s = n_UNCONST(appp->app_dat[i + appp->app_idx]));
3509 if(!n_string_can_swallow(sp, in.l)){
3510 jeover:
3511 n_err(_("`vpospar': overflow: string too long!\n"));
3512 n_pstate_err_no = n_ERR_OVERFLOW;
3513 f = a_ERR;
3514 goto jleave;
3516 sp = n_shexp_quote(sp, &in, TRU1);
3519 varres = n_string_cp(sp);
3522 if(cacp->cac_vput == NULL){
3523 if(fprintf(n_stdout, "%s\n", varres) < 0){
3524 n_pstate_err_no = n_err_no;
3525 f |= a_ERR;
3527 }else if(!n_var_vset(cacp->cac_vput, (uintptr_t)varres)){
3528 n_pstate_err_no = n_ERR_NOTSUP;
3529 f |= a_ERR;
3532 jleave:
3533 NYD_LEAVE;
3534 return (f & a_ERR) ? 1 : 0;
3537 /* s-it-mode */