(BWDIC!) Drop a_amv_var_canonify(): URL normalizes hostnames to lowercase..
[s-mailx.git] / accmacvar.c
blob97451b571a330a530e35fc9becdd25ab7b252b81
1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
2 *@ Account, macro and variable handling.
3 *@ HOWTO add a new non-dynamic boolean or value option:
4 *@ - add an entry to nail.h:enum okeys
5 *@ - run make-okey-map.pl
6 *@ - update the manual!
7 *@ TODO . Improve support for chains so that we can apply the property checks
8 *@ TODO of the base variable to -HOST and -USER@HOST variants! Add some!!
9 *@ TODO Now with `local' modifier we need to be able to catch that, as well
10 *@ TODO as linked environment variables: both of which are not catched and
11 *@ TODO can be shadowed, which is documented for `set' - change on change!!
12 *@ TODO . `localopts' should act like an automatic permanent `scope' command
13 *@ TODO modifier! We need an OnScopeLeaveEvent, then.
14 *@ TODO Also see the a_GO_SPLICE comment in go.c.
15 *@ TODO . Optimize: with the dynamic hashmaps, and the object based approach
16 *@ TODO it should become possible to strip down the implementation again.
17 *@ TODO E.g., FREEZE is much too complicated: use an overlay object ptr,
18 *@ TODO UNLIKELY() it, and add a OnProgramStartupCompletedEvent to
19 *@ TODO incorporate what it tracks, then drop it. Etc.
20 *@ TODO Global -> Scope -> Local, all "overlay" objects.
22 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
23 * Copyright (c) 2012 - 2018 Steffen (Daode) Nurpmeso <steffen@sdaoden.eu>.
26 * Copyright (c) 1980, 1993
27 * The Regents of the University of California. All rights reserved.
29 * Redistribution and use in source and binary forms, with or without
30 * modification, are permitted provided that the following conditions
31 * are met:
32 * 1. Redistributions of source code must retain the above copyright
33 * notice, this list of conditions and the following disclaimer.
34 * 2. Redistributions in binary form must reproduce the above copyright
35 * notice, this list of conditions and the following disclaimer in the
36 * documentation and/or other materials provided with the distribution.
37 * 3. Neither the name of the University nor the names of its contributors
38 * may be used to endorse or promote products derived from this software
39 * without specific prior written permission.
41 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
42 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
45 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51 * SUCH DAMAGE.
53 #undef n_FILE
54 #define n_FILE accmacvar
56 #ifndef HAVE_AMALGAMATION
57 # include "nail.h"
58 #endif
60 #if !defined HAVE_SETENV && !defined HAVE_PUTENV
61 # error Exactly one of HAVE_SETENV and HAVE_PUTENV
62 #endif
64 /* Positional parameter maximum (macro arguments, `vexpr' "splitifs") */
65 #define a_AMV_POSPAR_MAX SI16_MAX
67 /* Special "pseudo macro" that stabs you from the back */
68 #define a_AMV_MACKY_MACK ((struct a_amv_mac*)-1)
70 /* Note: changing the hash function must be reflected in `vexpr' "hash",
71 * because that is used by the hashtable creator scripts! */
72 #define a_AMV_PRIME HSHSIZE
73 #define a_AMV_NAME2HASH(N) n_torek_hash(N)
74 #define a_AMV_HASH2PRIME(H) ((H) % a_AMV_PRIME)
76 enum a_amv_mac_flags{
77 a_AMV_MF_NONE = 0,
78 a_AMV_MF_ACCOUNT = 1u<<0, /* This macro is an `account' */
79 a_AMV_MF_TYPE_MASK = a_AMV_MF_ACCOUNT,
80 a_AMV_MF_UNDEF = 1u<<1, /* Unlink after lookup */
81 a_AMV_MF_DELETE = 1u<<7, /* Delete in progress, free once refcnt==0 */
82 a_AMV_MF__MAX = 0xFFu
85 enum a_amv_loflags{
86 a_AMV_LF_NONE = 0,
87 a_AMV_LF_SCOPE = 1u<<0, /* Current scope `localopts' on */
88 a_AMV_LF_SCOPE_FIXATE = 1u<<1, /* Ditto, but fixated */
89 a_AMV_LF_SCOPE_MASK = a_AMV_LF_SCOPE | a_AMV_LF_SCOPE_FIXATE,
90 a_AMV_LF_CALL = 1u<<2, /* `localopts' on for `call'ed scopes */
91 a_AMV_LF_CALL_FIXATE = 1u<<3, /* Ditto, but fixated */
92 a_AMV_LF_CALL_MASK = a_AMV_LF_CALL | a_AMV_LF_CALL_FIXATE,
93 a_AMV_LF_CALL_TO_SCOPE_SHIFT = 2
96 /* make-okey-map.pl ensures that _VIRT implies _RDONLY and _NODEL, and that
97 * _IMPORT implies _ENV; it doesn't verify anything...
98 * More description at nail.h:enum okeys */
99 enum a_amv_var_flags{
100 a_AMV_VF_NONE = 0,
102 /* The basic set of flags, also present in struct a_amv_var_map.avm_flags */
103 a_AMV_VF_BOOL = 1u<<0, /* ok_b_* */
104 a_AMV_VF_CHAIN = 1u<<1, /* Is variable chain (-USER{,@HOST} variants) */
105 a_AMV_VF_VIRT = 1u<<2, /* "Stateless" automatic variable */
106 a_AMV_VF_VIP = 1u<<3, /* Wants _var_check_vips() evaluation */
107 a_AMV_VF_RDONLY = 1u<<4, /* May not be set by user */
108 a_AMV_VF_NODEL = 1u<<5, /* May not be deleted */
109 a_AMV_VF_I3VAL = 1u<<6, /* Has an initial value */
110 a_AMV_VF_DEFVAL = 1u<<7, /* Has a default value */
111 a_AMV_VF_IMPORT = 1u<<8, /* Import ONLY from env (pre n_PSO_STARTED) */
112 a_AMV_VF_ENV = 1u<<9, /* Update environment on change */
113 a_AMV_VF_NOLOPTS = 1u<<10, /* May not be tracked by `localopts' */
114 a_AMV_VF_NOTEMPTY = 1u<<11, /* May not be assigned an empty value */
115 /* TODO _VF_NUM, _VF_POSNUM: we also need 64-bit limit numbers! */
116 a_AMV_VF_NUM = 1u<<12, /* Value must be a 32-bit number */
117 a_AMV_VF_POSNUM = 1u<<13, /* Value must be positive 32-bit number */
118 a_AMV_VF_LOWER = 1u<<14, /* Values will be stored in lowercase version */
119 a_AMV_VF_OBSOLETE = 1u<<15, /* Is obsolete? */
120 a_AMV_VF__MASK = (1u<<(15+1)) - 1,
122 /* Extended flags, not part of struct a_amv_var_map.avm_flags */
123 a_AMV_VF_EXT_LOCAL = 1u<<23, /* `local' */
124 a_AMV_VF_EXT_LINKED = 1u<<24, /* `environ' link'ed */
125 a_AMV_VF_EXT_FROZEN = 1u<<25, /* Has been set by -S,.. */
126 a_AMV_VF_EXT_FROZEN_UNSET = 1u<<26, /* ..and was used to unset a variable */
127 a_AMV_VF_EXT__FROZEN_MASK = a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET,
128 a_AMV_VF_EXT__MASK = (1u<<(26+1)) - 1
131 enum a_amv_var_lookup_flags{
132 a_AMV_VLOOK_NONE = 0,
133 a_AMV_VLOOK_LOCAL = 1u<<0, /* Query `local' layer first */
134 a_AMV_VLOOK_LOCAL_ONLY = 1u<<1, /* MUST be a `local' variable */
135 /* Do not allocate new var for _I3VAL, see _var_lookup() for more */
136 a_AMV_VLOOK_I3VAL_NONEW = 1u<<2,
137 a_AMV_VLOOK_I3VAL_NONEW_REPORT = 1u<<3
140 enum a_amv_var_setclr_flags{
141 a_AMV_VSETCLR_NONE = 0,
142 a_AMV_VSETCLR_LOCAL = 1u<<0, /* Use `local' variables only */
143 a_AMV_VSETCLR_ENV = 1u<<1 /* `environ' or otherwise environ */
146 /* We support some special parameter names for one(+)-letter variable names;
147 * note these have counterparts in the code that manages shell expansion!
148 * All these special variables are solely backed by n_var_vlook(), and besides
149 * there is only a_amv_var_revlookup() which knows about them */
150 enum a_amv_var_special_category{
151 a_AMV_VSC_NONE, /* Normal variable, no special treatment */
152 a_AMV_VSC_GLOBAL, /* ${[?!]} are specially mapped, but global */
153 a_AMV_VSC_MULTIPLEX, /* ${^.*} circumflex accent multiplexer */
154 a_AMV_VSC_POSPAR, /* ${[1-9][0-9]*} positional parameters */
155 a_AMV_VSC_POSPAR_ENV /* ${[*@#]} positional parameter support variables */
158 enum a_amv_var_special_type{
159 /* _VSC_GLOBAL */
160 a_AMV_VST_QM, /* ? */
161 a_AMV_VST_EM, /* ! */
162 /* _VSC_MULTIPLEX */
163 /* This is special in that it is a multiplex indicator, the ^ is followed by
164 * a normal variable */
165 a_AMV_VST_CACC, /* ^ (circumflex accent) */
166 /* _VSC_POSPAR_ENV */
167 a_AMV_VST_STAR, /* * */
168 a_AMV_VST_AT, /* @ */
169 a_AMV_VST_NOSIGN /* # */
172 enum a_amv_var_vip_mode{
173 a_AMV_VIP_SET_PRE,
174 a_AMV_VIP_SET_POST,
175 a_AMV_VIP_CLEAR
178 struct a_amv_pospar{
179 ui16_t app_maxcount; /* == slots in .app_dat */
180 ui16_t app_count; /* Maximum a_AMV_POSPAR_MAX */
181 ui16_t app_idx; /* `shift' moves this one, decs .app_count */
182 bool_t app_not_heap; /* .app_dat stuff not dynamically allocated */
183 ui8_t app__dummy[1];
184 char const **app_dat; /* NULL terminated (for "$@" explosion support) */
186 n_CTA(a_AMV_POSPAR_MAX <= SI16_MAX, "Limit exceeds datatype capabilities");
188 struct a_amv_mac{
189 struct a_amv_mac *am_next;
190 ui32_t am_maxlen; /* of any line in .am_line_dat */
191 ui32_t am_line_cnt; /* of *.am_line_dat (but NULL terminated) */
192 struct a_amv_mac_line **am_line_dat; /* TODO use deque? */
193 struct a_amv_var *am_lopts; /* `localopts' unroll list */
194 ui32_t am_refcnt; /* 0-based for `un{account,define}' purposes */
195 ui8_t am_flags; /* enum a_amv_mac_flags */
196 char am_name[n_VFIELD_SIZE(3)]; /* of this macro */
198 n_CTA(a_AMV_MF__MAX <= UI8_MAX, "Enumeration excesses storage datatype");
200 struct a_amv_mac_line{
201 ui32_t aml_len;
202 ui32_t aml_prespc; /* Number of leading SPACEs, for display purposes */
203 char aml_dat[n_VFIELD_SIZE(0)];
206 struct a_amv_mac_call_args{
207 char const *amca_name; /* For MACKY_MACK, this is *0*! */
208 struct a_amv_mac *amca_amp; /* "const", but for am_refcnt */
209 struct a_amv_var **amca_unroller;
210 void (*amca_hook_pre)(void *);
211 void *amca_hook_arg;
212 ui8_t amca_loflags;
213 bool_t amca_ps_hook_mask;
214 bool_t amca_no_xcall; /* We want n_GO_INPUT_NO_XCALL for this */
215 ui8_t amca__pad[5];
216 struct a_amv_var *(*amca_local_vars)[a_AMV_PRIME]; /* `local's, or NULL */
217 struct a_amv_pospar amca_pospar;
220 struct a_amv_lostack{
221 struct a_amv_lostack *as_global_saved; /* Saved global XXX due to jump */
222 struct a_amv_mac_call_args *as_amcap;
223 struct a_amv_lostack *as_up; /* Outer context */
224 struct a_amv_var *as_lopts;
225 ui8_t as_loflags; /* enum a_amv_mac_loflags */
226 ui8_t avs__pad[7];
229 struct a_amv_var{
230 struct a_amv_var *av_link;
231 char *av_value;
232 #ifdef HAVE_PUTENV
233 char *av_env; /* Actively managed putenv(3) memory, or NULL */
234 #endif
235 ui32_t av_flags; /* enum a_amv_var_flags inclusive extended bits */
236 char av_name[n_VFIELD_SIZE(4)];
238 n_CTA(a_AMV_VF_EXT__MASK <= UI32_MAX, "Enumeration excesses storage datatype");
240 struct a_amv_var_map{
241 ui32_t avm_hash;
242 ui16_t avm_keyoff;
243 ui16_t avm_flags; /* enum a_amv_var_flags without extended bits */
245 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
247 struct a_amv_var_virt{
248 ui32_t avv_okey;
249 ui8_t avv__dummy[4];
250 struct a_amv_var const *avv_var;
253 struct a_amv_var_defval{
254 ui32_t avdv_okey;
255 ui8_t avdv__pad[4];
256 char const *avdv_value; /* Only for !BOOL (otherwise plain existence) */
259 struct a_amv_var_carrier{
260 char const *avc_name;
261 ui32_t avc_hash;
262 ui32_t avc_prime;
263 struct a_amv_var *avc_var;
264 struct a_amv_var_map const *avc_map;
265 enum okeys avc_okey;
266 ui8_t avc__pad[1];
267 ui8_t avc_special_cat;
268 /* Numerical parameter name if .avc_special_cat=a_AMV_VSC_POSPAR,
269 * otherwise a enum a_amv_var_special_type */
270 ui16_t avc_special_prop;
273 /* Include constant make-okey-map.pl output, and the generated version data */
274 #include <gen-version.h>
275 #include <gen-okeys.h>
277 /* The currently active account */
278 static struct a_amv_mac *a_amv_acc_curr;
280 static struct a_amv_mac *a_amv_macs[a_AMV_PRIME]; /* TODO dynamically spaced */
282 /* Unroll list of currently running macro stack */
283 static struct a_amv_lostack *a_amv_lopts;
285 static struct a_amv_var *a_amv_vars[a_AMV_PRIME]; /* TODO dynamically spaced */
287 /* Global (i.e., non-local) a_AMV_VSC_POSPAR stack */
288 static struct a_amv_pospar a_amv_pospar;
290 /* TODO We really deserve localopts support for *folder-hook*s, so hack it in
291 * TODO today via a static lostack, it should be a field in mailbox, once that
292 * TODO is a real multi-instance object */
293 static struct a_amv_var *a_amv_folder_hook_lopts;
295 /* TODO Rather ditto (except for storage -> cmd_ctx), compose hooks */
296 static struct a_amv_var *a_amv_compose_lopts;
298 /* Lookup for macros/accounts: if newamp is not NULL it will be linked in the
299 * map, if _MF_UNDEF is set a possibly existing entry will be removed (first).
300 * Returns NULL if a lookup failed, or if newamp was set, the found entry in
301 * plain lookup cases or when _UNDEF was performed on a currently active entry
302 * (the entry will have been unlinked, and the _MF_DELETE will be honoured once
303 * the reference count reaches 0), and (*)-1 if an _UNDEF was performed */
304 static struct a_amv_mac *a_amv_mac_lookup(char const *name,
305 struct a_amv_mac *newamp, enum a_amv_mac_flags amf);
307 /* `call', `call_if' (and `xcall' via go.c -> c_call()) */
308 static int a_amv_mac_call(void *v, bool_t silent_nexist);
310 /* Execute a macro; amcap must reside in LOFI memory */
311 static bool_t a_amv_mac_exec(struct a_amv_mac_call_args *amcap);
313 static void a_amv_mac__finalize(void *vp);
315 /* User display helpers */
316 static bool_t a_amv_mac_show(enum a_amv_mac_flags amf);
318 /* _def() returns error for faulty definitions and already existing * names,
319 * _undef() returns error if a named thing doesn't exist */
320 static bool_t a_amv_mac_def(char const *name, enum a_amv_mac_flags amf);
321 static bool_t a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf);
323 /* */
324 static void a_amv_mac_free(struct a_amv_mac *amp);
326 /* Update replay-log */
327 static void a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
328 struct a_amv_var *oavp);
329 static void a_amv_lopts_unroll(struct a_amv_var **avpp);
331 /* Special cased value string allocation */
332 static char *a_amv_var_copy(char const *str);
333 static void a_amv_var_free(char *cp);
335 /* Check for special housekeeping. _VIP_SET_POST and _VIP_CLEAR do not fail
336 * (or propagate errors), _VIP_SET_PRE may and should case abortion */
337 static bool_t a_amv_var_check_vips(enum a_amv_var_vip_mode avvm,
338 enum okeys okey, char const **val);
340 /* _VF_NUM / _VF_POSNUM */
341 static bool_t a_amv_var_check_num(char const *val, bool_t posnum);
343 /* Try to reverse lookup an option name to an enum okeys mapping.
344 * Updates .avc_name and .avc_hash; .avc_map is NULL if none found */
345 static bool_t a_amv_var_revlookup(struct a_amv_var_carrier *avcp,
346 char const *name);
348 /* Lookup a variable from .avc_(map|name|hash), return whether it was found.
349 * Sets .avc_prime; .avc_var is NULL if not found.
350 * Here it is where we care for _I3VAL and _DEFVAL.
351 * An _I3VAL will be "consumed" as necessary anyway, but it won't be used to
352 * create a new variable if _VLOOK_I3VAL_NONEW is set; if
353 * _VLOOK_I3VAL_NONEW_REPORT is set then we set .avc_var to -1 and return true
354 * if that was the case, otherwise we'll return FAL0, then! */
355 static bool_t a_amv_var_lookup(struct a_amv_var_carrier *avcp,
356 enum a_amv_var_lookup_flags avlf);
358 /* Lookup functions for special category variables, _pospar drives all
359 * positional parameter etc. special categories */
360 static char const *a_amv_var_vsc_global(struct a_amv_var_carrier *avcp);
361 static char const *a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp);
362 static char const *a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp);
364 /* Set var from .avc_(map|name|hash), return success */
365 static bool_t a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
366 enum a_amv_var_setclr_flags avscf);
368 static bool_t a_amv_var__putenv(struct a_amv_var_carrier *avcp,
369 struct a_amv_var *avp);
371 /* Clear var from .avc_(map|name|hash); sets .avc_var=NULL, return success */
372 static bool_t a_amv_var_clear(struct a_amv_var_carrier *avcp,
373 enum a_amv_var_setclr_flags avscf);
375 static bool_t a_amv_var__clearenv(char const *name, struct a_amv_var *avp);
377 /* List all variables */
378 static void a_amv_var_show_all(void);
380 static int a_amv_var__show_cmp(void const *s1, void const *s2);
382 /* Actually do print one, return number of lines written */
383 static size_t a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp);
385 /* Shared c_set() and c_environ():set impl, return success */
386 static bool_t a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf);
388 static struct a_amv_mac *
389 a_amv_mac_lookup(char const *name, struct a_amv_mac *newamp,
390 enum a_amv_mac_flags amf){
391 struct a_amv_mac *amp, **ampp;
392 ui32_t h;
393 enum a_amv_mac_flags save_amf;
394 NYD2_ENTER;
396 save_amf = amf;
397 amf &= a_AMV_MF_TYPE_MASK;
398 h = a_AMV_NAME2HASH(name);
399 h = a_AMV_HASH2PRIME(h);
400 ampp = &a_amv_macs[h];
402 for(amp = *ampp; amp != NULL; ampp = &(*ampp)->am_next, amp = amp->am_next){
403 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf &&
404 !strcmp(amp->am_name, name)){
405 if(n_LIKELY((save_amf & a_AMV_MF_UNDEF) == 0))
406 goto jleave;
408 *ampp = amp->am_next;
410 if(amp->am_refcnt > 0){
411 amp->am_flags |= a_AMV_MF_DELETE;
412 if(n_poption & n_PO_D_V)
413 n_err(_("Delayed deletion of currently active %s: %s\n"),
414 (amp->am_flags & a_AMV_MF_ACCOUNT ? "account" : "define"),
415 name);
416 }else{
417 a_amv_mac_free(amp);
418 amp = (struct a_amv_mac*)-1;
420 break;
424 if(newamp != NULL){
425 ampp = &a_amv_macs[h];
426 newamp->am_next = *ampp;
427 *ampp = newamp;
428 amp = NULL;
430 jleave:
431 NYD2_LEAVE;
432 return amp;
435 static int
436 a_amv_mac_call(void *v, bool_t silent_nexist){
437 struct a_amv_mac *amp;
438 int rv;
439 char const *name;
440 struct n_cmd_arg_ctx *cacp;
441 NYD_ENTER;
443 cacp = v;
445 if(cacp->cac_no == 0){
446 n_err(_("Synopsis: call(_if)?: name [:<arguments>:]\n"));
447 n_pstate_err_no = n_ERR_INVAL;
448 rv = 1;
449 goto jleave;
452 name = cacp->cac_arg->ca_arg.ca_str.s;
454 if(n_UNLIKELY(cacp->cac_no > a_AMV_POSPAR_MAX)){
455 n_err(_("Too many arguments to macro `call': %s\n"), name);
456 n_pstate_err_no = n_ERR_OVERFLOW;
457 rv = 1;
458 }else if(n_UNLIKELY((amp = a_amv_mac_lookup(name, NULL, a_AMV_MF_NONE)
459 ) == NULL)){
460 if(!silent_nexist)
461 n_err(_("Undefined macro called: %s\n"), n_shexp_quote_cp(name, FAL0));
462 n_pstate_err_no = n_ERR_NOENT;
463 rv = 1;
464 }else{
465 char const **argv;
466 struct a_amv_mac_call_args *amcap;
467 size_t argc;
469 argc = cacp->cac_no + 1;
470 amcap = n_lofi_alloc(sizeof *amcap + (argc * sizeof *argv));
471 argv = (void*)&amcap[1];
473 for(argc = 0; (cacp->cac_arg = cacp->cac_arg->ca_next) != NULL; ++argc)
474 argv[argc] = cacp->cac_arg->ca_arg.ca_str.s;
475 argv[argc] = NULL;
477 memset(amcap, 0, sizeof *amcap);
478 amcap->amca_name = name;
479 amcap->amca_amp = amp;
480 if(a_amv_lopts != NULL)
481 amcap->amca_loflags = (a_amv_lopts->as_loflags & a_AMV_LF_CALL_MASK
482 ) >> a_AMV_LF_CALL_TO_SCOPE_SHIFT;
483 if(argc > 0){
484 amcap->amca_pospar.app_count = (ui16_t)argc;
485 amcap->amca_pospar.app_not_heap = TRU1;
486 amcap->amca_pospar.app_dat = argv;
489 (void)a_amv_mac_exec(amcap);
490 rv = n_pstate_ex_no;
492 jleave:
493 NYD_LEAVE;
494 return rv;
497 static bool_t
498 a_amv_mac_exec(struct a_amv_mac_call_args *amcap){
499 struct a_amv_lostack *losp;
500 struct a_amv_mac_line **amlp;
501 char **args_base, **args;
502 struct a_amv_mac *amp;
503 bool_t rv;
504 NYD2_ENTER;
506 amp = amcap->amca_amp;
507 assert(amp != NULL && amp != a_AMV_MACKY_MACK);
508 ++amp->am_refcnt;
509 /* XXX Unfortunately we yet need to dup the macro lines! :( */
510 args_base = args = n_alloc(sizeof(*args) * (amp->am_line_cnt +1));
511 for(amlp = amp->am_line_dat; *amlp != NULL; ++amlp)
512 *(args++) = sbufdup((*amlp)->aml_dat, (*amlp)->aml_len);
513 *args = NULL;
515 losp = n_lofi_alloc(sizeof *losp);
516 losp->as_global_saved = a_amv_lopts;
517 if((losp->as_amcap = amcap)->amca_unroller == NULL){
518 losp->as_up = losp->as_global_saved;
519 losp->as_lopts = NULL;
520 }else{
521 losp->as_up = NULL;
522 losp->as_lopts = *amcap->amca_unroller;
524 losp->as_loflags = amcap->amca_loflags;
526 a_amv_lopts = losp;
527 if(amcap->amca_hook_pre != NULL)
528 n_PS_ROOT_BLOCK((*amcap->amca_hook_pre)(amcap->amca_hook_arg));
529 rv = n_go_macro((n_GO_INPUT_NONE |
530 (amcap->amca_no_xcall ? n_GO_INPUT_NO_XCALL : 0)),
531 amp->am_name, args_base, &a_amv_mac__finalize, losp);
532 NYD2_LEAVE;
533 return rv;
536 static void
537 a_amv_mac__finalize(void *vp){
538 struct a_amv_mac *amp;
539 struct a_amv_mac_call_args *amcap;
540 struct a_amv_lostack *losp;
541 NYD2_ENTER;
543 losp = vp;
544 a_amv_lopts = losp->as_global_saved;
546 amcap = losp->as_amcap;
548 /* Delete positional parameter stack */
549 if(!amcap->amca_pospar.app_not_heap && amcap->amca_pospar.app_maxcount > 0){
550 ui16_t i;
552 for(i = amcap->amca_pospar.app_maxcount; i-- != 0;)
553 n_free(n_UNCONST(amcap->amca_pospar.app_dat[i]));
554 n_free(amcap->amca_pospar.app_dat);
557 /* `local' variable hashmap. These have no environment map, never */
558 if(amcap->amca_local_vars != NULL){
559 struct a_amv_var **avpp_base, **avpp, *avp;
561 for(avpp_base = *amcap->amca_local_vars, avpp = &avpp_base[a_AMV_PRIME];
562 avpp-- != avpp_base;)
563 while((avp = *avpp)){
564 assert((avp->av_flags & (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL)) ==
565 (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL));
566 assert(!(avp->av_flags &
567 ((a_AMV_VF__MASK | a_AMV_VF_EXT__MASK) &
568 ~(a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL))));
569 *avpp = avp->av_link;
570 a_amv_var_free(avp->av_value);
571 n_free(avp);
573 n_free(avpp_base);
576 /* Unroll `localopts', if applicable */
577 if(amcap->amca_unroller == NULL){
578 if(losp->as_lopts != NULL)
579 a_amv_lopts_unroll(&losp->as_lopts);
580 }else
581 *amcap->amca_unroller = losp->as_lopts;
583 if(amcap->amca_ps_hook_mask)
584 n_pstate &= ~n_PS_HOOK_MASK;
586 if((amp = amcap->amca_amp) != a_AMV_MACKY_MACK && amp != NULL &&
587 --amp->am_refcnt == 0 && (amp->am_flags & a_AMV_MF_DELETE))
588 a_amv_mac_free(amp);
590 n_lofi_free(losp);
591 n_lofi_free(amcap);
592 NYD2_LEAVE;
595 static bool_t
596 a_amv_mac_show(enum a_amv_mac_flags amf){
597 size_t lc, mc, ti, i;
598 char const *typestr;
599 FILE *fp;
600 bool_t rv;
601 NYD2_ENTER;
603 rv = FAL0;
605 if((fp = Ftmp(NULL, "deflist", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
606 NULL){
607 n_perr(_("`define' or `account' list: cannot create temporary file"), 0);
608 goto jleave;
611 amf &= a_AMV_MF_TYPE_MASK;
612 typestr = (amf & a_AMV_MF_ACCOUNT) ? "account" : "define";
614 for(lc = mc = ti = 0; ti < a_AMV_PRIME; ++ti){
615 struct a_amv_mac *amp;
617 for(amp = a_amv_macs[ti]; amp != NULL; amp = amp->am_next){
618 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
619 struct a_amv_mac_line **amlpp;
621 if(++mc > 1){
622 putc('\n', fp);
623 ++lc;
625 ++lc;
626 fprintf(fp, "%s %s {\n", typestr, amp->am_name);
627 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++lc, ++amlpp){
628 for(i = (*amlpp)->aml_prespc; i > 0; --i)
629 putc(' ', fp);
630 fputs((*amlpp)->aml_dat, fp);
631 putc('\n', fp);
633 fputs("}\n", fp);
634 ++lc;
638 if(mc > 0)
639 page_or_print(fp, lc);
641 rv = (ferror(fp) == 0);
642 Fclose(fp);
643 jleave:
644 NYD2_LEAVE;
645 return rv;
648 static bool_t
649 a_amv_mac_def(char const *name, enum a_amv_mac_flags amf){
650 struct str line;
651 ui32_t line_cnt, maxlen;
652 struct linelist{
653 struct linelist *ll_next;
654 struct a_amv_mac_line *ll_amlp;
655 } *llp, *ll_head, *ll_tail;
656 union {size_t s; int i; ui32_t ui; size_t l;} n;
657 struct a_amv_mac *amp;
658 bool_t rv;
659 NYD2_ENTER;
661 memset(&line, 0, sizeof line);
662 rv = FAL0;
663 amp = NULL;
665 /* TODO We should have our input state machine which emits Line events,
666 * TODO and hook different consumers dependent on our content, as stated
667 * TODO in i think lex_input: like this local macros etc. would become
668 * TODO possible from the input side */
669 /* Read in the lines which form the macro content */
670 for(ll_tail = ll_head = NULL, line_cnt = maxlen = 0;;){
671 ui32_t leaspc;
672 char *cp;
674 n.i = n_go_input(n_GO_INPUT_CTX_DEFAULT | n_GO_INPUT_NL_ESC, n_empty,
675 &line.s, &line.l, NULL, NULL);
676 if(n.ui == 0)
677 continue;
678 if(n.i < 0){
679 n_err(_("Unterminated %s definition: %s\n"),
680 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
681 goto jerr;
684 /* Trim WS, remember amount of leading spaces for display purposes */
685 for(cp = line.s, leaspc = 0; n.ui > 0; ++cp, --n.ui)
686 if(*cp == '\t')
687 leaspc = (leaspc + 8u) & ~7u;
688 else if(*cp == ' ')
689 ++leaspc;
690 else
691 break;
692 for(; n.ui > 0 && spacechar(cp[n.ui - 1]); --n.ui)
694 if(n.ui == 0)
695 continue;
697 maxlen = n_MAX(maxlen, n.ui);
698 cp[n.ui++] = '\0';
700 /* Is is the closing brace? */
701 if(*cp == '}')
702 break;
704 if(n_LIKELY(++line_cnt < UI32_MAX)){
705 struct a_amv_mac_line *amlp;
707 llp = n_autorec_alloc(sizeof *llp);
708 if(ll_head == NULL)
709 ll_head = llp;
710 else
711 ll_tail->ll_next = llp;
712 ll_tail = llp;
713 llp->ll_next = NULL;
714 llp->ll_amlp = amlp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac_line,
715 aml_dat) + n.ui);
716 amlp->aml_len = n.ui -1;
717 amlp->aml_prespc = leaspc;
718 memcpy(amlp->aml_dat, cp, n.ui);
719 }else{
720 n_err(_("Too much content in %s definition: %s\n"),
721 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
722 goto jerr;
726 /* Create the new macro */
727 n.s = strlen(name) +1;
728 amp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name) + n.s);
729 memset(amp, 0, n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name));
730 amp->am_maxlen = maxlen;
731 amp->am_line_cnt = line_cnt;
732 amp->am_flags = amf;
733 memcpy(amp->am_name, name, n.s);
734 /* C99 */{
735 struct a_amv_mac_line **amlpp;
737 amp->am_line_dat = amlpp = n_alloc(sizeof(*amlpp) * ++line_cnt);
738 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
739 *amlpp++ = llp->ll_amlp;
740 *amlpp = NULL;
743 /* Create entry, replace a yet existing one as necessary */
744 a_amv_mac_lookup(name, amp, amf | a_AMV_MF_UNDEF);
745 rv = TRU1;
746 jleave:
747 if(line.s != NULL)
748 n_free(line.s);
749 NYD2_LEAVE;
750 return rv;
752 jerr:
753 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
754 n_free(llp->ll_amlp);
756 * if(amp != NULL){
757 * n_free(amp->am_line_dat);
758 * n_free(amp);
759 *}*/
760 goto jleave;
763 static bool_t
764 a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf){
765 struct a_amv_mac *amp;
766 bool_t rv;
767 NYD2_ENTER;
769 rv = TRU1;
771 if(n_LIKELY(name[0] != '*' || name[1] != '\0')){
772 if((amp = a_amv_mac_lookup(name, NULL, amf | a_AMV_MF_UNDEF)) == NULL){
773 n_err(_("%s not defined: %s\n"),
774 (amf & a_AMV_MF_ACCOUNT ? "Account" : "Macro"), name);
775 rv = FAL0;
777 }else{
778 struct a_amv_mac **ampp, *lamp;
780 for(ampp = a_amv_macs; PTRCMP(ampp, <, &a_amv_macs[n_NELEM(a_amv_macs)]);
781 ++ampp)
782 for(lamp = NULL, amp = *ampp; amp != NULL;){
783 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
784 /* xxx Expensive but rare: be simple */
785 a_amv_mac_lookup(amp->am_name, NULL, amf | a_AMV_MF_UNDEF);
786 amp = (lamp == NULL) ? *ampp : lamp->am_next;
787 }else{
788 lamp = amp;
789 amp = amp->am_next;
793 NYD2_LEAVE;
794 return rv;
797 static void
798 a_amv_mac_free(struct a_amv_mac *amp){
799 struct a_amv_mac_line **amlpp;
800 NYD2_ENTER;
802 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++amlpp)
803 n_free(*amlpp);
804 n_free(amp->am_line_dat);
805 n_free(amp);
806 NYD2_LEAVE;
809 static void
810 a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
811 struct a_amv_var *oavp){
812 struct a_amv_var *avp;
813 size_t nl, vl;
814 NYD2_ENTER;
816 /* Propagate unrolling up the stack, as necessary */
817 assert(alp != NULL);
818 for(;;){
819 if(alp->as_loflags & a_AMV_LF_SCOPE_MASK)
820 break;
821 if((alp = alp->as_up) == NULL)
822 goto jleave;
825 /* Check whether this variable is handled yet XXX Boost: ID check etc.!! */
826 for(avp = alp->as_lopts; avp != NULL; avp = avp->av_link)
827 if(!strcmp(avp->av_name, name))
828 goto jleave;
830 nl = strlen(name) +1;
831 vl = (oavp != NULL) ? strlen(oavp->av_value) +1 : 0;
832 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + nl + vl);
833 avp->av_link = alp->as_lopts;
834 alp->as_lopts = avp;
835 if(vl != 0){
836 avp->av_value = &avp->av_name[nl];
837 avp->av_flags = oavp->av_flags;
838 memcpy(avp->av_value, oavp->av_value, vl);
840 memcpy(avp->av_name, name, nl);
841 jleave:
842 NYD2_LEAVE;
845 static void
846 a_amv_lopts_unroll(struct a_amv_var **avpp){
847 struct a_amv_lostack *save_alp;
848 struct a_amv_var *x, *avp;
849 NYD2_ENTER;
851 avp = *avpp;
852 *avpp = NULL;
854 save_alp = a_amv_lopts;
855 a_amv_lopts = NULL;
856 while(avp != NULL){
857 x = avp;
858 avp = avp->av_link;
859 n_PS_ROOT_BLOCK(n_var_vset(x->av_name, (uintptr_t)x->av_value));
860 n_free(x);
862 a_amv_lopts = save_alp;
863 NYD2_LEAVE;
866 static char *
867 a_amv_var_copy(char const *str){
868 char *news;
869 size_t len;
870 NYD2_ENTER;
872 if(*str == '\0')
873 news = n_UNCONST(n_empty);
874 else if(str[1] == '\0'){
875 if(str[0] == '1')
876 news = n_UNCONST(n_1);
877 else if(str[0] == '0')
878 news = n_UNCONST(n_0);
879 else
880 goto jheap;
881 }else if(str[2] == '\0' && str[0] == '-' && str[1] == '1')
882 news = n_UNCONST(n_m1);
883 else{
884 jheap:
885 len = strlen(str) +1;
886 news = n_alloc(len);
887 memcpy(news, str, len);
889 NYD2_LEAVE;
890 return news;
893 static void
894 a_amv_var_free(char *cp){
895 NYD2_ENTER;
896 if(cp[0] != '\0' && cp != n_0 && cp != n_1 && cp != n_m1)
897 n_free(cp);
898 NYD2_LEAVE;
901 static bool_t
902 a_amv_var_check_vips(enum a_amv_var_vip_mode avvm, enum okeys okey,
903 char const **val){
904 char const *emsg;
905 bool_t ok;
906 NYD2_ENTER;
908 ok = TRU1;
910 if(avvm == a_AMV_VIP_SET_PRE){
911 switch(okey){
912 default:
913 break;
914 case ok_v_charset_7bit:
915 case ok_v_charset_8bit:
916 case ok_v_charset_unknown_8bit:
917 case ok_v_ttycharset:
918 if((*val = n_iconv_normalize_name(*val)) == NULL)
919 ok = FAL0;
920 break;
921 case ok_v_customhdr:{
922 char const *vp;
923 char *buf;
924 struct n_header_field *hflp, **hflpp, *hfp;
925 NYD_ENTER;
927 buf = savestr(*val);
928 hflp = NULL;
929 hflpp = &hflp;
931 while((vp = n_strsep_esc(&buf, ',', TRU1)) != NULL){
932 if(!n_header_add_custom(hflpp, vp, TRU1)){
933 emsg = N_("Invalid *customhdr* entry: %s\n");
934 ok = FAL0;
935 break;
937 hflpp = &(*hflpp)->hf_next;
940 hflpp = ok ? &n_customhdr_list : &hflp;
941 while((hfp = *hflpp) != NULL){
942 *hflpp = hfp->hf_next;
943 n_free(hfp);
945 if(!ok)
946 goto jerr;
947 n_customhdr_list = hflp;
948 } break;
949 case ok_v_from:
950 case ok_v_sender:{
951 struct name *np;
953 if((np = lextract(*val, GEXTRA | GFULL)) == NULL){
954 jefrom:
955 emsg = N_("*from* / *sender*: invalid address(es): %s\n");
956 goto jerr;
957 }else if(okey == ok_v_sender && np->n_flink != NULL){
958 emsg = N_("*sender*: may not contain multiple addresses: %s\n");
959 goto jerr;
960 }else for(; np != NULL; np = np->n_flink)
961 if(is_addr_invalid(np, EACM_STRICT | EACM_NOLOG | EACM_NONAME))
962 goto jefrom;
963 } break;
964 case ok_v_HOME:
965 /* Note this gets called from main.c during initialization, and they
966 * simply set this to pw_dir as a fallback: don't verify _that_ call.
967 * See main.c! */
968 if(!(n_pstate & n_PS_ROOT) && !n_is_dir(*val, TRUM1)){
969 emsg = N_("$HOME is not a directory or not accessible: %s\n");
970 goto jerr;
972 break;
973 case ok_v_hostname:
974 case ok_v_smtp_hostname:
975 #ifdef HAVE_IDNA
976 if(**val != '\0'){
977 struct n_string cnv;
979 n_string_creat_auto(&cnv);
980 if(!n_idna_to_ascii(&cnv, *val, UIZ_MAX)){
981 /*n_string_gut(&res);*/
982 emsg = N_("*hostname*/*smtp_hostname*: "
983 "IDNA encoding failed: %s\n");
984 goto jerr;
986 *val = n_string_cp(&cnv);
987 /*n_string_drop_ownership(&cnv);*/
989 #endif
990 break;
991 case ok_v_quote_chars:{
992 char c;
993 char const *cp;
995 for(cp = *val; (c = *cp++) != '\0';)
996 if(!asciichar(c) || blankspacechar(c)){
997 ok = FAL0;
998 break;
1000 } break;
1001 case ok_v_sendcharsets:{
1002 struct n_string s, *sp = &s;
1003 char *csv, *cp;
1005 sp = n_string_creat_auto(sp);
1006 csv = savestr(*val);
1008 while((cp = n_strsep(&csv, ',', TRU1)) != NULL){
1009 if((cp = n_iconv_normalize_name(cp)) == NULL){
1010 ok = FAL0;
1011 break;
1013 if(sp->s_len > 0)
1014 sp = n_string_push_c(sp, ',');
1015 sp = n_string_push_cp(sp, cp);
1018 *val = n_string_cp(sp);
1019 /* n_string_drop_ownership(sp); */
1020 } break;
1021 case ok_v_TMPDIR:
1022 if(!n_is_dir(*val, TRU1)){
1023 emsg = N_("$TMPDIR is not a directory or not accessible: %s\n");
1024 goto jerr;
1026 break;
1027 case ok_v_umask:
1028 if(**val != '\0'){
1029 ui64_t uib;
1031 n_idec_ui64_cp(&uib, *val, 0, NULL);
1032 if(uib & ~0777u){ /* (is valid _VF_POSNUM) */
1033 emsg = N_("Invalid *umask* setting: %s\n");
1034 goto jerr;
1037 break;
1039 }else if(avvm == a_AMV_VIP_SET_POST){
1040 switch(okey){
1041 default:
1042 break;
1043 case ok_b_ask:
1044 ok_bset(asksub);
1045 break;
1046 case ok_b_debug:
1047 n_poption |= n_PO_DEBUG;
1048 break;
1049 case ok_v_HOME:
1050 /* Invalidate any resolved folder then, too
1051 * FALLTHRU */
1052 case ok_v_folder:
1053 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1054 break;
1055 case ok_v_ifs:{
1056 char *x_b, *x, c;
1057 char const *cp;
1059 cp = *val;
1060 x_b = x = n_autorec_alloc(strlen(cp) +1);
1061 while((c = *cp++) != '\0')
1062 if(spacechar(c))
1063 *x++ = c;
1064 *x = '\0';
1065 n_PS_ROOT_BLOCK(ok_vset(ifs_ws, x_b));
1066 } break;
1067 #ifdef HAVE_SETLOCALE
1068 case ok_v_LANG:
1069 case ok_v_LC_ALL:
1070 case ok_v_LC_CTYPE:
1071 n_locale_init();
1072 break;
1073 #endif
1074 case ok_b_memdebug:
1075 n_poption |= n_PO_MEMDEBUG;
1076 break;
1077 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1078 if(!(n_pstate & n_PS_ROOT))
1079 n_PS_ROOT_BLOCK(ok_bset(posix));
1080 break;
1081 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1082 if(!(n_pstate & n_PS_ROOT))
1083 n_PS_ROOT_BLOCK(ok_bset(POSIXLY_CORRECT));
1084 break;
1085 case ok_b_skipemptybody:
1086 n_poption |= n_PO_E_FLAG;
1087 break;
1088 case ok_b_typescript_mode:
1089 ok_bset(colour_disable);
1090 ok_bset(line_editor_disable);
1091 if(!(n_psonce & n_PSO_STARTED))
1092 ok_bset(termcap_disable);
1093 break;
1094 case ok_v_umask:
1095 if(**val != '\0'){
1096 ui64_t uib;
1098 n_idec_ui64_cp(&uib, *val, 0, NULL);
1099 umask((mode_t)uib);
1101 break;
1102 case ok_b_verbose:
1103 n_poption |= (n_poption & n_PO_VERB) ? n_PO_VERBVERB : n_PO_VERB;
1104 break;
1106 }else{
1107 switch(okey){
1108 default:
1109 break;
1110 case ok_b_ask:
1111 ok_bclear(asksub);
1112 break;
1113 case ok_b_debug:
1114 n_poption &= ~n_PO_DEBUG;
1115 break;
1116 case ok_v_customhdr:{
1117 struct n_header_field *hfp;
1119 while((hfp = n_customhdr_list) != NULL){
1120 n_customhdr_list = hfp->hf_next;
1121 n_free(hfp);
1123 } break;
1124 case ok_v_HOME:
1125 /* Invalidate any resolved folder then, too
1126 * FALLTHRU */
1127 case ok_v_folder:
1128 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1129 break;
1130 case ok_b_memdebug:
1131 n_poption &= ~n_PO_MEMDEBUG;
1132 break;
1133 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1134 if(!(n_pstate & n_PS_ROOT))
1135 n_PS_ROOT_BLOCK(ok_bclear(posix));
1136 break;
1137 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1138 if(!(n_pstate & n_PS_ROOT))
1139 n_PS_ROOT_BLOCK(ok_bclear(POSIXLY_CORRECT));
1140 break;
1141 case ok_b_skipemptybody:
1142 n_poption &= ~n_PO_E_FLAG;
1143 break;
1144 case ok_b_verbose:
1145 n_poption &= ~(n_PO_VERB | n_PO_VERBVERB);
1146 break;
1150 jleave:
1151 NYD2_LEAVE;
1152 return ok;
1153 jerr:
1154 n_err(V_(emsg), n_shexp_quote_cp(*val, FAL0));
1155 ok = FAL0;
1156 goto jleave;
1159 static bool_t
1160 a_amv_var_check_num(char const *val, bool_t posnum){
1161 /* TODO The internal/environment variables which are num= or posnum= should
1162 * TODO gain special lookup functions, or the return should be void* and
1163 * TODO castable to integer; i.e. no more strtoX() should be needed.
1164 * TODO I.e., the result of this function should instead be stored */
1165 bool_t rv;
1166 NYD2_ENTER;
1168 rv = TRU1;
1170 if(*val != '\0'){ /* Would be _VF_NOTEMPTY if not allowed */
1171 ui64_t uib;
1172 enum n_idec_state ids;
1174 ids = n_idec_cp(&uib, val, 0,
1175 (n_IDEC_MODE_LIMIT_32BIT |
1176 (posnum ? n_IDEC_MODE_SIGNED_TYPE : n_IDEC_MODE_NONE)), NULL);
1177 if((ids & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
1178 ) != n_IDEC_STATE_CONSUMED)
1179 rv = FAL0;
1180 /* TODO Unless we store integers we need to look and forbid, because
1181 * TODO callee may not be able to swallow, e.g., "-1" */
1182 if(posnum && (ids & n_IDEC_STATE_SEEN_MINUS))
1183 rv = FAL0;
1185 NYD2_LEAVE;
1186 return rv;
1189 static bool_t
1190 a_amv_var_revlookup(struct a_amv_var_carrier *avcp, char const *name){
1191 ui32_t hash, i, j;
1192 struct a_amv_var_map const *avmp;
1193 char c;
1194 NYD2_ENTER;
1196 /* It may be a special a.k.a. macro-local or one-letter parameter */
1197 c = name[0];
1198 if(n_UNLIKELY(digitchar(c))){
1199 /* (Inline dec. atoi, ugh) */
1200 for(j = (ui8_t)c - '0', i = 1;; ++i){
1201 c = name[i];
1202 if(c == '\0')
1203 break;
1204 if(!digitchar(c))
1205 goto jno_special_param;
1206 j = j * 10 + (ui8_t)c - '0';
1208 if(j <= a_AMV_POSPAR_MAX){
1209 avcp->avc_special_cat = a_AMV_VSC_POSPAR;
1210 goto jspecial_param;
1212 }else if(n_UNLIKELY(name[1] == '\0')){
1213 switch(c){
1214 case '?':
1215 case '!':
1216 avcp->avc_special_cat = a_AMV_VSC_GLOBAL;
1217 j = (c == '?') ? a_AMV_VST_QM : a_AMV_VST_EM;
1218 goto jspecial_param;
1219 case '^':
1220 goto jmultiplex;
1221 case '*':
1222 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1223 j = a_AMV_VST_STAR;
1224 goto jspecial_param;
1225 case '@':
1226 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1227 j = a_AMV_VST_AT;
1228 goto jspecial_param;
1229 case '#':
1230 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1231 j = a_AMV_VST_NOSIGN;
1232 goto jspecial_param;
1233 default:
1234 break;
1236 }else if(c == '^'){
1237 jmultiplex:
1238 avcp->avc_special_cat = a_AMV_VSC_MULTIPLEX;
1239 j = a_AMV_VST_CACC;
1240 goto jspecial_param;
1243 /* Normal reverse lookup, walk over the hashtable */
1244 jno_special_param:
1245 avcp->avc_special_cat = a_AMV_VSC_NONE;
1246 avcp->avc_name = name;
1247 avcp->avc_hash = hash = a_AMV_NAME2HASH(name);
1249 for(i = hash % a_AMV_VAR_REV_PRIME, j = 0; j <= a_AMV_VAR_REV_LONGEST; ++j){
1250 ui32_t x;
1252 if((x = a_amv_var_revmap[i]) == a_AMV_VAR_REV_ILL)
1253 break;
1255 avmp = &a_amv_var_map[x];
1256 if(avmp->avm_hash == hash &&
1257 !strcmp(&a_amv_var_names[avmp->avm_keyoff], name)){
1258 avcp->avc_map = avmp;
1259 avcp->avc_okey = (enum okeys)x;
1260 goto jleave;
1263 if(++i == a_AMV_VAR_REV_PRIME){
1264 #ifdef a_AMV_VAR_REV_WRAPAROUND
1265 i = 0;
1266 #else
1267 break;
1268 #endif
1272 avcp->avc_map = NULL;
1273 avcp = NULL;
1274 jleave:
1275 assert(avcp == NULL || avcp->avc_map != NULL ||
1276 avcp->avc_special_cat == a_AMV_VSC_NONE);
1277 NYD2_LEAVE;
1278 return (avcp != NULL);
1280 /* All these are mapped to *--special-param* */
1281 jspecial_param:
1282 avcp->avc_name = name;
1283 avcp->avc_special_prop = (ui16_t)j;
1284 avmp = &a_amv_var_map[a_AMV_VAR__SPECIAL_PARAM_MAP_IDX];
1285 avcp->avc_hash = avmp->avm_hash;
1286 avcp->avc_map = avmp;
1287 avcp->avc_okey = ok_v___special_param;
1288 goto jleave;
1291 static bool_t
1292 a_amv_var_lookup(struct a_amv_var_carrier *avcp,
1293 enum a_amv_var_lookup_flags avlf){
1294 size_t i;
1295 char const *cp;
1296 ui32_t f;
1297 struct a_amv_var_map const *avmp;
1298 struct a_amv_var *avp;
1299 NYD2_ENTER;
1301 assert(!(avlf & a_AMV_VLOOK_LOCAL_ONLY) || (avlf & a_AMV_VLOOK_LOCAL));
1302 assert(!(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT) ||
1303 (avlf & a_AMV_VLOOK_I3VAL_NONEW));
1305 /* C99 */{
1306 struct a_amv_var **avpp, *lavp;
1308 avcp->avc_prime = a_AMV_HASH2PRIME(avcp->avc_hash);
1310 /* Optionally macro-`local' variables first */
1311 if(avlf & a_AMV_VLOOK_LOCAL){
1312 if(a_amv_lopts != NULL &&
1313 (avpp = *a_amv_lopts->as_amcap->amca_local_vars) != NULL){
1314 avpp += avcp->avc_prime;
1316 for(lavp = NULL, avp = *avpp; avp != NULL;
1317 lavp = avp, avp = avp->av_link)
1318 if(!strcmp(avp->av_name, avcp->avc_name)){
1319 /* Relink as head, hope it "sorts on usage" over time.
1320 * The code relies on this behaviour! */
1321 if(lavp != NULL){
1322 lavp->av_link = avp->av_link;
1323 avp->av_link = *avpp;
1324 *avpp = avp;
1326 goto jleave;
1330 if(avlf & a_AMV_VLOOK_LOCAL_ONLY)
1331 goto jerr;
1334 /* Global variable map */
1335 avpp = &a_amv_vars[avcp->avc_prime];
1337 for(lavp = NULL, avp = *avpp; avp != NULL; lavp = avp, avp = avp->av_link)
1338 if(!strcmp(avp->av_name, avcp->avc_name)){
1339 /* Relink as head, hope it "sorts on usage" over time.
1340 * The code relies on this behaviour! */
1341 if(lavp != NULL){
1342 lavp->av_link = avp->av_link;
1343 avp->av_link = *avpp;
1344 *avpp = avp;
1347 /* If this setting has been established via -S and we still have
1348 * not reached the _STARTED_CONFIG program state, it may have been
1349 * an explicit "clearance" that is to be treated as unset.
1350 * Because that is a special condition that (has been hacked in
1351 * later and) needs to be encapsulated in lower levels, but not
1352 * of interest if _set() or _clear() called us */
1353 switch(avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK){
1354 case a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET:
1355 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW)){
1356 avcp->avc_var = avp;
1357 avp = NULL;
1358 goto j_leave;
1360 /* FALLTHRU */
1361 default:
1362 break;
1364 goto jleave;
1368 /* If this is not an assembled variable we need to consider some special
1369 * initialization cases and eventually create the variable anew */
1370 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1371 f = avmp->avm_flags;
1373 /* Does it have an import-from-environment flag? */
1374 if(n_UNLIKELY((f & (a_AMV_VF_IMPORT | a_AMV_VF_ENV)) != 0)){
1375 if(n_LIKELY((cp = getenv(avcp->avc_name)) != NULL)){
1376 /* May be better not to use that one, though? */
1377 /* TODO Outsource the tests into a _shared_ test function! */
1378 bool_t isempty, isbltin;
1380 isempty = (*cp == '\0' && (f & a_AMV_VF_NOTEMPTY) != 0);
1381 isbltin = ((f & (a_AMV_VF_I3VAL | a_AMV_VF_DEFVAL)) != 0);
1383 if(n_UNLIKELY(isempty)){
1384 n_err(_("Environment variable must not be empty: %s\n"),
1385 avcp->avc_name);
1386 if(!isbltin)
1387 goto jerr;
1388 }else if(n_LIKELY(*cp != '\0')){
1389 if(n_UNLIKELY((f & a_AMV_VF_NUM) &&
1390 !a_amv_var_check_num(cp, FAL0))){
1391 n_err(_("Environment variable value not a number "
1392 "or out of range: %s\n"), avcp->avc_name);
1393 goto jerr;
1395 if(n_UNLIKELY((f & a_AMV_VF_POSNUM) &&
1396 !a_amv_var_check_num(cp, TRU1))){
1397 n_err(_("Environment variable value not a number, "
1398 "negative or out of range: %s\n"), avcp->avc_name);
1399 goto jerr;
1401 goto jnewval;
1402 }else
1403 goto jnewval;
1407 /* A first-time init switch is to be handled now and here */
1408 if(n_UNLIKELY((f & a_AMV_VF_I3VAL) != 0)){
1409 static struct a_amv_var_defval const **arr,
1410 *arr_base[a_AMV_VAR_I3VALS_CNT +1];
1412 if(arr == NULL){
1413 arr = &arr_base[0];
1414 arr[i = a_AMV_VAR_I3VALS_CNT] = NULL;
1415 while(i-- > 0)
1416 arr[i] = &a_amv_var_i3vals[i];
1419 for(i = 0; arr[i] != NULL; ++i)
1420 if(arr[i]->avdv_okey == avcp->avc_okey){
1421 cp = (f & a_AMV_VF_BOOL) ? n_1 : arr[i]->avdv_value;
1422 /* Remove this entry, hope entire block becomes no-op asap */
1424 arr[i] = arr[i + 1];
1425 while(arr[i++] != NULL);
1427 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW))
1428 goto jnewval;
1429 if(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT)
1430 avp = (struct a_amv_var*)-1;
1431 goto jleave;
1435 /* */
1436 jdefval:
1437 if(n_UNLIKELY(f & a_AMV_VF_DEFVAL) != 0){
1438 for(i = 0; i < a_AMV_VAR_DEFVALS_CNT; ++i)
1439 if(a_amv_var_defvals[i].avdv_okey == avcp->avc_okey){
1440 cp = (f & a_AMV_VF_BOOL) ? n_1
1441 : a_amv_var_defvals[i].avdv_value;
1442 goto jnewval;
1446 /* The virtual variables */
1447 if(n_UNLIKELY((f & a_AMV_VF_VIRT) != 0)){
1448 for(i = 0; i < a_AMV_VAR_VIRTS_CNT; ++i)
1449 if(a_amv_var_virts[i].avv_okey == avcp->avc_okey){
1450 avp = n_UNCONST(a_amv_var_virts[i].avv_var);
1451 goto jleave;
1453 /* Not reached */
1457 jerr:
1458 avp = NULL;
1459 jleave:
1460 avcp->avc_var = avp;
1461 j_leave:
1462 NYD2_LEAVE;
1463 return (avp != NULL);
1465 jnewval:
1466 assert(avmp != NULL);
1467 assert(f == avmp->avm_flags);
1468 /* E.g., $TMPDIR may be set to non-existent, so we need to be able to catch
1469 * that and redirect to a possible default value */
1470 if((f & a_AMV_VF_VIP) &&
1471 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, &cp)){
1472 #ifdef HAVE_SETENV
1473 if(f & (a_AMV_VF_IMPORT | a_AMV_VF_ENV))
1474 unsetenv(avcp->avc_name);
1475 #endif
1476 if(n_UNLIKELY(f & a_AMV_VF_DEFVAL) != 0)
1477 goto jdefval;
1478 goto jerr;
1479 }else{
1480 struct a_amv_var **avpp;
1481 size_t l;
1483 l = strlen(avcp->avc_name) +1;
1484 avcp->avc_var =
1485 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1486 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1487 *avpp = avp;
1488 avp->av_flags = f;
1489 avp->av_value = a_amv_var_copy(cp);
1490 memcpy(avp->av_name, avcp->avc_name, l);
1492 if(f & a_AMV_VF_ENV)
1493 a_amv_var__putenv(avcp, avp);
1494 if(f & a_AMV_VF_VIP)
1495 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, &cp);
1496 goto jleave;
1500 static char const *
1501 a_amv_var_vsc_global(struct a_amv_var_carrier *avcp){
1502 char itoabuf[32];
1503 char const *rv;
1504 si32_t *ep;
1505 struct a_amv_var_map const *avmp;
1506 NYD2_ENTER;
1508 /* Not function local, TODO but lazy evaluted for now */
1509 if(avcp->avc_special_prop == a_AMV_VST_QM){
1510 avmp = &a_amv_var_map[a_AMV_VAR__QM_MAP_IDX];
1511 avcp->avc_okey = ok_v___qm;
1512 ep = &n_pstate_ex_no;
1513 }else{
1514 avmp = &a_amv_var_map[a_AMV_VAR__EM_MAP_IDX];
1515 avcp->avc_okey = ok_v___em;
1516 ep = &n_pstate_err_no;
1519 /* XXX Oh heaven, we are responsible to ensure that $?/! is up-to-date
1520 * TODO we could num=1 ok_v___[qe]m, but the thing is still a string
1521 * TODO and thus conversion takes places over and over again; also
1522 * TODO for now that would have to occur before we set _that_ value
1523 * TODO so let's special treat it until we store ints as such */
1524 switch(*ep){
1525 case 0: rv = n_0; break;
1526 case 1: rv = n_1; break;
1527 default:
1528 snprintf(itoabuf, sizeof itoabuf, "%d", *ep);
1529 rv = itoabuf;
1530 break;
1532 n_PS_ROOT_BLOCK(n_var_okset(avcp->avc_okey, (uintptr_t)rv));
1534 avcp->avc_hash = avmp->avm_hash;
1535 avcp->avc_map = avmp;
1536 rv = a_amv_var_lookup(avcp, a_AMV_VLOOK_NONE)
1537 ? avcp->avc_var->av_value : NULL;
1538 NYD2_LEAVE;
1539 return rv;
1542 static char const *
1543 a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp){
1544 char itoabuf[32];
1545 si32_t e;
1546 size_t i;
1547 char const *rv;
1548 NYD2_ENTER;
1550 i = strlen(rv = &avcp->avc_name[1]);
1552 /* ERR, ERRDOC, ERRNAME, plus *-NAME variants */
1553 if(rv[0] == 'E' && i >= 3 && rv[1] == 'R' && rv[2] == 'R'){
1554 if(i == 3){
1555 e = n_pstate_err_no;
1556 goto jeno;
1557 }else if(rv[3] == '-'){
1558 e = n_err_from_name(&rv[4]);
1559 jeno:
1560 switch(e){
1561 case 0: rv = n_0; break;
1562 case 1: rv = n_1; break;
1563 default:
1564 snprintf(itoabuf, sizeof itoabuf, "%d", e);
1565 rv = savestr(itoabuf); /* XXX yet, cannot do numbers */
1566 break;
1568 goto jleave;
1569 }else if(i >= 6){
1570 if(!memcmp(&rv[3], "DOC", 3)){
1571 rv += 6;
1572 switch(*rv){
1573 case '\0': e = n_pstate_err_no; break;
1574 case '-': e = n_err_from_name(&rv[1]); break;
1575 default: goto jerr;
1577 rv = n_err_to_doc(e);
1578 goto jleave;
1579 }else if(i >= 7 && !memcmp(&rv[3], "NAME", 4)){
1580 rv += 7;
1581 switch(*rv){
1582 case '\0': e = n_pstate_err_no; break;
1583 case '-': e = n_err_from_name(&rv[1]); break;
1584 default: goto jerr;
1586 rv = n_err_to_name(e);
1587 goto jleave;
1592 jerr:
1593 rv = NULL;
1594 jleave:
1595 NYD2_LEAVE;
1596 return rv;
1599 static char const *
1600 a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp){
1601 size_t i, j;
1602 ui16_t argc;
1603 char const *rv, **argv;
1604 NYD2_ENTER;
1606 rv = NULL;
1608 /* If in a macro/xy.. */
1609 if(a_amv_lopts != NULL){
1610 bool_t ismacky;
1611 struct a_amv_mac_call_args *amcap;
1613 amcap = a_amv_lopts->as_amcap;
1614 argc = amcap->amca_pospar.app_count;
1615 argv = amcap->amca_pospar.app_dat;
1616 argv += amcap->amca_pospar.app_idx;
1618 /* ..in a `call'ed macro only, to be exact. Or in a_AMV_MACKY_MACK */
1619 if(!(ismacky = (amcap->amca_amp == a_AMV_MACKY_MACK)) &&
1620 (amcap->amca_ps_hook_mask ||
1621 (assert(amcap->amca_amp != NULL),
1622 (amcap->amca_amp->am_flags & a_AMV_MF_TYPE_MASK
1623 ) == a_AMV_MF_ACCOUNT)))
1624 goto jleave;
1626 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1627 if(avcp->avc_special_prop > 0){
1628 if(argc >= avcp->avc_special_prop)
1629 rv = argv[avcp->avc_special_prop - 1];
1630 }else if(ismacky)
1631 rv = amcap->amca_name;
1632 else
1633 rv = (a_amv_lopts->as_up != NULL
1634 ? a_amv_lopts->as_up->as_amcap->amca_name : n_empty);
1635 goto jleave;
1637 /* MACKY_MACK doesn't know about [*@#] */
1638 /*else*/ if(ismacky){
1639 if(n_poption & n_PO_D_V)
1640 n_err(_("Cannot use $*/$@/$# in this context: %s\n"),
1641 n_shexp_quote_cp(avcp->avc_name, FAL0));
1642 goto jleave;
1644 }else{
1645 argc = a_amv_pospar.app_count;
1646 argv = a_amv_pospar.app_dat;
1647 argv += a_amv_pospar.app_idx;
1649 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1650 if(avcp->avc_special_prop > 0){
1651 if(argc >= avcp->avc_special_prop)
1652 rv = argv[avcp->avc_special_prop - 1];
1653 }else
1654 rv = n_progname;
1655 goto jleave;
1659 switch(avcp->avc_special_prop){ /* XXX OPTIMIZE */
1660 case a_AMV_VST_STAR:{
1661 char sep;
1663 sep = *ok_vlook(ifs);
1664 if(0){
1665 case a_AMV_VST_AT:
1666 sep = ' ';
1668 for(i = j = 0; i < argc; ++i)
1669 j += strlen(argv[i]) + 1;
1670 if(j == 0)
1671 rv = n_empty;
1672 else{
1673 char *cp;
1675 rv = cp = n_autorec_alloc(j);
1676 for(i = j = 0; i < argc; ++i){
1677 j = strlen(argv[i]);
1678 memcpy(cp, argv[i], j);
1679 cp += j;
1680 if(sep != '\0')
1681 *cp++ = sep;
1683 if(sep != '\0')
1684 --cp;
1685 *cp = '\0';
1687 }break;
1688 case a_AMV_VST_NOSIGN:{
1689 char *cp;
1691 rv = cp = n_autorec_alloc(sizeof("65535"));
1692 snprintf(cp, sizeof("65535"), "%hu", argc);
1693 }break;
1694 default:
1695 rv = n_empty;
1696 break;
1698 jleave:
1699 NYD2_LEAVE;
1700 return rv;
1703 static bool_t
1704 a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
1705 enum a_amv_var_setclr_flags avscf){
1706 struct a_amv_var *avp;
1707 char *oval;
1708 struct a_amv_var_map const *avmp;
1709 bool_t rv;
1710 NYD2_ENTER;
1712 if(value == NULL){
1713 rv = a_amv_var_clear(avcp, avscf);
1714 goto jleave;
1717 if((avmp = avcp->avc_map) != NULL){
1718 rv = FAL0;
1720 /* Validity checks */
1721 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_RDONLY) != 0 &&
1722 !(n_pstate & n_PS_ROOT))){
1723 value = N_("Variable is read-only: %s\n");
1724 goto jeavmp;
1726 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOTEMPTY) && *value == '\0')){
1727 value = N_("Variable must not be empty: %s\n");
1728 goto jeavmp;
1730 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1731 !a_amv_var_check_num(value, FAL0))){
1732 value = N_("Variable value not a number or out of range: %s\n");
1733 goto jeavmp;
1735 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1736 !a_amv_var_check_num(value, TRU1))){
1737 value = _("Variable value not a number, negative, "
1738 "or out of range: %s\n");
1739 goto jeavmp;
1742 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_IMPORT) != 0 &&
1743 !(n_psonce & n_PSO_STARTED) && !(n_pstate & n_PS_ROOT))){
1744 value = N_("Variable cannot be set in a resource file: %s\n");
1745 goto jeavmp;
1748 /* Any more complicated inter-dependency? */
1749 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1750 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, &value))){
1751 value = N_("Assignment of variable aborted: %s\n");
1752 jeavmp:
1753 n_err(V_(value), avcp->avc_name);
1754 goto jleave;
1757 /* Transformations */
1758 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_LOWER)){
1759 char c;
1761 oval = savestr(value);
1762 value = oval;
1763 for(; (c = *oval) != '\0'; ++oval)
1764 *oval = lowerconv(c);
1767 /* Obsoletion warning */
1768 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_OBSOLETE) != 0))
1769 n_OBSOLETE2(_("variable superseded or obsoleted"), avcp->avc_name);
1772 /* Lookup possibly existing var. For */
1774 rv = TRU1;
1775 a_amv_var_lookup(avcp, (a_AMV_VLOOK_I3VAL_NONEW |
1776 ((avscf & a_AMV_VSETCLR_LOCAL)
1777 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0)));
1778 avp = avcp->avc_var;
1780 /* A `local' setting is never covered by `localopts' nor frozen */
1781 if(avscf & a_AMV_VSETCLR_LOCAL)
1782 goto jislocal;
1784 /* If this setting had been established via -S and we still have not reached
1785 * the _STARTED_CONFIG program state, silently ignore request! */
1786 if(n_UNLIKELY(avp != NULL) &&
1787 n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1788 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1789 if((n_pstate & n_PS_ROOT) ||
1790 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1791 (n_poption & n_PO_S_FLAG_TEMPORARY)))
1792 goto joval_and_go;
1793 if(n_poption & n_PO_D_VV)
1794 n_err(_("Temporarily frozen by -S, not `set'ing: %s\n"),
1795 avcp->avc_name);
1796 goto jleave;
1799 /* Otherwise, if -S freezing was an `unset' request, be very simple and
1800 * avoid tampering with that very special case we are not really prepared
1801 * for just one more line of code: throw the old thing away! */
1802 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET))
1803 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1804 else{
1805 assert(avp->av_value == n_empty);
1806 assert(a_amv_vars[avcp->avc_prime] == avp);
1807 a_amv_vars[avcp->avc_prime] = avp->av_link;
1808 n_free(avp);
1809 avcp->avc_var = avp = NULL;
1813 /* Optionally cover by `localopts' */
1814 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1815 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1816 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1818 jislocal:
1819 if(avp != NULL)
1820 joval_and_go:
1821 oval = avp->av_value;
1822 else{
1823 size_t l;
1824 struct a_amv_var **avpp;
1826 if(avscf & a_AMV_VSETCLR_LOCAL){
1827 if((avpp = *a_amv_lopts->as_amcap->amca_local_vars) == NULL)
1828 avpp = *(a_amv_lopts->as_amcap->amca_local_vars =
1829 n_calloc(1, sizeof(*a_amv_lopts->as_amcap->amca_local_vars)));
1830 avpp += avcp->avc_prime;
1831 }else
1832 avpp = &a_amv_vars[avcp->avc_prime];
1834 l = strlen(avcp->avc_name) +1;
1835 avcp->avc_var = avp = n_calloc(1,
1836 n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1837 avp->av_link = *avpp;
1838 *avpp = avp;
1839 avp->av_flags = (avscf & a_AMV_VSETCLR_LOCAL)
1840 ? a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL
1841 : (avmp != NULL) ? avmp->avm_flags : 0;
1842 memcpy(avp->av_name, avcp->avc_name, l);
1843 oval = n_UNCONST(n_empty);
1846 if(avmp == NULL)
1847 avp->av_value = a_amv_var_copy(value);
1848 else{
1849 assert(!(avscf & a_AMV_VSETCLR_LOCAL));
1850 /* Via `set' etc. the user may give even boolean options non-boolean
1851 * values, ignore that and force boolean */
1852 if(!(avp->av_flags & a_AMV_VF_BOOL))
1853 avp->av_value = a_amv_var_copy(value);
1854 else{
1855 if(!(n_pstate & n_PS_ROOT) && (n_poption & n_PO_D_V) &&
1856 *value != '\0')
1857 n_err(_("Ignoring value of boolean variable: %s: %s\n"),
1858 avcp->avc_name, value);
1859 avp->av_value = n_UNCONST(n_1);
1863 /* A `local' setting can skip all the crude special things */
1864 if(!(avscf & a_AMV_VSETCLR_LOCAL)){
1865 ui32_t f;
1867 f = avp->av_flags;
1869 if((avscf & a_AMV_VSETCLR_ENV) && !(f & a_AMV_VF_ENV))
1870 f |= a_AMV_VF_EXT_LINKED;
1871 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1872 rv = a_amv_var__putenv(avcp, avp);
1873 if(f & a_AMV_VF_VIP)
1874 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, &value);
1876 f &= ~a_AMV_VF_EXT__FROZEN_MASK;
1877 if(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1878 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0)
1879 f |= a_AMV_VF_EXT_FROZEN;
1881 avp->av_flags = f;
1884 a_amv_var_free(oval);
1885 jleave:
1886 NYD2_LEAVE;
1887 return rv;
1890 static bool_t
1891 a_amv_var__putenv(struct a_amv_var_carrier *avcp, struct a_amv_var *avp){
1892 #ifndef HAVE_SETENV
1893 char *cp;
1894 #endif
1895 bool_t rv;
1896 NYD2_ENTER;
1898 #ifdef HAVE_SETENV
1899 rv = (setenv(avcp->avc_name, avp->av_value, 1) == 0);
1900 #else
1901 cp = sstrdup(savecatsep(avcp->avc_name, '=', avp->av_value));
1903 if((rv = (putenv(cp) == 0))){
1904 char *ocp;
1906 ocp = avp->av_env;
1907 avp->av_env = cp;
1908 cp = ocp;
1911 if(cp != NULL)
1912 n_free(cp);
1913 #endif
1914 NYD2_LEAVE;
1915 return rv;
1918 static bool_t
1919 a_amv_var_clear(struct a_amv_var_carrier *avcp,
1920 enum a_amv_var_setclr_flags avscf){
1921 struct a_amv_var **avpp, *avp;
1922 ui32_t f;
1923 struct a_amv_var_map const *avmp;
1924 bool_t rv;
1925 NYD2_ENTER;
1927 rv = FAL0;
1928 f = 0;
1930 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1931 if(n_UNLIKELY(((f = avmp->avm_flags) & a_AMV_VF_NODEL) != 0 &&
1932 !(n_pstate & n_PS_ROOT))){
1933 n_err(_("Variable may not be unset: %s\n"), avcp->avc_name);
1934 goto jleave;
1936 if(n_UNLIKELY((f & a_AMV_VF_VIP) != 0 &&
1937 !a_amv_var_check_vips(a_AMV_VIP_CLEAR, avcp->avc_okey, NULL))){
1938 n_err(_("Clearance of variable aborted: %s\n"), avcp->avc_name);
1939 goto jleave;
1943 rv = TRU1;
1945 if(n_UNLIKELY(!a_amv_var_lookup(avcp,
1946 (((avscf & a_AMV_VSETCLR_LOCAL)
1947 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0) |
1948 a_AMV_VLOOK_I3VAL_NONEW | a_AMV_VLOOK_I3VAL_NONEW_REPORT)))){
1949 assert(avcp->avc_var == NULL);
1950 /* This may be a clearance request from the command line, via -S, and we
1951 * need to keep track of that! Unfortunately we are not prepared for
1952 * this, really, so we need to create a fake entry that is known and
1953 * handled correctly by the lowermost variable layer!
1954 * However, all this cannot happen for plain unset of `local' variables */
1955 if(avscf & a_AMV_VSETCLR_LOCAL)
1956 goto jleave;
1957 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
1958 (n_poption & n_PO_S_FLAG_TEMPORARY)) Jfreeze:{
1959 size_t l;
1961 l = strlen(avcp->avc_name) +1;
1962 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1963 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1964 *avpp = avp;
1965 avp->av_value = n_UNCONST(n_empty); /* Sth. covered by _var_free()! */
1966 assert(f == (avmp != NULL ? avmp->avm_flags : 0));
1967 avp->av_flags = f | a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET;
1968 memcpy(avp->av_name, avcp->avc_name, l);
1970 if((avscf & a_AMV_VSETCLR_ENV) || (f & a_AMV_VF_ENV))
1971 a_amv_var__clearenv(avcp->avc_name, NULL);
1972 }else if(avscf & a_AMV_VSETCLR_ENV){
1973 jforce_env:
1974 if(!(rv = a_amv_var__clearenv(avcp->avc_name, NULL)))
1975 goto jerr_env_unset;
1976 }else{
1977 /* TODO "cannot unset undefined variable" not echoed in "ROBOT" state,
1978 * TODO should only be like that with "ignerr"! */
1979 jerr_env_unset:
1980 if(!(n_pstate & (n_PS_ROOT | n_PS_ROBOT)) && (n_poption & n_PO_D_V))
1981 n_err(_("Cannot unset undefined variable: %s\n"), avcp->avc_name);
1983 goto jleave;
1984 }else if((avp = avcp->avc_var) == (struct a_amv_var*)-1){
1985 /* Clearance request from command line, via -S? As above.. */
1986 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1987 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0))
1988 goto Jfreeze;
1989 avcp->avc_var = NULL;
1990 if(avscf & a_AMV_VSETCLR_ENV)
1991 goto jforce_env;
1992 goto jleave;
1994 assert(avcp->avc_var != NULL);
1996 /* `local' variables bypass "frozen" checks and `localopts' coverage etc. */
1997 if((f = avp->av_flags) & a_AMV_VF_EXT_LOCAL)
1998 goto jdefault_path;
2000 /* If this setting has been established via -S and we still have not reached
2001 * the _STARTED_CONFIG program state, silently ignore request!
2002 * XXX All this is very complicated for the tenth of a second */
2003 /*else*/ if(n_UNLIKELY((f & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
2004 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
2005 if((n_pstate & n_PS_ROOT) ||
2006 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
2007 (n_poption & n_PO_S_FLAG_TEMPORARY))){
2008 /* Be aware this may turn a set into an unset! */
2009 if(!(f & a_AMV_VF_EXT_FROZEN_UNSET)){
2010 if(f & a_AMV_VF_DEFVAL)
2011 goto jdefault_path;
2012 a_amv_var_free(avp->av_value);
2013 f |= a_AMV_VF_EXT_FROZEN_UNSET;
2014 avp->av_flags = f;
2015 avp->av_value = n_UNCONST(n_empty); /* _var_free() covered */
2016 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
2017 goto jforce_env;
2019 goto jleave;
2021 if(n_poption & n_PO_D_VV)
2022 n_err(_("Temporarily frozen by -S, not `unset'ting: %s\n"),
2023 avcp->avc_name);
2024 goto jleave;
2026 f &= ~a_AMV_VF_EXT__FROZEN_MASK;
2027 avp->av_flags = f;
2030 if(n_UNLIKELY(a_amv_lopts != NULL) &&
2031 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
2032 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
2034 jdefault_path:
2035 assert(avp == avcp->avc_var);
2036 avcp->avc_var = NULL;
2037 avpp = &(((f = avp->av_flags) & a_AMV_VF_EXT_LOCAL)
2038 ? *a_amv_lopts->as_amcap->amca_local_vars : a_amv_vars
2039 )[avcp->avc_prime];
2040 assert(*avpp == avp); /* (always listhead after lookup()) */
2041 *avpp = (*avpp)->av_link;
2043 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
2044 rv = a_amv_var__clearenv(avp->av_name, avp);
2045 a_amv_var_free(avp->av_value);
2046 n_free(avp);
2048 /* XXX Fun part, extremely simple-minded for now: if this variable has
2049 * XXX a default value, immediately reinstantiate it! TODO Heh? */
2050 if(n_UNLIKELY(avmp != NULL && (avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)){
2051 a_amv_var_lookup(avcp, a_AMV_VLOOK_I3VAL_NONEW);
2052 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
2053 (n_poption & n_PO_S_FLAG_TEMPORARY))
2054 avcp->avc_var->av_flags |= a_AMV_VF_EXT_FROZEN;
2056 jleave:
2057 NYD2_LEAVE;
2058 return rv;
2061 static bool_t
2062 a_amv_var__clearenv(char const *name, struct a_amv_var *avp){
2063 extern char **environ;
2064 char **ecpp;
2065 bool_t rv;
2066 NYD2_ENTER;
2067 n_UNUSED(avp);
2069 rv = FAL0;
2070 ecpp = environ;
2072 #ifndef HAVE_SETENV
2073 if(avp != NULL && avp->av_env != NULL){
2074 for(; *ecpp != NULL; ++ecpp)
2075 if(*ecpp == avp->av_env){
2077 ecpp[0] = ecpp[1];
2078 while(*ecpp++ != NULL);
2079 n_free(avp->av_env);
2080 avp->av_env = NULL;
2081 rv = TRU1;
2082 break;
2084 }else
2085 #endif
2087 size_t l;
2089 if((l = strlen(name)) > 0){
2090 for(; *ecpp != NULL; ++ecpp)
2091 if(!strncmp(*ecpp, name, l) && (*ecpp)[l] == '='){
2092 #ifdef HAVE_SETENV
2093 unsetenv(name);
2094 #else
2096 ecpp[0] = ecpp[1];
2097 while(*ecpp++ != NULL);
2098 #endif
2099 rv = TRU1;
2100 break;
2104 NYD2_LEAVE;
2105 return rv;
2108 static void
2109 a_amv_var_show_all(void){
2110 struct n_string msg, *msgp;
2111 FILE *fp;
2112 size_t no, i;
2113 struct a_amv_var *avp;
2114 char const **vacp, **cap;
2115 NYD2_ENTER;
2117 if((fp = Ftmp(NULL, "setlist", OF_RDWR | OF_UNLINK | OF_REGISTER)) == NULL){
2118 n_perr(_("`set' list: cannot create temporary file"), 0);
2119 goto jleave;
2122 /* We need to instantiate first-time-inits and default values here, so that
2123 * they will be regular members of our _vars[] table */
2124 for(i = a_AMV_VAR_I3VALS_CNT; i-- > 0;)
2125 n_var_oklook(a_amv_var_i3vals[i].avdv_okey);
2126 for(i = a_AMV_VAR_DEFVALS_CNT; i-- > 0;)
2127 n_var_oklook(a_amv_var_defvals[i].avdv_okey);
2129 for(no = i = 0; i < a_AMV_PRIME; ++i)
2130 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2131 ++no;
2132 no += a_AMV_VAR_VIRTS_CNT;
2134 vacp = n_autorec_alloc(no * sizeof(*vacp));
2136 for(cap = vacp, i = 0; i < a_AMV_PRIME; ++i)
2137 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2138 *cap++ = avp->av_name;
2139 for(i = a_AMV_VAR_VIRTS_CNT; i-- > 0;)
2140 *cap++ = a_amv_var_virts[i].avv_var->av_name;
2142 if(no > 1)
2143 qsort(vacp, no, sizeof *vacp, &a_amv_var__show_cmp);
2145 msgp = &msg;
2146 msgp = n_string_reserve(n_string_creat(msgp), 80);
2147 for(i = 0, cap = vacp; no != 0; ++cap, --no)
2148 i += a_amv_var_show(*cap, fp, msgp);
2149 n_string_gut(&msg);
2151 page_or_print(fp, i);
2152 Fclose(fp);
2153 jleave:
2154 NYD2_LEAVE;
2157 static int
2158 a_amv_var__show_cmp(void const *s1, void const *s2){
2159 int rv;
2160 NYD2_ENTER;
2162 rv = strcmp(*(char**)n_UNCONST(s1), *(char**)n_UNCONST(s2));
2163 NYD2_LEAVE;
2164 return rv;
2167 static size_t
2168 a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp){
2169 struct a_amv_var_carrier avc;
2170 char const *quote;
2171 struct a_amv_var *avp;
2172 bool_t isset;
2173 size_t i;
2174 NYD2_ENTER;
2176 msgp = n_string_trunc(msgp, 0);
2177 i = 0;
2179 a_amv_var_revlookup(&avc, name);
2180 isset = a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
2181 avp = avc.avc_var;
2183 if(n_poption & n_PO_D_V){
2184 if(avc.avc_map == NULL){
2185 msgp = n_string_push_cp(msgp, "#assembled variable with value");
2186 i = 1;
2187 }else{
2188 struct{
2189 ui16_t flag;
2190 char msg[22];
2191 } const tbase[] = {
2192 {a_AMV_VF_CHAIN, "variable chain"},
2193 {a_AMV_VF_VIRT, "virtual"},
2194 {a_AMV_VF_RDONLY, "read-only"},
2195 {a_AMV_VF_NODEL, "nodelete"},
2196 {a_AMV_VF_I3VAL, "initial-value"},
2197 {a_AMV_VF_DEFVAL, "default-value"},
2198 {a_AMV_VF_IMPORT, "import-environ-first\0"}, /* assert NUL in max */
2199 {a_AMV_VF_ENV, "sync-environ"},
2200 {a_AMV_VF_NOLOPTS, "no-localopts"},
2201 {a_AMV_VF_NOTEMPTY, "notempty"},
2202 {a_AMV_VF_NUM, "number"},
2203 {a_AMV_VF_POSNUM, "positive-number"},
2204 {a_AMV_VF_OBSOLETE, "obsoleted"},
2205 }, *tp;
2207 for(tp = tbase; PTRCMP(tp, <, &tbase[n_NELEM(tbase)]); ++tp)
2208 if(isset ? (avp->av_flags & tp->flag)
2209 : (avc.avc_map->avm_flags & tp->flag)){
2210 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2211 msgp = n_string_push_cp(msgp, tp->msg);
2215 if(isset && (avp->av_flags & a_AMV_VF_EXT_FROZEN)){
2216 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2217 msgp = n_string_push_cp(msgp, "(un)?set via -S");
2220 if(i > 0)
2221 msgp = n_string_push_cp(msgp, "\n ");
2224 /* (Read-only variables are generally shown via comments..) */
2225 if(!isset || (avp->av_flags & a_AMV_VF_RDONLY)){
2226 msgp = n_string_push_c(msgp, n_ns[0]);
2227 if(!isset){
2228 if(avc.avc_map != NULL && (avc.avc_map->avm_flags & a_AMV_VF_BOOL))
2229 msgp = n_string_push_cp(msgp, "boolean; ");
2230 msgp = n_string_push_cp(msgp, "variable not set: ");
2231 msgp = n_string_push_cp(msgp, n_shexp_quote_cp(name, FAL0));
2232 goto jleave;
2236 n_UNINIT(quote, NULL);
2237 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2238 quote = n_shexp_quote_cp(avp->av_value, TRU1);
2239 if(strcmp(quote, avp->av_value))
2240 msgp = n_string_push_cp(msgp, "wysh ");
2241 }else if(n_poption & n_PO_D_V)
2242 msgp = n_string_push_cp(msgp, "wysh "); /* (for shell-style comment) */
2244 if(avp->av_flags & a_AMV_VF_EXT_LINKED)
2245 msgp = n_string_push_cp(msgp, "environ ");
2246 msgp = n_string_push_cp(msgp, "set ");
2247 msgp = n_string_push_cp(msgp, name);
2249 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2250 msgp = n_string_push_c(msgp, '=');
2251 msgp = n_string_push_cp(msgp, quote);
2252 }else if(n_poption & n_PO_D_V)
2253 msgp = n_string_push_cp(msgp, " #boolean");
2255 jleave:
2256 msgp = n_string_push_c(msgp, '\n');
2257 fputs(n_string_cp(msgp), fp);
2258 NYD2_ENTER;
2259 return (i > 0 ? 2 : 1);
2262 static bool_t
2263 a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf){
2264 char *cp, *cp2, *varbuf, c;
2265 size_t errs;
2266 NYD2_ENTER;
2268 errs = 0;
2269 jouter:
2270 while((cp = *ap++) != NULL){
2271 /* Isolate key */
2272 cp2 = varbuf = n_autorec_alloc(strlen(cp) +1);
2274 for(; (c = *cp) != '=' && c != '\0'; ++cp){
2275 if(cntrlchar(c) || spacechar(c)){
2276 n_err(_("Variable name with control or space character ignored: "
2277 "%s\n"), ap[-1]);
2278 ++errs;
2279 goto jouter;
2281 *cp2++ = c;
2283 *cp2 = '\0';
2284 if(c == '\0')
2285 cp = n_UNCONST(n_empty);
2286 else
2287 ++cp;
2289 if(varbuf == cp2){
2290 n_err(_("Empty variable name ignored\n"));
2291 ++errs;
2292 }else{
2293 struct a_amv_var_carrier avc;
2294 bool_t isunset;
2296 if((isunset = (varbuf[0] == 'n' && varbuf[1] == 'o')))
2297 varbuf = &varbuf[2];
2299 a_amv_var_revlookup(&avc, varbuf);
2301 if((avscf & a_AMV_VSETCLR_LOCAL) && avc.avc_map != NULL){
2302 if(n_poption & n_PO_D_V)
2303 n_err(_("Builtin variable not overwritten by `local': %s\n"),
2304 varbuf);
2305 ++errs;
2306 }else if(isunset)
2307 errs += !a_amv_var_clear(&avc, avscf);
2308 else
2309 errs += !a_amv_var_set(&avc, cp, avscf);
2312 NYD2_LEAVE;
2313 return (errs == 0);
2316 FL int
2317 c_define(void *v){
2318 int rv;
2319 char **args;
2320 NYD_ENTER;
2322 rv = 1;
2324 if((args = v)[0] == NULL){
2325 rv = (a_amv_mac_show(a_AMV_MF_NONE) == FAL0);
2326 goto jleave;
2329 if(args[1] == NULL || args[1][0] != '{' || args[1][1] != '\0' ||
2330 args[2] != NULL){
2331 n_err(_("Synopsis: define: <name> {\n"));
2332 goto jleave;
2335 rv = (a_amv_mac_def(args[0], a_AMV_MF_NONE) == FAL0);
2336 jleave:
2337 NYD_LEAVE;
2338 return rv;
2341 FL int
2342 c_undefine(void *v){
2343 int rv;
2344 char **args;
2345 NYD_ENTER;
2347 rv = 0;
2348 args = v;
2350 rv |= !a_amv_mac_undef(*args, a_AMV_MF_NONE);
2351 while(*++args != NULL);
2352 NYD_LEAVE;
2353 return rv;
2356 FL int
2357 c_call(void *vp){
2358 int rv;
2359 NYD_ENTER;
2361 rv = a_amv_mac_call(vp, FAL0);
2362 NYD_LEAVE;
2363 return rv;
2366 FL int
2367 c_call_if(void *vp){
2368 int rv;
2369 NYD_ENTER;
2371 rv = a_amv_mac_call(vp, TRU1);
2372 NYD_LEAVE;
2373 return rv;
2376 FL int
2377 c_account(void *v){
2378 struct a_amv_mac_call_args *amcap;
2379 struct a_amv_mac *amp;
2380 char **args;
2381 int rv, i, oqf, nqf;
2382 NYD_ENTER;
2384 rv = 1;
2386 if((args = v)[0] == NULL){
2387 rv = (a_amv_mac_show(a_AMV_MF_ACCOUNT) == FAL0);
2388 goto jleave;
2391 if(args[1] && args[1][0] == '{' && args[1][1] == '\0'){
2392 if(args[2] != NULL){
2393 n_err(_("Synopsis: account: <name> {\n"));
2394 goto jleave;
2396 if(!asccasecmp(args[0], ACCOUNT_NULL)){
2397 n_err(_("`account': cannot use reserved name: %s\n"),
2398 ACCOUNT_NULL);
2399 goto jleave;
2401 rv = (a_amv_mac_def(args[0], a_AMV_MF_ACCOUNT) == FAL0);
2402 goto jleave;
2405 if(n_pstate & n_PS_HOOK_MASK){
2406 n_err(_("`account': cannot change account from within a hook\n"));
2407 goto jleave;
2410 save_mbox_for_possible_quitstuff();
2412 amp = NULL;
2413 if(asccasecmp(args[0], ACCOUNT_NULL) != 0 &&
2414 (amp = a_amv_mac_lookup(args[0], NULL, a_AMV_MF_ACCOUNT)) == NULL){
2415 n_err(_("`account': account does not exist: %s\n"), args[0]);
2416 goto jleave;
2419 oqf = savequitflags();
2421 /* Shutdown the active account */
2422 if(a_amv_acc_curr != NULL){
2423 char const *cp;
2424 char *var;
2426 /* Is there a cleanup hook? */
2427 var = savecat("on-account-cleanup-", a_amv_acc_curr->am_name);
2428 if((cp = n_var_vlook(var, FAL0)) != NULL ||
2429 (cp = ok_vlook(on_account_cleanup)) != NULL){
2430 struct a_amv_mac *amphook;
2432 if((amphook = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) != NULL){
2433 amcap = n_lofi_alloc(sizeof *amcap);
2434 memset(amcap, 0, sizeof *amcap);
2435 amcap->amca_name = cp;
2436 amcap->amca_amp = amphook;
2437 amcap->amca_unroller = &a_amv_acc_curr->am_lopts;
2438 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2439 amcap->amca_no_xcall = TRU1;
2440 n_pstate |= n_PS_HOOK;
2441 rv = a_amv_mac_exec(amcap);
2442 n_pstate &= ~n_PS_HOOK_MASK;
2443 }else
2444 n_err(_("*on-account-leave* hook %s does not exist\n"),
2445 n_shexp_quote_cp(cp, FAL0));
2448 /* `localopts'? */
2449 if(a_amv_acc_curr->am_lopts != NULL)
2450 a_amv_lopts_unroll(&a_amv_acc_curr->am_lopts);
2452 /* For accounts this lingers */
2453 --a_amv_acc_curr->am_refcnt;
2454 if(a_amv_acc_curr->am_flags & a_AMV_MF_DELETE)
2455 a_amv_mac_free(a_amv_acc_curr);
2458 a_amv_acc_curr = amp;
2460 /* And switch to any non-"null" account */
2461 if(amp != NULL){
2462 assert(amp->am_lopts == NULL);
2463 amcap = n_lofi_alloc(sizeof *amcap);
2464 memset(amcap, 0, sizeof *amcap);
2465 amcap->amca_name = amp->am_name;
2466 amcap->amca_amp = amp;
2467 amcap->amca_unroller = &amp->am_lopts;
2468 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2469 amcap->amca_no_xcall = TRU1;
2470 ++amp->am_refcnt; /* We may not run 0 to avoid being deleted! */
2471 if(!a_amv_mac_exec(amcap)){
2472 /* XXX account switch incomplete, unroll? */
2473 n_err(_("`account': failed to switch to account: %s\n"), amp->am_name);
2474 goto jleave;
2478 n_PS_ROOT_BLOCK((amp != NULL ? ok_vset(account, amp->am_name)
2479 : ok_vclear(account)));
2481 if(n_psonce & n_PSO_STARTED){
2482 assert(!(n_pstate & n_PS_HOOK_MASK));
2483 nqf = savequitflags(); /* TODO obsolete (leave -> void -> new box!) */
2484 restorequitflags(oqf);
2485 if((i = setfile("%", 0)) < 0)
2486 goto jleave;
2487 temporary_folder_hook_check(FAL0);
2488 if(i > 0 && !ok_blook(emptystart))
2489 goto jleave;
2490 n_folder_announce(n_ANNOUNCE_CHANGE);
2491 restorequitflags(nqf);
2493 rv = 0;
2494 jleave:
2495 NYD_LEAVE;
2496 return rv;
2499 FL int
2500 c_unaccount(void *v){
2501 int rv;
2502 char **args;
2503 NYD_ENTER;
2505 rv = 0;
2506 args = v;
2508 rv |= !a_amv_mac_undef(*args, a_AMV_MF_ACCOUNT);
2509 while(*++args != NULL);
2510 NYD_LEAVE;
2511 return rv;
2514 FL int
2515 c_localopts(void *vp){
2516 enum a_amv_loflags alf, alm;
2517 char const **argv;
2518 int rv;
2519 NYD_ENTER;
2521 rv = 1;
2523 if(a_amv_lopts == NULL){
2524 n_err(_("Cannot use `localopts' in this context\n"));
2525 goto jleave;
2528 if((argv = vp)[1] == NULL || is_asccaseprefix((++argv)[-1], "scope"))
2529 alf = alm = a_AMV_LF_SCOPE;
2530 else if(is_asccaseprefix(argv[-1], "call"))
2531 alf = a_AMV_LF_CALL, alm = a_AMV_LF_CALL_MASK;
2532 else if(is_asccaseprefix(argv[-1], "call-fixate"))
2533 alf = a_AMV_LF_CALL_FIXATE, alm = a_AMV_LF_CALL_MASK;
2534 else{
2535 jesynopsis:
2536 n_err(_("Synopsis: localopts: [<scope|call|call-fixate>] <boolean>\n"));
2537 goto jleave;
2540 if(alf == a_AMV_LF_SCOPE &&
2541 (a_amv_lopts->as_loflags & a_AMV_LF_SCOPE_FIXATE)){
2542 if(n_poption & n_PO_D_V)
2543 n_err(_("Cannot turn off `localopts', setting is fixated\n"));
2544 goto jleave;
2547 if((rv = n_boolify(*argv, UIZ_MAX, FAL0)) < FAL0)
2548 goto jesynopsis;
2549 a_amv_lopts->as_loflags &= ~alm;
2550 if(rv > FAL0)
2551 a_amv_lopts->as_loflags |= alf;
2552 rv = 0;
2553 jleave:
2554 NYD_LEAVE;
2555 return rv;
2558 FL int
2559 c_shift(void *vp){ /* xxx move to bottom, not in macro part! */
2560 struct a_amv_pospar *appp;
2561 ui16_t i;
2562 int rv;
2563 NYD_ENTER;
2565 rv = 1;
2567 if((vp = *(char**)vp) == NULL)
2568 i = 1;
2569 else{
2570 si16_t sib;
2572 if((n_idec_si16_cp(&sib, vp, 10, NULL
2573 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2574 ) != n_IDEC_STATE_CONSUMED || sib < 0){
2575 n_err(_("`shift': invalid argument: %s\n"), vp);
2576 goto jleave;
2578 i = (ui16_t)sib;
2581 /* If in in a macro/xy */
2582 if(a_amv_lopts != NULL){
2583 struct a_amv_mac const *amp;
2584 struct a_amv_mac_call_args *amcap;
2586 /* Explicitly do allow `vpospar' created things! */
2587 amp = (amcap = a_amv_lopts->as_amcap)->amca_amp;
2588 if((amp == NULL || amcap->amca_ps_hook_mask ||
2589 (amp->am_flags & a_AMV_MF_TYPE_MASK) == a_AMV_MF_ACCOUNT) &&
2590 amcap->amca_pospar.app_not_heap){
2591 n_err(_("Cannot use `shift' in `account's or hook macros etc.\n"));
2592 goto jleave;
2594 appp = &amcap->amca_pospar;
2595 }else
2596 appp = &a_amv_pospar;
2598 if(i > appp->app_count){
2599 n_err(_("`shift': cannot shift %hu of %hu parameters\n"),
2600 i, appp->app_count);
2601 goto jleave;
2602 }else{
2603 appp->app_idx += i;
2604 appp->app_count -= i;
2605 rv = 0;
2607 jleave:
2608 NYD_LEAVE;
2609 return rv;
2612 FL int
2613 c_return(void *vp){ /* TODO the exit status should be m_si64! */
2614 int rv;
2615 NYD_ENTER;
2617 if(a_amv_lopts != NULL){
2618 char const **argv;
2620 n_go_input_force_eof();
2621 n_pstate_err_no = n_ERR_NONE;
2622 rv = 0;
2624 if((argv = vp)[0] != NULL){
2625 si32_t i;
2627 if((n_idec_si32_cp(&i, argv[0], 10, NULL
2628 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2629 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2630 rv = (int)i;
2631 else{
2632 n_err(_("`return': return value argument is invalid: %s\n"),
2633 argv[0]);
2634 n_pstate_err_no = n_ERR_INVAL;
2635 rv = 1;
2638 if(argv[1] != NULL){
2639 if((n_idec_si32_cp(&i, argv[1], 10, NULL
2640 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2641 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2642 n_pstate_err_no = i;
2643 else{
2644 n_err(_("`return': error number argument is invalid: %s\n"),
2645 argv[1]);
2646 n_pstate_err_no = n_ERR_INVAL;
2647 rv = 1;
2651 }else{
2652 n_err(_("Can only use `return' in a macro\n"));
2653 n_pstate_err_no = n_ERR_OPNOTSUPP;
2654 rv = 1;
2656 NYD_LEAVE;
2657 return rv;
2660 FL bool_t
2661 temporary_folder_hook_check(bool_t nmail){ /* TODO temporary, v15: drop */
2662 struct a_amv_mac_call_args *amcap;
2663 struct a_amv_mac *amp;
2664 size_t len;
2665 char const *cp;
2666 char *var;
2667 bool_t rv;
2668 NYD_ENTER;
2670 rv = TRU1;
2671 var = n_autorec_alloc(len = strlen(mailname) + sizeof("folder-hook-") -1 +1);
2673 /* First try the fully resolved path */
2674 snprintf(var, len, "folder-hook-%s", mailname);
2675 if((cp = n_var_vlook(var, FAL0)) != NULL)
2676 goto jmac;
2678 /* If we are under *folder*, try the usual +NAME syntax, too */
2679 if(displayname[0] == '+'){
2680 char *x;
2682 for(x = &mailname[len]; x != mailname; --x)
2683 if(x[-1] == '/'){
2684 snprintf(var, len, "folder-hook-+%s", x);
2685 if((cp = n_var_vlook(var, FAL0)) != NULL)
2686 goto jmac;
2687 break;
2691 /* Plain *folder-hook* is our last try */
2692 if((cp = ok_vlook(folder_hook)) == NULL)
2693 goto jleave;
2695 jmac:
2696 if((amp = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) == NULL){
2697 n_err(_("Cannot call *folder-hook* for %s: macro does not exist: %s\n"),
2698 n_shexp_quote_cp(displayname, FAL0), cp);
2699 rv = FAL0;
2700 goto jleave;
2703 amcap = n_lofi_alloc(sizeof *amcap);
2704 memset(amcap, 0, sizeof *amcap);
2705 amcap->amca_name = cp;
2706 amcap->amca_amp = amp;
2707 n_pstate &= ~n_PS_HOOK_MASK;
2708 if(nmail){
2709 amcap->amca_unroller = NULL;
2710 n_pstate |= n_PS_HOOK_NEWMAIL;
2711 }else{
2712 amcap->amca_unroller = &a_amv_folder_hook_lopts;
2713 n_pstate |= n_PS_HOOK;
2715 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2716 amcap->amca_ps_hook_mask = TRU1;
2717 amcap->amca_no_xcall = TRU1;
2718 rv = a_amv_mac_exec(amcap);
2719 n_pstate &= ~n_PS_HOOK_MASK;
2721 jleave:
2722 NYD_LEAVE;
2723 return rv;
2726 FL void
2727 temporary_folder_hook_unroll(void){ /* XXX intermediate hack */
2728 NYD_ENTER;
2729 if(a_amv_folder_hook_lopts != NULL){
2730 void *save = a_amv_lopts;
2732 a_amv_lopts = NULL;
2733 a_amv_lopts_unroll(&a_amv_folder_hook_lopts);
2734 assert(a_amv_folder_hook_lopts == NULL);
2735 a_amv_lopts = save;
2737 NYD_LEAVE;
2740 FL void
2741 temporary_compose_mode_hook_call(char const *macname,
2742 void (*hook_pre)(void *), void *hook_arg){
2743 /* TODO compose_mode_hook_call() temporary, v15: generalize; see a_GO_SPLICE
2744 * TODO comment in go.c for the right way of doing things! */
2745 static struct a_amv_lostack *cmh_losp;
2746 struct a_amv_mac_call_args *amcap;
2747 struct a_amv_mac *amp;
2748 NYD_ENTER;
2750 amp = NULL;
2752 if(macname == (char*)-1){
2753 a_amv_mac__finalize(cmh_losp);
2754 cmh_losp = NULL;
2755 }else if(macname != NULL &&
2756 (amp = a_amv_mac_lookup(macname, NULL, a_AMV_MF_NONE)) == NULL)
2757 n_err(_("Cannot call *on-compose-**: macro does not exist: %s\n"),
2758 macname);
2759 else{
2760 amcap = n_lofi_alloc(sizeof *amcap);
2761 memset(amcap, 0, sizeof *amcap);
2762 amcap->amca_name = (macname != NULL) ? macname
2763 : "*on-compose-splice(-shell)?*";
2764 amcap->amca_amp = amp;
2765 amcap->amca_unroller = &a_amv_compose_lopts;
2766 amcap->amca_hook_pre = hook_pre;
2767 amcap->amca_hook_arg = hook_arg;
2768 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2769 amcap->amca_ps_hook_mask = TRU1;
2770 amcap->amca_no_xcall = TRU1;
2771 n_pstate &= ~n_PS_HOOK_MASK;
2772 n_pstate |= n_PS_HOOK;
2773 if(macname != NULL)
2774 a_amv_mac_exec(amcap);
2775 else{
2776 cmh_losp = n_lofi_alloc(sizeof *cmh_losp);
2777 memset(cmh_losp, 0, sizeof *cmh_losp);
2778 cmh_losp->as_global_saved = a_amv_lopts;
2779 cmh_losp->as_lopts = *(cmh_losp->as_amcap = amcap)->amca_unroller;
2780 cmh_losp->as_loflags = a_AMV_LF_SCOPE_FIXATE;
2781 a_amv_lopts = cmh_losp;
2784 NYD_LEAVE;
2787 FL void
2788 temporary_compose_mode_hook_unroll(void){ /* XXX intermediate hack */
2789 NYD_ENTER;
2790 if(a_amv_compose_lopts != NULL){
2791 void *save = a_amv_lopts;
2793 a_amv_lopts = NULL;
2794 a_amv_lopts_unroll(&a_amv_compose_lopts);
2795 assert(a_amv_compose_lopts == NULL);
2796 a_amv_lopts = save;
2798 NYD_LEAVE;
2801 FL bool_t
2802 n_var_is_user_writable(char const *name){
2803 struct a_amv_var_carrier avc;
2804 struct a_amv_var_map const *avmp;
2805 bool_t rv;
2806 NYD_ENTER;
2808 a_amv_var_revlookup(&avc, name);
2809 if((avmp = avc.avc_map) == NULL)
2810 rv = TRU1;
2811 else
2812 rv = ((avmp->avm_flags & (a_AMV_VF_BOOL | a_AMV_VF_RDONLY)) == 0);
2813 NYD_LEAVE;
2814 return rv;
2817 FL char *
2818 n_var_oklook(enum okeys okey){
2819 struct a_amv_var_carrier avc;
2820 char *rv;
2821 struct a_amv_var_map const *avmp;
2822 NYD_ENTER;
2824 avc.avc_map = avmp = &a_amv_var_map[okey];
2825 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2826 avc.avc_hash = avmp->avm_hash;
2827 avc.avc_okey = okey;
2829 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE))
2830 rv = avc.avc_var->av_value;
2831 else
2832 rv = NULL;
2833 NYD_LEAVE;
2834 return rv;
2837 FL bool_t
2838 n_var_okset(enum okeys okey, uintptr_t val){
2839 struct a_amv_var_carrier avc;
2840 bool_t ok;
2841 struct a_amv_var_map const *avmp;
2842 NYD_ENTER;
2844 avc.avc_map = avmp = &a_amv_var_map[okey];
2845 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2846 avc.avc_hash = avmp->avm_hash;
2847 avc.avc_okey = okey;
2849 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
2850 a_AMV_VSETCLR_NONE);
2851 NYD_LEAVE;
2852 return ok;
2855 FL bool_t
2856 n_var_okclear(enum okeys okey){
2857 struct a_amv_var_carrier avc;
2858 bool_t rv;
2859 struct a_amv_var_map const *avmp;
2860 NYD_ENTER;
2862 avc.avc_map = avmp = &a_amv_var_map[okey];
2863 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2864 avc.avc_hash = avmp->avm_hash;
2865 avc.avc_okey = okey;
2867 rv = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
2868 NYD_LEAVE;
2869 return rv;
2872 FL char const *
2873 n_var_vlook(char const *vokey, bool_t try_getenv){
2874 struct a_amv_var_carrier avc;
2875 char const *rv;
2876 NYD_ENTER;
2878 a_amv_var_revlookup(&avc, vokey);
2880 switch((enum a_amv_var_special_category)avc.avc_special_cat){
2881 default: /* silence CC */
2882 case a_AMV_VSC_NONE:
2883 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_LOCAL))
2884 rv = avc.avc_var->av_value;
2885 /* Only check the environment for something that is otherwise unknown */
2886 else if(try_getenv && avc.avc_map == NULL)
2887 rv = getenv(vokey);
2888 else
2889 rv = NULL;
2890 break;
2891 case a_AMV_VSC_GLOBAL:
2892 rv = a_amv_var_vsc_global(&avc);
2893 break;
2894 case a_AMV_VSC_MULTIPLEX:
2895 rv = a_amv_var_vsc_multiplex(&avc);
2896 break;
2897 case a_AMV_VSC_POSPAR:
2898 case a_AMV_VSC_POSPAR_ENV:
2899 rv = a_amv_var_vsc_pospar(&avc);
2900 break;
2902 NYD_LEAVE;
2903 return rv;
2906 FL bool_t
2907 n_var_vexplode(void const **cookie){
2908 struct a_amv_pospar *appp;
2909 NYD_ENTER;
2911 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
2912 : &a_amv_pospar;
2913 *cookie = (appp->app_count > 0) ? &appp->app_dat[appp->app_idx] : NULL;
2914 NYD_LEAVE;
2915 return (*cookie != NULL);
2918 FL bool_t
2919 n_var_vset(char const *vokey, uintptr_t val){
2920 struct a_amv_var_carrier avc;
2921 bool_t ok;
2922 NYD_ENTER;
2924 a_amv_var_revlookup(&avc, vokey);
2926 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
2927 a_AMV_VSETCLR_NONE);
2928 NYD_LEAVE;
2929 return ok;
2932 FL bool_t
2933 n_var_vclear(char const *vokey){
2934 struct a_amv_var_carrier avc;
2935 bool_t ok;
2936 NYD_ENTER;
2938 a_amv_var_revlookup(&avc, vokey);
2940 ok = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
2941 NYD_LEAVE;
2942 return ok;
2945 #ifdef HAVE_SOCKETS
2946 FL char *
2947 n_var_xoklook(enum okeys okey, struct url const *urlp,
2948 enum okey_xlook_mode oxm){
2949 struct a_amv_var_carrier avc;
2950 struct str const *us;
2951 size_t nlen;
2952 char *nbuf, *rv;
2953 struct a_amv_var_map const *avmp;
2954 NYD_ENTER;
2956 assert(oxm & (OXM_PLAIN | OXM_H_P | OXM_U_H_P));
2958 /* For simplicity: allow this case too */
2959 if(!(oxm & (OXM_H_P | OXM_U_H_P))){
2960 nbuf = NULL;
2961 goto jplain;
2964 avc.avc_map = avmp = &a_amv_var_map[okey];
2965 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2966 avc.avc_okey = okey;
2968 us = (oxm & OXM_U_H_P) ? &urlp->url_u_h_p : &urlp->url_h_p;
2969 nlen = strlen(avc.avc_name);
2970 nbuf = n_lofi_alloc(nlen + 1 + us->l +1);
2971 memcpy(nbuf, avc.avc_name, nlen);
2973 /* One of .url_u_h_p and .url_h_p we test in here */
2974 /*avc.avc_is_chain_variant = TRU1;*/
2975 nbuf[nlen++] = '-';
2976 memcpy(&nbuf[nlen], us->s, us->l +1);
2977 avc.avc_name = nbuf;
2978 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2979 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE))
2980 goto jvar;
2982 /* The second */
2983 if((oxm & (OXM_U_H_P | OXM_H_P)) == (OXM_U_H_P | OXM_H_P)){
2984 us = &urlp->url_h_p;
2985 memcpy(&nbuf[nlen], us->s, us->l +1);
2986 avc.avc_name = nbuf;
2987 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2988 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE)){
2989 jvar:
2990 rv = avc.avc_var->av_value;
2991 goto jleave;
2995 jplain:
2996 /*avc.avc_is_chain_variant = FAL0;*/
2997 rv = (oxm & OXM_PLAIN) ? n_var_oklook(okey) : NULL;
2998 jleave:
2999 if(nbuf != NULL)
3000 n_lofi_free(nbuf);
3001 NYD_LEAVE;
3002 return rv;
3004 #endif /* HAVE_SOCKETS */
3006 FL int
3007 c_set(void *vp){
3008 int err;
3009 char **ap;
3010 NYD_ENTER;
3012 if(*(ap = vp) == NULL){
3013 a_amv_var_show_all();
3014 err = 0;
3015 }else{
3016 enum a_amv_var_setclr_flags avscf;
3018 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
3019 avscf = a_AMV_VSETCLR_NONE;
3020 else{
3021 if(a_amv_lopts == NULL){
3022 n_err(_("`set': cannot use `local' in this context\n"));
3023 err = 1;
3024 goto jleave;
3026 avscf = a_AMV_VSETCLR_LOCAL;
3028 err = !a_amv_var_c_set(ap, avscf);
3030 jleave:
3031 NYD_LEAVE;
3032 return err;
3035 FL int
3036 c_unset(void *vp){
3037 struct a_amv_var_carrier avc;
3038 char **ap;
3039 int err;
3040 enum a_amv_var_setclr_flags avscf;
3041 NYD_ENTER;
3043 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
3044 avscf = a_AMV_VSETCLR_NONE;
3045 else{
3046 if(a_amv_lopts == NULL){
3047 n_err(_("`unset': cannot use `local' in this context\n"));
3048 err = 1;
3049 goto jleave;
3051 avscf = a_AMV_VSETCLR_LOCAL;
3054 for(err = 0, ap = vp; *ap != NULL; ++ap){
3055 a_amv_var_revlookup(&avc, *ap);
3057 err |= !a_amv_var_clear(&avc, avscf);
3059 jleave:
3060 NYD_LEAVE;
3061 return err;
3064 FL int
3065 c_varshow(void *v){
3066 char **ap;
3067 NYD_ENTER;
3069 if(*(ap = v) == NULL)
3070 v = NULL;
3071 else{
3072 struct n_string msg, *msgp = &msg;
3074 msgp = n_string_creat(msgp);
3075 for(; *ap != NULL; ++ap)
3076 a_amv_var_show(*ap, n_stdout, msgp);
3077 n_string_gut(msgp);
3079 NYD_LEAVE;
3080 return (v == NULL ? !STOP : !OKAY); /* xxx 1:bad 0:good -- do some */
3083 FL int
3084 c_varedit(void *v){ /* TODO v15 drop */
3085 struct a_amv_var_carrier avc;
3086 FILE *of, *nf;
3087 char *val, **argv;
3088 int err;
3089 sighandler_type sigint;
3090 NYD_ENTER;
3092 sigint = safe_signal(SIGINT, SIG_IGN);
3094 for(err = 0, argv = v; *argv != NULL; ++argv){
3095 memset(&avc, 0, sizeof avc);
3097 a_amv_var_revlookup(&avc, *argv);
3099 if(avc.avc_map != NULL){
3100 if(avc.avc_map->avm_flags & a_AMV_VF_BOOL){
3101 n_err(_("`varedit': cannot edit boolean variable: %s\n"),
3102 avc.avc_name);
3103 continue;
3105 if(avc.avc_map->avm_flags & a_AMV_VF_RDONLY){
3106 n_err(_("`varedit': cannot edit read-only variable: %s\n"),
3107 avc.avc_name);
3108 continue;
3112 a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
3114 if((of = Ftmp(NULL, "varedit", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
3115 NULL){
3116 n_perr(_("`varedit': cannot create temporary file"), 0);
3117 err = 1;
3118 break;
3119 }else if(avc.avc_var != NULL && *(val = avc.avc_var->av_value) != '\0' &&
3120 sizeof *val != fwrite(val, strlen(val), sizeof *val, of)){
3121 n_perr(_("`varedit' failed to write old value to temporary file"), 0);
3122 Fclose(of);
3123 err = 1;
3124 continue;
3127 fflush_rewind(of);
3128 nf = n_run_editor(of, (off_t)-1, 'e', FAL0, NULL,NULL, SEND_MBOX, sigint,
3129 NULL);
3130 Fclose(of);
3132 if(nf != NULL){
3133 int c;
3134 char const *varres;
3135 off_t l;
3137 l = fsize(nf);
3138 if(UICMP(64, l, >=, UIZ_MAX -42)){
3139 n_err(_("`varedit': not enough memory to store variable: %s\n"),
3140 avc.avc_name);
3141 varres = n_empty;
3142 err = 1;
3143 }else{
3144 varres = val = n_autorec_alloc(l +1);
3145 for(; l > 0 && (c = getc(nf)) != EOF; --l)
3146 *val++ = c;
3147 *val++ = '\0';
3148 if(l != 0){
3149 n_err(_("`varedit': I/O while reading new value of: %s\n"),
3150 avc.avc_name);
3151 err = 1;
3155 if(!a_amv_var_set(&avc, varres, a_AMV_VSETCLR_NONE))
3156 err = 1;
3158 Fclose(nf);
3159 }else{
3160 n_err(_("`varedit': cannot start $EDITOR\n"));
3161 err = 1;
3162 break;
3166 safe_signal(SIGINT, sigint);
3167 NYD_LEAVE;
3168 return err;
3171 FL int
3172 c_environ(void *v){
3173 struct a_amv_var_carrier avc;
3174 int err;
3175 char **ap;
3176 bool_t islnk;
3177 NYD_ENTER;
3179 if((islnk = is_asccaseprefix(*(ap = v), "link")) ||
3180 is_asccaseprefix(*ap, "unlink")){
3181 for(err = 0; *++ap != NULL;){
3182 a_amv_var_revlookup(&avc, *ap);
3184 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE) && (islnk ||
3185 (avc.avc_var->av_flags & a_AMV_VF_EXT_LINKED))){
3186 if(!islnk){
3187 avc.avc_var->av_flags &= ~a_AMV_VF_EXT_LINKED;
3188 continue;
3189 }else if(avc.avc_var->av_flags &
3190 (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)){
3191 if(n_poption & n_PO_D_V)
3192 n_err(_("`environ': link: already established: %s\n"), *ap);
3193 continue;
3195 avc.avc_var->av_flags |= a_AMV_VF_EXT_LINKED;
3196 if(!(avc.avc_var->av_flags & a_AMV_VF_ENV))
3197 a_amv_var__putenv(&avc, avc.avc_var);
3198 }else if(!islnk){
3199 n_err(_("`environ': unlink: no link established: %s\n"), *ap);
3200 err = 1;
3201 }else{
3202 char const *evp = getenv(*ap);
3204 if(evp != NULL)
3205 err |= !a_amv_var_set(&avc, evp, a_AMV_VSETCLR_ENV);
3206 else{
3207 n_err(_("`environ': link: cannot link to non-existent: %s\n"),
3208 *ap);
3209 err = 1;
3213 }else if(is_asccaseprefix(*ap, "set"))
3214 err = !a_amv_var_c_set(++ap, a_AMV_VSETCLR_ENV);
3215 else if(is_asccaseprefix(*ap, "unset")){
3216 for(err = 0; *++ap != NULL;){
3217 a_amv_var_revlookup(&avc, *ap);
3219 if(!a_amv_var_clear(&avc, a_AMV_VSETCLR_ENV))
3220 err = 1;
3222 }else{
3223 n_err(_("Synopsis: environ: <link|set|unset> <variable>...\n"));
3224 err = 1;
3226 NYD_LEAVE;
3227 return err;
3230 FL int
3231 c_vexpr(void *v){ /* TODO POSIX expr(1) comp. exit status; overly complicat. */
3232 char pbase, op, iencbuf[2+1/* BASE# prefix*/ + n_IENC_BUFFER_SIZE + 1];
3233 size_t i;
3234 enum n_idec_state ids;
3235 enum n_idec_mode idm;
3236 si64_t lhv, rhv;
3237 char const **argv, *varname, *varres, *cp;
3238 enum{
3239 a_ERR = 1u<<0,
3240 a_SOFTOVERFLOW = 1u<<1,
3241 a_ISNUM = 1u<<2,
3242 a_ISDECIMAL = 1u<<3, /* Print only decimal result */
3243 a_SATURATED = 1u<<4,
3244 a_ICASE = 1u<<5,
3245 a_UNSIGNED_OP = 1u<<6, /* Unsigned right shift (share bit ok) */
3246 a_PBASE = 1u<<7, /* Print additional number base */
3247 a_TMP = 1u<<30
3248 } f;
3249 NYD_ENTER;
3251 f = a_ERR;
3252 argv = v;
3253 varname = (n_pstate & n_PS_ARGMOD_VPUT) ? *argv++ : NULL;
3254 n_UNINIT(varres, n_empty);
3255 n_UNINIT(pbase, '\0');
3257 if((cp = argv[0])[0] == '\0')
3258 goto jesubcmd;
3260 if(cp[1] == '\0'){
3261 op = cp[0];
3262 jnumop:
3263 f |= a_ISNUM;
3264 switch(op){
3265 case '=':
3266 case '~':
3267 if(argv[1] == NULL || argv[2] != NULL)
3268 goto jesynopsis;
3270 if(*(cp = *++argv) == '\0')
3271 lhv = 0;
3272 else{
3273 idm = ((*cp == 'u' || *cp == 'U')
3274 ? (++cp, n_IDEC_MODE_NONE)
3275 : ((*cp == 's' || *cp == 'S')
3276 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3277 : n_IDEC_MODE_SIGNED_TYPE |
3278 n_IDEC_MODE_POW2BASE_UNSIGNED));
3279 if(((ids = n_idec_cp(&lhv, cp, 0, idm, NULL)
3280 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3281 ) != n_IDEC_STATE_CONSUMED){
3282 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3283 goto jenum_range;
3284 f |= a_SOFTOVERFLOW;
3285 break;
3288 if(op == '~')
3289 lhv = ~lhv;
3290 break;
3292 case '+':
3293 case '-':
3294 case '*':
3295 case '/':
3296 case '%':
3297 case '|':
3298 case '&':
3299 case '^':
3300 case '<':
3301 case '>':
3302 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3303 goto jesynopsis;
3304 else{
3305 char xop;
3307 if(*(cp = *++argv) == '\0')
3308 lhv = 0;
3309 else{
3310 idm = ((*cp == 'u' || *cp == 'U')
3311 ? (++cp, n_IDEC_MODE_NONE)
3312 : ((*cp == 's' || *cp == 'S')
3313 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3314 : n_IDEC_MODE_SIGNED_TYPE |
3315 n_IDEC_MODE_POW2BASE_UNSIGNED));
3316 if(((ids = n_idec_cp(&lhv, cp, 0, idm, NULL)
3317 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3318 ) != n_IDEC_STATE_CONSUMED){
3319 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3320 goto jenum_range;
3321 f |= a_SOFTOVERFLOW;
3322 break;
3326 if(*(cp = *++argv) == '\0')
3327 rhv = 0;
3328 else{
3329 idm = ((*cp == 'u' || *cp == 'U')
3330 ? (++cp, n_IDEC_MODE_NONE)
3331 : ((*cp == 's' || *cp == 'S')
3332 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3333 : n_IDEC_MODE_SIGNED_TYPE |
3334 n_IDEC_MODE_POW2BASE_UNSIGNED));
3335 if(((ids = n_idec_cp(&rhv, cp, 0, idm, NULL)
3336 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3337 ) != n_IDEC_STATE_CONSUMED){
3338 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3339 goto jenum_range;
3340 f |= a_SOFTOVERFLOW;
3341 lhv = rhv;
3342 break;
3346 xop = op;
3347 jnumop_again:
3348 switch(xop){
3349 case '+':
3350 if(rhv < 0){
3351 if(rhv != SI64_MIN){
3352 rhv = -rhv;
3353 xop = '-';
3354 goto jnumop_again;
3355 }else if(lhv < 0)
3356 goto jenum_plusminus;
3357 else if(lhv == 0){
3358 lhv = rhv;
3359 break;
3361 }else if(SI64_MAX - rhv < lhv)
3362 goto jenum_plusminus;
3363 lhv += rhv;
3364 break;
3365 case '-':
3366 if(rhv < 0){
3367 if(rhv != SI64_MIN){
3368 rhv = -rhv;
3369 xop = '+';
3370 goto jnumop_again;
3371 }else if(lhv > 0)
3372 goto jenum_plusminus;
3373 else if(lhv == 0){
3374 lhv = rhv;
3375 break;
3377 }else if(SI64_MIN + rhv > lhv){
3378 jenum_plusminus:
3379 if(!(f & a_SATURATED))
3380 goto jenum_overflow;
3381 f |= a_SOFTOVERFLOW;
3382 lhv = (lhv < 0 || xop == '-') ? SI64_MIN : SI64_MAX;
3383 break;
3385 lhv -= rhv;
3386 break;
3387 case '*':
3388 /* Will the result be positive? */
3389 if((lhv < 0) == (rhv < 0)){
3390 if(lhv > 0){
3391 lhv = -lhv;
3392 rhv = -rhv;
3394 if(rhv != 0 && lhv != 0 && SI64_MAX / rhv > lhv){
3395 if(!(f & a_SATURATED))
3396 goto jenum_overflow;
3397 f |= a_SOFTOVERFLOW;
3398 lhv = SI64_MAX;
3399 }else
3400 lhv *= rhv;
3401 }else{
3402 if(rhv > 0){
3403 if(lhv != 0 && SI64_MIN / lhv < rhv){
3404 if(!(f & a_SATURATED))
3405 goto jenum_overflow;
3406 f |= a_SOFTOVERFLOW;
3407 lhv = SI64_MIN;
3408 }else
3409 lhv *= rhv;
3410 }else{
3411 if(rhv != 0 && lhv != 0 && SI64_MIN / rhv < lhv){
3412 if(!(f & a_SATURATED))
3413 goto jenum_overflow;
3414 f |= a_SOFTOVERFLOW;
3415 lhv = SI64_MIN;
3416 }else
3417 lhv *= rhv;
3420 break;
3421 case '/':
3422 if(rhv == 0){
3423 if(!(f & a_SATURATED))
3424 goto jenum_range;
3425 f |= a_SOFTOVERFLOW;
3426 lhv = SI64_MAX;
3427 }else
3428 lhv /= rhv;
3429 break;
3430 case '%':
3431 if(rhv == 0){
3432 if(!(f & a_SATURATED))
3433 goto jenum_range;
3434 f |= a_SOFTOVERFLOW;
3435 lhv = SI64_MAX;
3436 }else
3437 lhv %= rhv;
3438 break;
3439 case '|':
3440 lhv |= rhv;
3441 break;
3442 case '&':
3443 lhv &= rhv;
3444 break;
3445 case '^':
3446 lhv ^= rhv;
3447 break;
3448 case '<':
3449 case '>':
3450 if(!(f & a_TMP)){
3451 argv -= 2;
3452 goto jesubcmd;
3454 if(rhv > 63){ /* xxx 63? */
3455 if(!(f & a_SATURATED))
3456 goto jenum_overflow;
3457 rhv = 63;
3459 if(op == '<')
3460 lhv <<= (ui8_t)rhv;
3461 else if(f & a_UNSIGNED_OP)
3462 lhv = (ui64_t)lhv >> (ui8_t)rhv;
3463 else
3464 lhv >>= (ui8_t)rhv;
3465 break;
3468 break;
3469 default:
3470 goto jesubcmd;
3472 }else if(cp[0] == '<'){
3473 if(*++cp != '<')
3474 goto jesubcmd;
3475 if(*++cp == '@'){
3476 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3477 f |= a_SATURATED;
3478 ++cp;
3480 if(*cp != '\0')
3481 goto jesubcmd;
3482 f |= a_TMP;
3483 op = '<';
3484 goto jnumop;
3485 }else if(cp[0] == '>'){
3486 if(*++cp != '>')
3487 goto jesubcmd;
3488 if(*++cp == '>'){
3489 f |= a_UNSIGNED_OP;
3490 ++cp;
3492 if(*cp == '@'){
3493 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3494 f |= a_SATURATED;
3495 ++cp;
3497 if(*cp != '\0')
3498 goto jesubcmd;
3499 f |= a_TMP;
3500 op = '>';
3501 goto jnumop;
3502 }else if(cp[2] == '\0' && cp[1] == '@'){
3503 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix, not suffix"));
3504 f |= a_SATURATED;
3505 op = cp[0];
3506 goto jnumop;
3507 }else if(cp[0] == '@'){
3508 f |= a_SATURATED;
3509 op = *++cp;
3510 if(*++cp != '\0'){
3511 if(op != *cp)
3512 goto jesubcmd;
3513 switch(op){
3514 case '<':
3515 if(*++cp != '\0')
3516 goto jesubcmd;
3517 f |= a_TMP;
3518 break;
3519 case '>':
3520 if(*++cp != '\0'){
3521 if(*cp != '>' || *++cp != '\0')
3522 goto jesubcmd;
3523 f |= a_UNSIGNED_OP;
3525 f |= a_TMP;
3526 break;
3527 default:
3528 goto jesubcmd;
3531 goto jnumop;
3532 }else if(is_asccaseprefix(cp, "pbase")){
3533 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3534 goto jesynopsis;
3536 if(((ids = n_idec_si8_cp(&pbase, argv[1], 0, NULL)
3537 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3538 ) != n_IDEC_STATE_CONSUMED || pbase < 2 || pbase > 36)
3539 goto jenum_range;
3541 f |= a_PBASE;
3542 op = *(argv[0] = cp = "=");
3543 argv[1] = argv[2];
3544 argv[2] = NULL;
3545 goto jnumop;
3546 }else if(is_asccaseprefix(cp, "length")){
3547 f |= a_ISNUM | a_ISDECIMAL;
3548 if(argv[1] == NULL || argv[2] != NULL)
3549 goto jesynopsis;
3551 i = strlen(*++argv);
3552 if(UICMP(64, i, >, SI64_MAX))
3553 goto jestr_overflow;
3554 lhv = (si64_t)i;
3555 }else if(is_asccaseprefix(cp, "hash")){
3556 f |= a_ISNUM | a_ISDECIMAL;
3557 if(argv[1] == NULL || argv[2] != NULL)
3558 goto jesynopsis;
3560 i = n_torek_hash(*++argv);
3561 lhv = (si64_t)i;
3562 }else if(is_asccaseprefix(cp, "find")){
3563 f |= a_ISNUM | a_ISDECIMAL;
3564 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3565 goto jesynopsis;
3567 if((cp = strstr(argv[1], argv[2])) == NULL)
3568 goto jestr_nodata;
3569 i = PTR2SIZE(cp - argv[1]);
3570 if(UICMP(64, i, >, SI64_MAX))
3571 goto jestr_overflow;
3572 lhv = (si64_t)i;
3573 }else if(is_asccaseprefix(cp, "ifind")){
3574 f |= a_ISNUM | a_ISDECIMAL;
3575 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3576 goto jesynopsis;
3578 if((cp = asccasestr(argv[1], argv[2])) == NULL)
3579 goto jestr_nodata;
3580 i = PTR2SIZE(cp - argv[1]);
3581 if(UICMP(64, i, >, SI64_MAX))
3582 goto jestr_overflow;
3583 lhv = (si64_t)i;
3584 }else if(is_asccaseprefix(cp, "substring")){
3585 if(argv[1] == NULL || argv[2] == NULL)
3586 goto jesynopsis;
3587 if(argv[3] != NULL && argv[4] != NULL)
3588 goto jesynopsis;
3590 i = strlen(varres = *++argv);
3592 if(*(cp = *++argv) == '\0')
3593 lhv = 0;
3594 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3595 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3596 ) != n_IDEC_STATE_CONSUMED)
3597 goto jestr_numrange;
3598 if(lhv < 0){
3599 if(UICMP(64, i, <, -lhv))
3600 goto jesubstring_off;
3601 lhv = i + lhv;
3603 if(UICMP(64, i, >=, lhv)){
3604 i -= lhv;
3605 varres += lhv;
3606 }else{
3607 jesubstring_off:
3608 if(n_poption & n_PO_D_V)
3609 n_err(_("`vexpr': substring: offset argument too large: %s\n"),
3610 n_shexp_quote_cp(argv[-1], FAL0));
3611 f |= a_SOFTOVERFLOW;
3614 if(argv[1] != NULL){
3615 if(*(cp = *++argv) == '\0')
3616 lhv = 0;
3617 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3618 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3619 ) != n_IDEC_STATE_CONSUMED)
3620 goto jestr_numrange;
3621 if(lhv < 0){
3622 if(UICMP(64, i, <, -lhv))
3623 goto jesubstring_len;
3624 lhv = i + lhv;
3626 if(UICMP(64, i, >=, lhv)){
3627 if(UICMP(64, i, !=, lhv))
3628 varres = savestrbuf(varres, (size_t)lhv);
3629 }else{
3630 jesubstring_len:
3631 if(n_poption & n_PO_D_V)
3632 n_err(_("`vexpr': substring: length argument too large: %s\n"),
3633 n_shexp_quote_cp(argv[-2], FAL0));
3634 f |= a_SOFTOVERFLOW;
3637 }else if(is_asccaseprefix(cp, "trim")) Jtrim: {
3638 struct str trim;
3639 enum n_str_trim_flags stf;
3641 if(argv[1] == NULL || argv[2] != NULL)
3642 goto jesynopsis;
3644 if(is_asccaseprefix(cp, "trim-front"))
3645 stf = n_STR_TRIM_FRONT;
3646 else if(is_asccaseprefix(cp, "trim-end"))
3647 stf = n_STR_TRIM_END;
3648 else
3649 stf = n_STR_TRIM_BOTH;
3651 trim.l = strlen(trim.s = n_UNCONST(argv[1]));
3652 (void)n_str_trim(&trim, stf);
3653 varres = savestrbuf(trim.s, trim.l);
3654 }else if(is_asccaseprefix(cp, "trim-front"))
3655 goto Jtrim;
3656 else if(is_asccaseprefix(cp, "trim-end"))
3657 goto Jtrim;
3658 else if(is_asccaseprefix(cp, "random")){
3659 if(argv[1] == NULL || argv[2] != NULL)
3660 goto jesynopsis;
3662 if((n_idec_si64_cp(&lhv, argv[1], 0, NULL
3663 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3664 ) != n_IDEC_STATE_CONSUMED || lhv < 0 || lhv > PATH_MAX)
3665 goto jestr_numrange;
3666 if(lhv == 0)
3667 lhv = NAME_MAX;
3668 varres = n_random_create_cp((size_t)lhv, NULL);
3669 }else if(is_asccaseprefix(cp, "file-expand")){
3670 if(argv[1] == NULL || argv[2] != NULL)
3671 goto jesynopsis;
3673 if((varres = fexpand(argv[1], FEXP_NVAR | FEXP_NOPROTO)) == NULL)
3674 goto jestr_nodata;
3675 }else if(is_asccaseprefix(cp, "makeprint")){
3676 struct str sin, sout;
3678 if(argv[1] == NULL || argv[2] != NULL)
3679 goto jesynopsis;
3681 /* XXX using strlen for `vexpr makeprint' is wrong for UTF-16 */
3682 sin.l = strlen(sin.s = n_UNCONST(argv[1]));
3683 makeprint(&sin, &sout);
3684 varres = savestrbuf(sout.s, sout.l);
3685 n_free(sout.s);
3686 /* TODO `vexpr': (wide) string length, find, etc!! */
3687 #ifdef HAVE_REGEX
3688 }else if(is_asccaseprefix(cp, "regex")) Jregex:{
3689 regmatch_t rema[1 + n_VEXPR_REGEX_MAX];
3690 regex_t re;
3691 int reflrv;
3693 f |= a_ISNUM | a_ISDECIMAL;
3694 if(argv[1] == NULL || argv[2] == NULL ||
3695 (argv[3] != NULL && argv[4] != NULL))
3696 goto jesynopsis;
3698 reflrv = REG_EXTENDED;
3699 if(f & a_ICASE)
3700 reflrv |= REG_ICASE;
3701 if((reflrv = regcomp(&re, argv[2], reflrv))){
3702 n_err(_("`vexpr': invalid regular expression: %s: %s\n"),
3703 n_shexp_quote_cp(argv[2], FAL0), n_regex_err_to_doc(NULL, reflrv));
3704 assert(f & a_ERR);
3705 n_pstate_err_no = n_ERR_INVAL;
3706 goto jestr;
3708 reflrv = regexec(&re, argv[1], n_NELEM(rema), rema, 0);
3709 regfree(&re);
3710 if(reflrv == REG_NOMATCH)
3711 goto jestr_nodata;
3713 /* Search only? Else replace, which is a bit */
3714 if(argv[3] == NULL){
3715 if(UICMP(64, rema[0].rm_so, >, SI64_MAX))
3716 goto jestr_overflow;
3717 lhv = (si64_t)rema[0].rm_so;
3718 }else{
3719 /* We need to setup some kind of pseudo macro environment for this */
3720 struct a_amv_lostack los;
3721 struct a_amv_mac_call_args amca;
3722 char const **reargv;
3724 memset(&amca, 0, sizeof amca);
3725 amca.amca_name = savestrbuf(&argv[1][rema[0].rm_so],
3726 rema[0].rm_eo - rema[0].rm_so);
3727 amca.amca_amp = a_AMV_MACKY_MACK;
3728 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3730 amca.amca_pospar.app_count = (ui32_t)i - 1;
3731 amca.amca_pospar.app_not_heap = TRU1;
3732 amca.amca_pospar.app_dat =
3733 reargv = n_autorec_alloc(sizeof(char*) * i);
3734 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3735 *reargv++ = savestrbuf(&argv[1][rema[i].rm_so],
3736 rema[i].rm_eo - rema[i].rm_so);
3737 *reargv = NULL;
3739 memset(&los, 0, sizeof los);
3740 hold_all_sigs(); /* TODO DISLIKE! */
3741 los.as_global_saved = a_amv_lopts;
3742 los.as_amcap = &amca;
3743 los.as_up = los.as_global_saved;
3744 a_amv_lopts = &los;
3746 /* C99 */{
3747 struct str templ;
3748 struct n_string s_b;
3749 enum n_shexp_state shs;
3751 templ.s = n_UNCONST(argv[3]);
3752 templ.l = UIZ_MAX;
3753 shs = n_shexp_parse_token((n_SHEXP_PARSE_LOG |
3754 n_SHEXP_PARSE_IGNORE_EMPTY | n_SHEXP_PARSE_QUOTE_AUTO_FIXED |
3755 n_SHEXP_PARSE_QUOTE_AUTO_DSQ),
3756 n_string_creat_auto(&s_b), &templ, NULL);
3757 if((shs & (n_SHEXP_STATE_ERR_MASK | n_SHEXP_STATE_STOP)
3758 ) == n_SHEXP_STATE_STOP){
3759 varres = n_string_cp(&s_b);
3760 n_string_drop_ownership(&s_b);
3761 }else
3762 varres = NULL;
3765 a_amv_lopts = los.as_global_saved;
3766 rele_all_sigs(); /* TODO DISLIKE! */
3768 if(varres == NULL)
3769 goto jestr_nodata;
3770 f &= ~(a_ISNUM | a_ISDECIMAL);
3772 }else if(is_asccaseprefix(argv[0], "iregex")){
3773 f |= a_ICASE;
3774 goto Jregex;
3775 #endif /* HAVE_REGEX */
3776 }else
3777 goto jesubcmd;
3779 n_pstate_err_no = (f & a_SOFTOVERFLOW) ? n_ERR_OVERFLOW : n_ERR_NONE;
3780 f &= ~a_ERR;
3782 /* Generate the variable value content for numerics.
3783 * Anticipate in our handling below! (Don't do needless work) */
3784 jleave:
3785 if((f & a_ISNUM) && ((f & (a_ISDECIMAL | a_PBASE)) || varname != NULL)){
3786 cp = n_ienc_buf(iencbuf, lhv, (f & a_PBASE ? pbase : 10),
3787 n_IENC_MODE_SIGNED_TYPE);
3788 if(cp != NULL)
3789 varres = cp;
3790 else{
3791 f |= a_ERR;
3792 varres = n_empty;
3796 if(varname == NULL){
3797 /* If there was no error and we are printing a numeric result, print some
3798 * more bases for the fun of it */
3799 if((f & (a_ERR | a_ISNUM | a_ISDECIMAL)) == a_ISNUM){
3800 char binabuf[64 + 64 / 8 +1];
3801 size_t j;
3803 for(j = 1, i = 0; i < 64; ++i){
3804 binabuf[63 + 64 / 8 -j - i] = (lhv & ((ui64_t)1 << i)) ? '1' : '0';
3805 if((i & 7) == 7 && i != 63){
3806 ++j;
3807 binabuf[63 + 64 / 8 -j - i] = ' ';
3810 binabuf[64 + 64 / 8 -1] = '\0';
3812 if(fprintf(n_stdout,
3813 "0b %s\n0%" PRIo64 " | 0x%" PRIX64 " | %" PRId64 "\n",
3814 binabuf, lhv, lhv, lhv) < 0 ||
3815 ((f & a_PBASE) && (assert(varres != NULL),
3816 fprintf(n_stdout, "%s\n", varres) < 0))){
3817 n_pstate_err_no = n_err_no;
3818 f |= a_ERR;
3820 }else if(varres != NULL && fprintf(n_stdout, "%s\n", varres) < 0){
3821 n_pstate_err_no = n_err_no;
3822 f |= a_ERR;
3824 }else if(!n_var_vset(varname, (uintptr_t)varres)){
3825 n_pstate_err_no = n_ERR_NOTSUP;
3826 f |= a_ERR;
3828 NYD_LEAVE;
3829 return (f & a_ERR) ? 1 : 0;
3831 jerr:
3832 f = a_ERR | a_ISNUM;
3833 lhv = -1;
3834 goto jleave;
3835 jesubcmd:
3836 n_err(_("`vexpr': invalid subcommand: %s\n"),
3837 n_shexp_quote_cp(*argv, FAL0));
3838 n_pstate_err_no = n_ERR_INVAL;
3839 goto jerr;
3840 jesynopsis:
3841 n_err(_("Synopsis: vexpr: <operator> <:argument:>\n"));
3842 n_pstate_err_no = n_ERR_INVAL;
3843 goto jerr;
3845 jenum_range:
3846 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3847 n_shexp_quote_cp(*argv, FAL0));
3848 n_pstate_err_no = n_ERR_RANGE;
3849 goto jerr;
3850 jenum_overflow:
3851 n_err(_("`vexpr': expression overflows datatype: %" PRId64 " %c %" PRId64
3852 "\n"), lhv, op, rhv);
3853 n_pstate_err_no = n_ERR_OVERFLOW;
3854 goto jerr;
3856 jestr_numrange:
3857 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3858 n_shexp_quote_cp(*argv, FAL0));
3859 n_pstate_err_no = n_ERR_RANGE;
3860 goto jestr;
3861 jestr_overflow:
3862 n_err(_("`vexpr': string length or offset overflows datatype\n"));
3863 n_pstate_err_no = n_ERR_OVERFLOW;
3864 goto jestr;
3865 jestr_nodata:
3866 n_pstate_err_no = n_ERR_NODATA;
3867 /* FALLTHRU */
3868 jestr:
3869 varres = n_empty;
3870 f &= ~a_ISNUM;
3871 f |= a_ERR;
3872 goto jleave;
3875 FL int
3876 c_vpospar(void *v){
3877 struct n_cmd_arg *cap;
3878 size_t i;
3879 struct a_amv_pospar *appp;
3880 enum{
3881 a_NONE = 0,
3882 a_ERR = 1u<<0,
3883 a_SET = 1u<<1,
3884 a_CLEAR = 1u<<2,
3885 a_QUOTE = 1u<<3
3886 } f;
3887 char const *varres;
3888 struct n_cmd_arg_ctx *cacp;
3889 NYD_ENTER;
3891 n_pstate_err_no = n_ERR_NONE;
3892 n_UNINIT(varres, n_empty);
3893 cacp = v;
3894 cap = cacp->cac_arg;
3896 if(is_asccaseprefix(cap->ca_arg.ca_str.s, "set"))
3897 f = a_SET;
3898 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "clear"))
3899 f = a_CLEAR;
3900 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "quote"))
3901 f = a_QUOTE;
3902 else{
3903 n_err(_("`vpospar': invalid subcommand: %s\n"),
3904 n_shexp_quote_cp(cap->ca_arg.ca_str.s, FAL0));
3905 n_pstate_err_no = n_ERR_INVAL;
3906 f = a_ERR;
3907 goto jleave;
3909 --cacp->cac_no;
3911 if((f & (a_CLEAR | a_QUOTE)) && cap->ca_next != NULL){
3912 n_err(_("`vpospar': `%s': takes no argument\n"), cap->ca_arg.ca_str.s);
3913 n_pstate_err_no = n_ERR_INVAL;
3914 f = a_ERR;
3915 goto jleave;
3918 cap = cap->ca_next;
3920 /* If in a macro, we need to overwrite the local instead of global argv */
3921 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
3922 : &a_amv_pospar;
3924 if(f & (a_SET | a_CLEAR)){
3925 if(cacp->cac_vput != NULL)
3926 n_err(_("`vpospar': `vput' only supported for `quote' subcommand\n"));
3927 if(!appp->app_not_heap && appp->app_maxcount > 0){
3928 for(i = appp->app_maxcount; i-- != 0;)
3929 n_free(n_UNCONST(appp->app_dat[i]));
3930 n_free(appp->app_dat);
3932 memset(appp, 0, sizeof *appp);
3934 if(f & a_SET){
3935 if((i = cacp->cac_no) > a_AMV_POSPAR_MAX){
3936 n_err(_("`vpospar': overflow: %" PRIuZ " arguments!\n"), i);
3937 n_pstate_err_no = n_ERR_OVERFLOW;
3938 f = a_ERR;
3939 goto jleave;
3942 memset(appp, 0, sizeof *appp);
3943 if(i > 0){
3944 appp->app_maxcount = appp->app_count = (ui16_t)i;
3945 /* XXX Optimize: store it all in one chunk! */
3946 ++i;
3947 i *= sizeof *appp->app_dat;
3948 appp->app_dat = n_alloc(i);
3950 for(i = 0; cap != NULL; ++i, cap = cap->ca_next){
3951 appp->app_dat[i] = n_alloc(cap->ca_arg.ca_str.l +1);
3952 memcpy(n_UNCONST(appp->app_dat[i]), cap->ca_arg.ca_str.s,
3953 cap->ca_arg.ca_str.l +1);
3956 appp->app_dat[i] = NULL;
3959 }else{
3960 if(appp->app_count == 0)
3961 varres = n_empty;
3962 else{
3963 struct str in;
3964 struct n_string s, *sp;
3965 char sep1, sep2;
3967 sp = n_string_creat_auto(&s);
3969 sep1 = *ok_vlook(ifs);
3970 sep2 = *ok_vlook(ifs_ws);
3971 if(sep1 == sep2)
3972 sep2 = '\0';
3973 if(sep1 == '\0')
3974 sep1 = ' ';
3976 for(i = 0; i < appp->app_count; ++i){
3977 if(sp->s_len){
3978 if(!n_string_can_book(sp, 2))
3979 goto jeover;
3980 sp = n_string_push_c(sp, sep1);
3981 if(sep2 != '\0')
3982 sp = n_string_push_c(sp, sep2);
3984 in.l = strlen(in.s = n_UNCONST(appp->app_dat[i + appp->app_idx]));
3986 if(!n_string_can_book(sp, in.l)){
3987 jeover:
3988 n_err(_("`vpospar': overflow: string too long!\n"));
3989 n_pstate_err_no = n_ERR_OVERFLOW;
3990 f = a_ERR;
3991 goto jleave;
3993 sp = n_shexp_quote(sp, &in, TRU1);
3996 varres = n_string_cp(sp);
3999 if(cacp->cac_vput == NULL){
4000 if(fprintf(n_stdout, "%s\n", varres) < 0){
4001 n_pstate_err_no = n_err_no;
4002 f |= a_ERR;
4004 }else if(!n_var_vset(cacp->cac_vput, (uintptr_t)varres)){
4005 n_pstate_err_no = n_ERR_NOTSUP;
4006 f |= a_ERR;
4009 jleave:
4010 NYD_LEAVE;
4011 return (f & a_ERR) ? 1 : 0;
4014 /* s-it-mode */