`vexpr': obsolete @ as suffix, it must henceforth be prefix..
[s-mailx.git] / accmacvar.c
blob692c7bb5ff239bd12445f33b9fe000031655a5aa
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 enum a_amv_var_flags{
99 a_AMV_VF_NONE = 0,
101 /* The basic set of flags, also present in struct a_amv_var_map.avm_flags */
102 a_AMV_VF_BOOL = 1u<<0, /* ok_b_* */
103 a_AMV_VF_CHAIN = 1u<<1, /* Is variable chain (-USER{,@HOST} variants) */
104 a_AMV_VF_VIRT = 1u<<2, /* "Stateless" automatic variable */
105 a_AMV_VF_VIP = 1u<<3, /* Wants _var_check_vips() evaluation */
106 a_AMV_VF_RDONLY = 1u<<4, /* May not be set by user */
107 a_AMV_VF_NODEL = 1u<<5, /* May not be deleted */
108 a_AMV_VF_I3VAL = 1u<<6, /* Has an initial value */
109 a_AMV_VF_DEFVAL = 1u<<7, /* Has a default value */
110 a_AMV_VF_IMPORT = 1u<<8, /* Import ONLY from env (pre n_PSO_STARTED) */
111 a_AMV_VF_ENV = 1u<<9, /* Update environment on change */
112 a_AMV_VF_NOLOPTS = 1u<<10, /* May not be tracked by `localopts' */
113 a_AMV_VF_NOTEMPTY = 1u<<11, /* May not be assigned an empty value */
114 /* TODO _VF_NUM, _VF_POSNUM: we also need 64-bit limit numbers! */
115 a_AMV_VF_NUM = 1u<<12, /* Value must be a 32-bit number */
116 a_AMV_VF_POSNUM = 1u<<13, /* Value must be positive 32-bit number */
117 a_AMV_VF_LOWER = 1u<<14, /* Values will be stored in lowercase version */
118 a_AMV_VF_OBSOLETE = 1u<<15, /* Is obsolete? */
119 a_AMV_VF__MASK = (1u<<(15+1)) - 1,
121 /* Extended flags, not part of struct a_amv_var_map.avm_flags */
122 a_AMV_VF_EXT_LOCAL = 1u<<23, /* `local' */
123 a_AMV_VF_EXT_LINKED = 1u<<24, /* `environ' link'ed */
124 a_AMV_VF_EXT_FROZEN = 1u<<25, /* Has been set by -S,.. */
125 a_AMV_VF_EXT_FROZEN_UNSET = 1u<<26, /* ..and was used to unset a variable */
126 a_AMV_VF_EXT__FROZEN_MASK = a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET,
127 a_AMV_VF_EXT__MASK = (1u<<(26+1)) - 1
130 enum a_amv_var_lookup_flags{
131 a_AMV_VLOOK_NONE = 0,
132 a_AMV_VLOOK_LOCAL = 1u<<0, /* Query `local' layer first */
133 a_AMV_VLOOK_LOCAL_ONLY = 1u<<1, /* MUST be a `local' variable */
134 /* Do not allocate new var for _I3VAL, see _var_lookup() for more */
135 a_AMV_VLOOK_I3VAL_NONEW = 1u<<2,
136 a_AMV_VLOOK_I3VAL_NONEW_REPORT = 1u<<3
139 enum a_amv_var_setclr_flags{
140 a_AMV_VSETCLR_NONE = 0,
141 a_AMV_VSETCLR_LOCAL = 1u<<0, /* Use `local' variables only */
142 a_AMV_VSETCLR_ENV = 1u<<1 /* `environ' or otherwise environ */
145 /* We support some special parameter names for one(+)-letter variable names;
146 * note these have counterparts in the code that manages shell expansion!
147 * All these special variables are solely backed by n_var_vlook(), and besides
148 * there is only a_amv_var_revlookup() which knows about them */
149 enum a_amv_var_special_category{
150 a_AMV_VSC_NONE, /* Normal variable, no special treatment */
151 a_AMV_VSC_GLOBAL, /* ${[?!]} are specially mapped, but global */
152 a_AMV_VSC_MULTIPLEX, /* ${^.*} circumflex accent multiplexer */
153 a_AMV_VSC_POSPAR, /* ${[1-9][0-9]*} positional parameters */
154 a_AMV_VSC_POSPAR_ENV /* ${[*@#]} positional parameter support variables */
157 enum a_amv_var_special_type{
158 /* _VSC_GLOBAL */
159 a_AMV_VST_QM, /* ? */
160 a_AMV_VST_EM, /* ! */
161 /* _VSC_MULTIPLEX */
162 /* This is special in that it is a multiplex indicator, the ^ is followed by
163 * a normal variable */
164 a_AMV_VST_CACC, /* ^ (circumflex accent) */
165 /* _VSC_POSPAR_ENV */
166 a_AMV_VST_STAR, /* * */
167 a_AMV_VST_AT, /* @ */
168 a_AMV_VST_NOSIGN /* # */
171 enum a_amv_var_vip_mode{
172 a_AMV_VIP_SET_PRE,
173 a_AMV_VIP_SET_POST,
174 a_AMV_VIP_CLEAR
177 struct a_amv_pospar{
178 ui16_t app_maxcount; /* == slots in .app_dat */
179 ui16_t app_count; /* Maximum a_AMV_POSPAR_MAX */
180 ui16_t app_idx; /* `shift' moves this one, decs .app_count */
181 bool_t app_not_heap; /* .app_dat stuff not dynamically allocated */
182 ui8_t app__dummy[1];
183 char const **app_dat; /* NULL terminated (for "$@" explosion support) */
185 n_CTA(a_AMV_POSPAR_MAX <= SI16_MAX, "Limit exceeds datatype capabilities");
187 struct a_amv_mac{
188 struct a_amv_mac *am_next;
189 ui32_t am_maxlen; /* of any line in .am_line_dat */
190 ui32_t am_line_cnt; /* of *.am_line_dat (but NULL terminated) */
191 struct a_amv_mac_line **am_line_dat; /* TODO use deque? */
192 struct a_amv_var *am_lopts; /* `localopts' unroll list */
193 ui32_t am_refcnt; /* 0-based for `un{account,define}' purposes */
194 ui8_t am_flags; /* enum a_amv_mac_flags */
195 char am_name[n_VFIELD_SIZE(3)]; /* of this macro */
197 n_CTA(a_AMV_MF__MAX <= UI8_MAX, "Enumeration excesses storage datatype");
199 struct a_amv_mac_line{
200 ui32_t aml_len;
201 ui32_t aml_prespc; /* Number of leading SPC, for display purposes */
202 char aml_dat[n_VFIELD_SIZE(0)];
205 struct a_amv_mac_call_args{
206 char const *amca_name; /* For MACKY_MACK, this is *0*! */
207 struct a_amv_mac *amca_amp; /* "const", but for am_refcnt */
208 struct a_amv_var **amca_unroller;
209 void (*amca_hook_pre)(void *);
210 void *amca_hook_arg;
211 ui8_t amca_loflags;
212 bool_t amca_ps_hook_mask;
213 bool_t amca_no_xcall; /* We want n_GO_INPUT_NO_XCALL for this */
214 ui8_t amca__pad[5];
215 struct a_amv_var *(*amca_local_vars)[a_AMV_PRIME]; /* `local's, or NULL */
216 struct a_amv_pospar amca_pospar;
219 struct a_amv_lostack{
220 struct a_amv_lostack *as_global_saved; /* Saved global XXX due to jump */
221 struct a_amv_mac_call_args *as_amcap;
222 struct a_amv_lostack *as_up; /* Outer context */
223 struct a_amv_var *as_lopts;
224 ui8_t as_loflags; /* enum a_amv_mac_loflags */
225 ui8_t avs__pad[7];
228 struct a_amv_var{
229 struct a_amv_var *av_link;
230 char *av_value;
231 #ifdef HAVE_PUTENV
232 char *av_env; /* Actively managed putenv(3) memory, or NULL */
233 #endif
234 ui32_t av_flags; /* enum a_amv_var_flags inclusive extended bits */
235 char av_name[n_VFIELD_SIZE(4)];
237 n_CTA(a_AMV_VF_EXT__MASK <= UI32_MAX, "Enumeration excesses storage datatype");
239 struct a_amv_var_map{
240 ui32_t avm_hash;
241 ui16_t avm_keyoff;
242 ui16_t avm_flags; /* enum a_amv_var_flags without extended bits */
244 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
246 struct a_amv_var_virt{
247 ui32_t avv_okey;
248 ui8_t avv__dummy[4];
249 struct a_amv_var const *avv_var;
252 struct a_amv_var_defval{
253 ui32_t avdv_okey;
254 ui8_t avdv__pad[4];
255 char const *avdv_value; /* Only for !BOOL (otherwise plain existence) */
258 struct a_amv_var_carrier{
259 char const *avc_name;
260 ui32_t avc_hash;
261 ui32_t avc_prime;
262 struct a_amv_var *avc_var;
263 struct a_amv_var_map const *avc_map;
264 enum okeys avc_okey;
265 ui8_t avc__pad[1];
266 ui8_t avc_special_cat;
267 /* Numerical parameter name if .avc_special_cat=a_AMV_VSC_POSPAR,
268 * otherwise a enum a_amv_var_special_type */
269 ui16_t avc_special_prop;
272 /* Include constant make-okey-map.pl output, and the generated version data */
273 #include <gen-version.h>
274 #include <gen-okeys.h>
276 /* The currently active account */
277 static struct a_amv_mac *a_amv_acc_curr;
279 static struct a_amv_mac *a_amv_macs[a_AMV_PRIME]; /* TODO dynamically spaced */
281 /* Unroll list of currently running macro stack */
282 static struct a_amv_lostack *a_amv_lopts;
284 static struct a_amv_var *a_amv_vars[a_AMV_PRIME]; /* TODO dynamically spaced */
286 /* Global (i.e., non-local) a_AMV_VSC_POSPAR stack */
287 static struct a_amv_pospar a_amv_pospar;
289 /* TODO We really deserve localopts support for *folder-hook*s, so hack it in
290 * TODO today via a static lostack, it should be a field in mailbox, once that
291 * TODO is a real multi-instance object */
292 static struct a_amv_var *a_amv_folder_hook_lopts;
294 /* TODO Rather ditto (except for storage -> cmd_ctx), compose hooks */
295 static struct a_amv_var *a_amv_compose_lopts;
297 /* Lookup for macros/accounts: if newamp is not NULL it will be linked in the
298 * map, if _MF_UNDEF is set a possibly existing entry will be removed (first).
299 * Returns NULL if a lookup failed, or if newamp was set, the found entry in
300 * plain lookup cases or when _UNDEF was performed on a currently active entry
301 * (the entry will have been unlinked, and the _MF_DELETE will be honoured once
302 * the reference count reaches 0), and (*)-1 if an _UNDEF was performed */
303 static struct a_amv_mac *a_amv_mac_lookup(char const *name,
304 struct a_amv_mac *newamp, enum a_amv_mac_flags amf);
306 /* `call', `call_if' (and `xcall' via go.c -> c_call()) */
307 static int a_amv_mac_call(void *v, bool_t silent_nexist);
309 /* Execute a macro; amcap must reside in LOFI memory */
310 static bool_t a_amv_mac_exec(struct a_amv_mac_call_args *amcap);
312 static void a_amv_mac__finalize(void *vp);
314 /* User display helpers */
315 static bool_t a_amv_mac_show(enum a_amv_mac_flags amf);
317 /* _def() returns error for faulty definitions and already existing * names,
318 * _undef() returns error if a named thing doesn't exist */
319 static bool_t a_amv_mac_def(char const *name, enum a_amv_mac_flags amf);
320 static bool_t a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf);
322 /* */
323 static void a_amv_mac_free(struct a_amv_mac *amp);
325 /* Update replay-log */
326 static void a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
327 struct a_amv_var *oavp);
328 static void a_amv_lopts_unroll(struct a_amv_var **avpp);
330 /* Special cased value string allocation */
331 static char *a_amv_var_copy(char const *str);
332 static void a_amv_var_free(char *cp);
334 /* Check for special housekeeping. _VIP_SET_POST and _VIP_CLEAR do not fail
335 * (or propagate errors), _VIP_SET_PRE may and should case abortion */
336 static bool_t a_amv_var_check_vips(enum a_amv_var_vip_mode avvm,
337 enum okeys okey, char const *val);
339 /* _VF_NUM / _VF_POSNUM */
340 static bool_t a_amv_var_check_num(char const *val, bool_t posnum);
342 /* If a variable name begins with a lowercase-character and contains at
343 * least one '@', it is converted to all-lowercase. This is necessary
344 * for lookups of names based on email addresses.
345 * Following the standard, only the part following the last '@' should
346 * be lower-cased, but practice has established otherwise here */
347 static char const *a_amv_var_canonify(char const *vn);
349 /* Try to reverse lookup an option name to an enum okeys mapping.
350 * Updates .avc_name and .avc_hash; .avc_map is NULL if none found */
351 static bool_t a_amv_var_revlookup(struct a_amv_var_carrier *avcp,
352 char const *name);
354 /* Lookup a variable from .avc_(map|name|hash), return whether it was found.
355 * Sets .avc_prime; .avc_var is NULL if not found.
356 * Here it is where we care for _I3VAL and _DEFVAL.
357 * An _I3VAL will be "consumed" as necessary anyway, but it won't be used to
358 * create a new variable if _VLOOK_I3VAL_NONEW is set; if
359 * _VLOOK_I3VAL_NONEW_REPORT is set then we set .avc_var to -1 and return true
360 * if that was the case, otherwise we'll return FAL0, then! */
361 static bool_t a_amv_var_lookup(struct a_amv_var_carrier *avcp,
362 enum a_amv_var_lookup_flags avlf);
364 /* Lookup functions for special category variables, _pospar drives all
365 * positional parameter etc. special categories */
366 static char const *a_amv_var_vsc_global(struct a_amv_var_carrier *avcp);
367 static char const *a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp);
368 static char const *a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp);
370 /* Set var from .avc_(map|name|hash), return success */
371 static bool_t a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
372 enum a_amv_var_setclr_flags avscf);
374 static bool_t a_amv_var__putenv(struct a_amv_var_carrier *avcp,
375 struct a_amv_var *avp);
377 /* Clear var from .avc_(map|name|hash); sets .avc_var=NULL, return success */
378 static bool_t a_amv_var_clear(struct a_amv_var_carrier *avcp,
379 enum a_amv_var_setclr_flags avscf);
381 static bool_t a_amv_var__clearenv(char const *name, struct a_amv_var *avp);
383 /* List all variables */
384 static void a_amv_var_show_all(void);
386 static int a_amv_var__show_cmp(void const *s1, void const *s2);
388 /* Actually do print one, return number of lines written */
389 static size_t a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp);
391 /* Shared c_set() and c_environ():set impl, return success */
392 static bool_t a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf);
394 static struct a_amv_mac *
395 a_amv_mac_lookup(char const *name, struct a_amv_mac *newamp,
396 enum a_amv_mac_flags amf){
397 struct a_amv_mac *amp, **ampp;
398 ui32_t h;
399 enum a_amv_mac_flags save_amf;
400 NYD2_ENTER;
402 save_amf = amf;
403 amf &= a_AMV_MF_TYPE_MASK;
404 h = a_AMV_NAME2HASH(name);
405 h = a_AMV_HASH2PRIME(h);
406 ampp = &a_amv_macs[h];
408 for(amp = *ampp; amp != NULL; ampp = &(*ampp)->am_next, amp = amp->am_next){
409 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf &&
410 !strcmp(amp->am_name, name)){
411 if(n_LIKELY((save_amf & a_AMV_MF_UNDEF) == 0))
412 goto jleave;
414 *ampp = amp->am_next;
416 if(amp->am_refcnt > 0){
417 amp->am_flags |= a_AMV_MF_DELETE;
418 if(n_poption & n_PO_D_V)
419 n_err(_("Delayed deletion of currently active %s: %s\n"),
420 (amp->am_flags & a_AMV_MF_ACCOUNT ? "account" : "define"),
421 name);
422 }else{
423 a_amv_mac_free(amp);
424 amp = (struct a_amv_mac*)-1;
426 break;
430 if(newamp != NULL){
431 ampp = &a_amv_macs[h];
432 newamp->am_next = *ampp;
433 *ampp = newamp;
434 amp = NULL;
436 jleave:
437 NYD2_LEAVE;
438 return amp;
441 static int
442 a_amv_mac_call(void *v, bool_t silent_nexist){
443 struct a_amv_mac *amp;
444 int rv;
445 char const *name;
446 struct n_cmd_arg_ctx *cacp;
447 NYD_ENTER;
449 cacp = v;
451 if(cacp->cac_no == 0){
452 n_err(_("Synopsis: call(_if)?: name [:<arguments>:]\n"));
453 n_pstate_err_no = n_ERR_INVAL;
454 rv = 1;
455 goto jleave;
458 name = cacp->cac_arg->ca_arg.ca_str.s;
460 if(n_UNLIKELY(cacp->cac_no > a_AMV_POSPAR_MAX)){
461 n_err(_("Too many arguments to macro `call': %s\n"), name);
462 n_pstate_err_no = n_ERR_OVERFLOW;
463 rv = 1;
464 }else if(n_UNLIKELY((amp = a_amv_mac_lookup(name, NULL, a_AMV_MF_NONE)
465 ) == NULL)){
466 if(!silent_nexist)
467 n_err(_("Undefined macro called: %s\n"), n_shexp_quote_cp(name, FAL0));
468 n_pstate_err_no = n_ERR_NOENT;
469 rv = 1;
470 }else{
471 char const **argv;
472 struct a_amv_mac_call_args *amcap;
473 size_t argc;
475 argc = cacp->cac_no + 1;
476 amcap = n_lofi_alloc(sizeof *amcap + (argc * sizeof *argv));
477 argv = (void*)&amcap[1];
479 for(argc = 0; (cacp->cac_arg = cacp->cac_arg->ca_next) != NULL; ++argc)
480 argv[argc] = cacp->cac_arg->ca_arg.ca_str.s;
481 argv[argc] = NULL;
483 memset(amcap, 0, sizeof *amcap);
484 amcap->amca_name = name;
485 amcap->amca_amp = amp;
486 if(a_amv_lopts != NULL)
487 amcap->amca_loflags = (a_amv_lopts->as_loflags & a_AMV_LF_CALL_MASK
488 ) >> a_AMV_LF_CALL_TO_SCOPE_SHIFT;
489 if(argc > 0){
490 amcap->amca_pospar.app_count = (ui16_t)argc;
491 amcap->amca_pospar.app_not_heap = TRU1;
492 amcap->amca_pospar.app_dat = argv;
495 (void)a_amv_mac_exec(amcap);
496 rv = n_pstate_ex_no;
498 jleave:
499 NYD_LEAVE;
500 return rv;
503 static bool_t
504 a_amv_mac_exec(struct a_amv_mac_call_args *amcap){
505 struct a_amv_lostack *losp;
506 struct a_amv_mac_line **amlp;
507 char **args_base, **args;
508 struct a_amv_mac *amp;
509 bool_t rv;
510 NYD2_ENTER;
512 amp = amcap->amca_amp;
513 assert(amp != NULL && amp != a_AMV_MACKY_MACK);
514 ++amp->am_refcnt;
515 /* XXX Unfortunately we yet need to dup the macro lines! :( */
516 args_base = args = n_alloc(sizeof(*args) * (amp->am_line_cnt +1));
517 for(amlp = amp->am_line_dat; *amlp != NULL; ++amlp)
518 *(args++) = sbufdup((*amlp)->aml_dat, (*amlp)->aml_len);
519 *args = NULL;
521 losp = n_lofi_alloc(sizeof *losp);
522 losp->as_global_saved = a_amv_lopts;
523 if((losp->as_amcap = amcap)->amca_unroller == NULL){
524 losp->as_up = losp->as_global_saved;
525 losp->as_lopts = NULL;
526 }else{
527 losp->as_up = NULL;
528 losp->as_lopts = *amcap->amca_unroller;
530 losp->as_loflags = amcap->amca_loflags;
532 a_amv_lopts = losp;
533 if(amcap->amca_hook_pre != NULL)
534 n_PS_ROOT_BLOCK((*amcap->amca_hook_pre)(amcap->amca_hook_arg));
535 rv = n_go_macro((n_GO_INPUT_NONE |
536 (amcap->amca_no_xcall ? n_GO_INPUT_NO_XCALL : 0)),
537 amp->am_name, args_base, &a_amv_mac__finalize, losp);
538 NYD2_LEAVE;
539 return rv;
542 static void
543 a_amv_mac__finalize(void *vp){
544 struct a_amv_mac *amp;
545 struct a_amv_mac_call_args *amcap;
546 struct a_amv_lostack *losp;
547 NYD2_ENTER;
549 losp = vp;
550 a_amv_lopts = losp->as_global_saved;
552 amcap = losp->as_amcap;
554 /* Delete positional parameter stack */
555 if(!amcap->amca_pospar.app_not_heap && amcap->amca_pospar.app_maxcount > 0){
556 ui16_t i;
558 for(i = amcap->amca_pospar.app_maxcount; i-- != 0;)
559 n_free(n_UNCONST(amcap->amca_pospar.app_dat[i]));
560 n_free(amcap->amca_pospar.app_dat);
563 /* `local' variable hashmap. These have no environment map, never */
564 if(amcap->amca_local_vars != NULL){
565 struct a_amv_var **avpp_base, **avpp, *avp;
567 for(avpp_base = *amcap->amca_local_vars, avpp = &avpp_base[a_AMV_PRIME];
568 avpp-- != avpp_base;)
569 while((avp = *avpp)){
570 assert((avp->av_flags & (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL)) ==
571 (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL));
572 assert(!(avp->av_flags &
573 ((a_AMV_VF__MASK | a_AMV_VF_EXT__MASK) &
574 ~(a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL))));
575 *avpp = avp->av_link;
576 a_amv_var_free(avp->av_value);
577 n_free(avp);
579 n_free(avpp_base);
582 /* Unroll `localopts', if applicable */
583 if(amcap->amca_unroller == NULL){
584 if(losp->as_lopts != NULL)
585 a_amv_lopts_unroll(&losp->as_lopts);
586 }else
587 *amcap->amca_unroller = losp->as_lopts;
589 if(amcap->amca_ps_hook_mask)
590 n_pstate &= ~n_PS_HOOK_MASK;
592 if((amp = amcap->amca_amp) != a_AMV_MACKY_MACK && amp != NULL &&
593 --amp->am_refcnt == 0 && (amp->am_flags & a_AMV_MF_DELETE))
594 a_amv_mac_free(amp);
596 n_lofi_free(losp);
597 n_lofi_free(amcap);
598 NYD2_LEAVE;
601 static bool_t
602 a_amv_mac_show(enum a_amv_mac_flags amf){
603 size_t lc, mc, ti, i;
604 char const *typestr;
605 FILE *fp;
606 bool_t rv;
607 NYD2_ENTER;
609 rv = FAL0;
611 if((fp = Ftmp(NULL, "deflist", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
612 NULL){
613 n_perr(_("`define' or `account' list: cannot create temporary file"), 0);
614 goto jleave;
617 amf &= a_AMV_MF_TYPE_MASK;
618 typestr = (amf & a_AMV_MF_ACCOUNT) ? "account" : "define";
620 for(lc = mc = ti = 0; ti < a_AMV_PRIME; ++ti){
621 struct a_amv_mac *amp;
623 for(amp = a_amv_macs[ti]; amp != NULL; amp = amp->am_next){
624 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
625 struct a_amv_mac_line **amlpp;
627 if(++mc > 1){
628 putc('\n', fp);
629 ++lc;
631 ++lc;
632 fprintf(fp, "%s %s {\n", typestr, amp->am_name);
633 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++lc, ++amlpp){
634 for(i = (*amlpp)->aml_prespc; i > 0; --i)
635 putc(' ', fp);
636 fputs((*amlpp)->aml_dat, fp);
637 putc('\n', fp);
639 fputs("}\n", fp);
640 ++lc;
644 if(mc > 0)
645 page_or_print(fp, lc);
647 rv = (ferror(fp) == 0);
648 Fclose(fp);
649 jleave:
650 NYD2_LEAVE;
651 return rv;
654 static bool_t
655 a_amv_mac_def(char const *name, enum a_amv_mac_flags amf){
656 struct str line;
657 ui32_t line_cnt, maxlen;
658 struct linelist{
659 struct linelist *ll_next;
660 struct a_amv_mac_line *ll_amlp;
661 } *llp, *ll_head, *ll_tail;
662 union {size_t s; int i; ui32_t ui; size_t l;} n;
663 struct a_amv_mac *amp;
664 bool_t rv;
665 NYD2_ENTER;
667 memset(&line, 0, sizeof line);
668 rv = FAL0;
669 amp = NULL;
671 /* TODO We should have our input state machine which emits Line events,
672 * TODO and hook different consumers dependent on our content, as stated
673 * TODO in i think lex_input: like this local macros etc. would become
674 * TODO possible from the input side */
675 /* Read in the lines which form the macro content */
676 for(ll_tail = ll_head = NULL, line_cnt = maxlen = 0;;){
677 ui32_t leaspc;
678 char *cp;
680 n.i = n_go_input(n_GO_INPUT_CTX_DEFAULT | n_GO_INPUT_NL_ESC, n_empty,
681 &line.s, &line.l, NULL, NULL);
682 if(n.ui == 0)
683 continue;
684 if(n.i < 0){
685 n_err(_("Unterminated %s definition: %s\n"),
686 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
687 goto jerr;
690 /* Trim WS, remember amount of leading spaces for display purposes */
691 for(cp = line.s, leaspc = 0; n.ui > 0; ++cp, --n.ui)
692 if(*cp == '\t')
693 leaspc = (leaspc + 8u) & ~7u;
694 else if(*cp == ' ')
695 ++leaspc;
696 else
697 break;
698 for(; n.ui > 0 && spacechar(cp[n.ui - 1]); --n.ui)
700 if(n.ui == 0)
701 continue;
703 maxlen = n_MAX(maxlen, n.ui);
704 cp[n.ui++] = '\0';
706 /* Is is the closing brace? */
707 if(*cp == '}')
708 break;
710 if(n_LIKELY(++line_cnt < UI32_MAX)){
711 struct a_amv_mac_line *amlp;
713 llp = n_autorec_alloc(sizeof *llp);
714 if(ll_head == NULL)
715 ll_head = llp;
716 else
717 ll_tail->ll_next = llp;
718 ll_tail = llp;
719 llp->ll_next = NULL;
720 llp->ll_amlp = amlp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac_line,
721 aml_dat) + n.ui);
722 amlp->aml_len = n.ui -1;
723 amlp->aml_prespc = leaspc;
724 memcpy(amlp->aml_dat, cp, n.ui);
725 }else{
726 n_err(_("Too much content in %s definition: %s\n"),
727 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
728 goto jerr;
732 /* Create the new macro */
733 n.s = strlen(name) +1;
734 amp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name) + n.s);
735 memset(amp, 0, n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name));
736 amp->am_maxlen = maxlen;
737 amp->am_line_cnt = line_cnt;
738 amp->am_flags = amf;
739 memcpy(amp->am_name, name, n.s);
740 /* C99 */{
741 struct a_amv_mac_line **amlpp;
743 amp->am_line_dat = amlpp = n_alloc(sizeof(*amlpp) * ++line_cnt);
744 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
745 *amlpp++ = llp->ll_amlp;
746 *amlpp = NULL;
749 /* Create entry, replace a yet existing one as necessary */
750 a_amv_mac_lookup(name, amp, amf | a_AMV_MF_UNDEF);
751 rv = TRU1;
752 jleave:
753 if(line.s != NULL)
754 n_free(line.s);
755 NYD2_LEAVE;
756 return rv;
758 jerr:
759 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
760 n_free(llp->ll_amlp);
762 * if(amp != NULL){
763 * n_free(amp->am_line_dat);
764 * n_free(amp);
765 *}*/
766 goto jleave;
769 static bool_t
770 a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf){
771 struct a_amv_mac *amp;
772 bool_t rv;
773 NYD2_ENTER;
775 rv = TRU1;
777 if(n_LIKELY(name[0] != '*' || name[1] != '\0')){
778 if((amp = a_amv_mac_lookup(name, NULL, amf | a_AMV_MF_UNDEF)) == NULL){
779 n_err(_("%s not defined: %s\n"),
780 (amf & a_AMV_MF_ACCOUNT ? "Account" : "Macro"), name);
781 rv = FAL0;
783 }else{
784 struct a_amv_mac **ampp, *lamp;
786 for(ampp = a_amv_macs; PTRCMP(ampp, <, &a_amv_macs[n_NELEM(a_amv_macs)]);
787 ++ampp)
788 for(lamp = NULL, amp = *ampp; amp != NULL;){
789 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
790 /* xxx Expensive but rare: be simple */
791 a_amv_mac_lookup(amp->am_name, NULL, amf | a_AMV_MF_UNDEF);
792 amp = (lamp == NULL) ? *ampp : lamp->am_next;
793 }else{
794 lamp = amp;
795 amp = amp->am_next;
799 NYD2_LEAVE;
800 return rv;
803 static void
804 a_amv_mac_free(struct a_amv_mac *amp){
805 struct a_amv_mac_line **amlpp;
806 NYD2_ENTER;
808 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++amlpp)
809 n_free(*amlpp);
810 n_free(amp->am_line_dat);
811 n_free(amp);
812 NYD2_LEAVE;
815 static void
816 a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
817 struct a_amv_var *oavp){
818 struct a_amv_var *avp;
819 size_t nl, vl;
820 NYD2_ENTER;
822 /* Propagate unrolling up the stack, as necessary */
823 assert(alp != NULL);
824 for(;;){
825 if(alp->as_loflags & a_AMV_LF_SCOPE_MASK)
826 break;
827 if((alp = alp->as_up) == NULL)
828 goto jleave;
831 /* Check whether this variable is handled yet XXX Boost: ID check etc.!! */
832 for(avp = alp->as_lopts; avp != NULL; avp = avp->av_link)
833 if(!strcmp(avp->av_name, name))
834 goto jleave;
836 nl = strlen(name) +1;
837 vl = (oavp != NULL) ? strlen(oavp->av_value) +1 : 0;
838 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + nl + vl);
839 avp->av_link = alp->as_lopts;
840 alp->as_lopts = avp;
841 if(vl != 0){
842 avp->av_value = &avp->av_name[nl];
843 avp->av_flags = oavp->av_flags;
844 memcpy(avp->av_value, oavp->av_value, vl);
845 #ifdef HAVE_PUTENV
846 if(oavp->av_env != NULL)
847 avp->av_env = sstrdup(oavp->av_env);
848 assert(avp->av_env == NULL ||
849 (avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)));
850 #endif
852 memcpy(avp->av_name, name, nl);
853 jleave:
854 NYD2_LEAVE;
857 static void
858 a_amv_lopts_unroll(struct a_amv_var **avpp){
859 struct a_amv_lostack *save_alp;
860 struct a_amv_var *x, *avp;
861 NYD2_ENTER;
863 avp = *avpp;
864 *avpp = NULL;
866 save_alp = a_amv_lopts;
867 a_amv_lopts = NULL;
868 while(avp != NULL){
869 x = avp;
870 avp = avp->av_link;
871 n_PS_ROOT_BLOCK(n_var_vset(x->av_name, (uintptr_t)x->av_value));
872 n_free(x);
874 a_amv_lopts = save_alp;
875 NYD2_LEAVE;
878 static char *
879 a_amv_var_copy(char const *str){
880 char *news;
881 size_t len;
882 NYD2_ENTER;
884 if(*str == '\0')
885 news = n_UNCONST(n_empty);
886 else if(str[1] == '\0'){
887 if(str[0] == '1')
888 news = n_UNCONST(n_1);
889 else if(str[0] == '0')
890 news = n_UNCONST(n_0);
891 else
892 goto jheap;
893 }else if(str[2] == '\0' && str[0] == '-' && str[1] == '1')
894 news = n_UNCONST(n_m1);
895 else{
896 jheap:
897 len = strlen(str) +1;
898 news = n_alloc(len);
899 memcpy(news, str, len);
901 NYD2_LEAVE;
902 return news;
905 static void
906 a_amv_var_free(char *cp){
907 NYD2_ENTER;
908 if(cp[0] != '\0' && cp != n_0 && cp != n_1 && cp != n_m1)
909 n_free(cp);
910 NYD2_LEAVE;
913 static bool_t
914 a_amv_var_check_vips(enum a_amv_var_vip_mode avvm, enum okeys okey,
915 char const *val){
916 bool_t ok;
917 NYD2_ENTER;
919 ok = TRU1;
921 if(avvm == a_AMV_VIP_SET_PRE){
922 switch(okey){
923 default:
924 break;
925 case ok_v_customhdr:{
926 char const *vp;
927 char *buf;
928 struct n_header_field *hflp, **hflpp, *hfp;
929 NYD_ENTER;
931 buf = savestr(val);
932 hflp = NULL;
933 hflpp = &hflp;
935 while((vp = n_strsep_esc(&buf, ',', TRU1)) != NULL){
936 if(!n_header_add_custom(hflpp, vp, TRU1)){
937 n_err(_("Invalid *customhdr* entry: %s\n"),
938 n_shexp_quote_cp(val, FAL0));
939 ok = FAL0;
940 break;
942 hflpp = &(*hflpp)->hf_next;
945 hflpp = ok ? &n_customhdr_list : &hflp;
946 while((hfp = *hflpp) != NULL){
947 *hflpp = hfp->hf_next;
948 n_free(hfp);
950 if(ok)
951 n_customhdr_list = hflp;
952 } break;
953 case ok_v_HOME:
954 /* Note this gets called from main.c during initialization, and they
955 * simply set this to pw_dir as a fallback: don't verify _that_ call.
956 * See main.c! */
957 if(!(n_pstate & n_PS_ROOT) && !n_is_dir(val, TRUM1)){
958 n_err(_("$HOME is not a directory or not accessible: %s\n"),
959 n_shexp_quote_cp(val, FAL0));
960 ok = FAL0;
962 break;
963 case ok_v_quote_chars:{
964 char c;
966 while((c = *val++) != '\0')
967 if(!asciichar(c) || blankspacechar(c)){
968 ok = FAL0;
969 break;
971 } break;
972 case ok_v_TMPDIR:
973 if(!n_is_dir(val, TRU1)){
974 n_err(_("$TMPDIR is not a directory or not accessible: %s\n"),
975 n_shexp_quote_cp(val, FAL0));
976 ok = FAL0;
978 break;
979 case ok_v_umask:
980 if(*val != '\0'){
981 ui64_t uib;
983 n_idec_ui64_cp(&uib, val, 0, NULL);
984 if(uib & ~0777u){ /* (is valid _VF_POSNUM) */
985 n_err(_("Invalid *umask* setting: %s\n"), val);
986 ok = FAL0;
989 break;
991 }else if(avvm == a_AMV_VIP_SET_POST){
992 switch(okey){
993 default:
994 break;
995 case ok_b_ask:
996 ok_bset(asksub);
997 break;
998 case ok_b_debug:
999 n_poption |= n_PO_DEBUG;
1000 break;
1001 case ok_v_HOME:
1002 /* Invalidate any resolved folder then, too
1003 * FALLTHRU */
1004 case ok_v_folder:
1005 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1006 break;
1007 case ok_v_ifs:{
1008 char *x_b, *x, c;
1010 x_b = x = n_autorec_alloc(strlen(val) +1);
1011 while((c = *val++) != '\0')
1012 if(spacechar(c))
1013 *x++ = c;
1014 *x = '\0';
1015 n_PS_ROOT_BLOCK(ok_vset(ifs_ws, x_b));
1016 } break;
1017 #ifdef HAVE_SETLOCALE
1018 case ok_v_LANG:
1019 case ok_v_LC_ALL:
1020 case ok_v_LC_CTYPE:
1021 n_locale_init();
1022 break;
1023 #endif
1024 case ok_b_memdebug:
1025 n_poption |= n_PO_MEMDEBUG;
1026 break;
1027 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1028 if(!(n_pstate & n_PS_ROOT))
1029 n_PS_ROOT_BLOCK(ok_bset(posix));
1030 break;
1031 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1032 if(!(n_pstate & n_PS_ROOT))
1033 n_PS_ROOT_BLOCK(ok_bset(POSIXLY_CORRECT));
1034 break;
1035 case ok_b_skipemptybody:
1036 n_poption |= n_PO_E_FLAG;
1037 break;
1038 case ok_b_typescript_mode:
1039 ok_bset(colour_disable);
1040 ok_bset(line_editor_disable);
1041 if(!(n_psonce & n_PSO_STARTED))
1042 ok_bset(termcap_disable);
1043 break;
1044 case ok_v_umask:
1045 if(*val != '\0'){
1046 ui64_t uib;
1048 n_idec_ui64_cp(&uib, val, 0, NULL);
1049 umask((mode_t)uib);
1051 break;
1052 case ok_b_verbose:
1053 n_poption |= (n_poption & n_PO_VERB) ? n_PO_VERBVERB : n_PO_VERB;
1054 break;
1056 }else{
1057 switch(okey){
1058 default:
1059 break;
1060 case ok_b_ask:
1061 ok_bclear(asksub);
1062 break;
1063 case ok_b_debug:
1064 n_poption &= ~n_PO_DEBUG;
1065 break;
1066 case ok_v_customhdr:{
1067 struct n_header_field *hfp;
1069 while((hfp = n_customhdr_list) != NULL){
1070 n_customhdr_list = hfp->hf_next;
1071 n_free(hfp);
1073 } break;
1074 case ok_v_HOME:
1075 /* Invalidate any resolved folder then, too
1076 * FALLTHRU */
1077 case ok_v_folder:
1078 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1079 break;
1080 case ok_b_memdebug:
1081 n_poption &= ~n_PO_MEMDEBUG;
1082 break;
1083 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1084 if(!(n_pstate & n_PS_ROOT))
1085 n_PS_ROOT_BLOCK(ok_bclear(posix));
1086 break;
1087 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1088 if(!(n_pstate & n_PS_ROOT))
1089 n_PS_ROOT_BLOCK(ok_bclear(POSIXLY_CORRECT));
1090 break;
1091 case ok_b_skipemptybody:
1092 n_poption &= ~n_PO_E_FLAG;
1093 break;
1094 case ok_b_verbose:
1095 n_poption &= ~(n_PO_VERB | n_PO_VERBVERB);
1096 break;
1099 NYD2_LEAVE;
1100 return ok;
1103 static bool_t
1104 a_amv_var_check_num(char const *val, bool_t posnum){
1105 /* TODO The internal/environment variables which are num= or posnum= should
1106 * TODO gain special lookup functions, or the return should be void* and
1107 * TODO castable to integer; i.e. no more strtoX() should be needed.
1108 * TODO I.e., the result of this function should instead be stored */
1109 bool_t rv;
1110 NYD2_ENTER;
1112 rv = TRU1;
1114 if(*val != '\0'){ /* Would be _VF_NOTEMPTY if not allowed */
1115 ui64_t uib;
1116 enum n_idec_state ids;
1118 ids = n_idec_cp(&uib, val, 0,
1119 (n_IDEC_MODE_LIMIT_32BIT |
1120 (posnum ? n_IDEC_MODE_SIGNED_TYPE : n_IDEC_MODE_NONE)), NULL);
1121 if((ids & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
1122 ) != n_IDEC_STATE_CONSUMED)
1123 rv = FAL0;
1124 /* TODO Unless we store integers we need to look and forbid, because
1125 * TODO callee may not be able to swallow, e.g., "-1" */
1126 if(posnum && (ids & n_IDEC_STATE_SEEN_MINUS))
1127 rv = FAL0;
1129 NYD2_LEAVE;
1130 return rv;
1133 static char const *
1134 a_amv_var_canonify(char const *vn){
1135 NYD2_ENTER;
1136 if(!upperchar(*vn)){
1137 char const *vp;
1139 for(vp = vn; *vp != '\0' && *vp != '@'; ++vp)
1141 vn = (*vp == '@') ? i_strdup(vn) : vn;
1143 NYD2_LEAVE;
1144 return vn;
1147 static bool_t
1148 a_amv_var_revlookup(struct a_amv_var_carrier *avcp, char const *name){
1149 ui32_t hash, i, j;
1150 struct a_amv_var_map const *avmp;
1151 char c;
1152 NYD2_ENTER;
1154 /* It may be a special a.k.a. macro-local or one-letter parameter */
1155 c = name[0];
1156 if(n_UNLIKELY(digitchar(c))){
1157 /* (Inline dec. atoi, ugh) */
1158 for(j = (ui8_t)c - '0', i = 1;; ++i){
1159 c = name[i];
1160 if(c == '\0')
1161 break;
1162 if(!digitchar(c))
1163 goto jno_special_param;
1164 j = j * 10 + (ui8_t)c - '0';
1166 if(j <= a_AMV_POSPAR_MAX){
1167 avcp->avc_special_cat = a_AMV_VSC_POSPAR;
1168 goto jspecial_param;
1170 }else if(n_UNLIKELY(name[1] == '\0')){
1171 switch(c){
1172 case '?':
1173 case '!':
1174 avcp->avc_special_cat = a_AMV_VSC_GLOBAL;
1175 j = (c == '?') ? a_AMV_VST_QM : a_AMV_VST_EM;
1176 goto jspecial_param;
1177 case '^':
1178 goto jmultiplex;
1179 case '*':
1180 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1181 j = a_AMV_VST_STAR;
1182 goto jspecial_param;
1183 case '@':
1184 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1185 j = a_AMV_VST_AT;
1186 goto jspecial_param;
1187 case '#':
1188 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1189 j = a_AMV_VST_NOSIGN;
1190 goto jspecial_param;
1191 default:
1192 break;
1194 }else if(c == '^'){
1195 jmultiplex:
1196 avcp->avc_special_cat = a_AMV_VSC_MULTIPLEX;
1197 j = a_AMV_VST_CACC;
1198 goto jspecial_param;
1201 /* Normal reverse lookup, walk over the hashtable */
1202 jno_special_param:
1203 avcp->avc_special_cat = a_AMV_VSC_NONE;
1204 avcp->avc_name = name = a_amv_var_canonify(name);
1205 avcp->avc_hash = hash = a_AMV_NAME2HASH(name);
1207 for(i = hash % a_AMV_VAR_REV_PRIME, j = 0; j <= a_AMV_VAR_REV_LONGEST; ++j){
1208 ui32_t x;
1210 if((x = a_amv_var_revmap[i]) == a_AMV_VAR_REV_ILL)
1211 break;
1213 avmp = &a_amv_var_map[x];
1214 if(avmp->avm_hash == hash &&
1215 !strcmp(&a_amv_var_names[avmp->avm_keyoff], name)){
1216 avcp->avc_map = avmp;
1217 avcp->avc_okey = (enum okeys)x;
1218 goto jleave;
1221 if(++i == a_AMV_VAR_REV_PRIME){
1222 #ifdef a_AMV_VAR_REV_WRAPAROUND
1223 i = 0;
1224 #else
1225 break;
1226 #endif
1230 avcp->avc_map = NULL;
1231 avcp = NULL;
1232 jleave:
1233 assert(avcp == NULL || avcp->avc_map != NULL ||
1234 avcp->avc_special_cat == a_AMV_VSC_NONE);
1235 NYD2_LEAVE;
1236 return (avcp != NULL);
1238 /* All these are mapped to *--special-param* */
1239 jspecial_param:
1240 avcp->avc_name = name;
1241 avcp->avc_special_prop = (ui16_t)j;
1242 avmp = &a_amv_var_map[a_AMV_VAR__SPECIAL_PARAM_MAP_IDX];
1243 avcp->avc_hash = avmp->avm_hash;
1244 avcp->avc_map = avmp;
1245 avcp->avc_okey = ok_v___special_param;
1246 goto jleave;
1249 static bool_t
1250 a_amv_var_lookup(struct a_amv_var_carrier *avcp,
1251 enum a_amv_var_lookup_flags avlf){
1252 size_t i;
1253 char const *cp;
1254 struct a_amv_var_map const *avmp;
1255 struct a_amv_var *avp;
1256 NYD2_ENTER;
1258 assert(!(avlf & a_AMV_VLOOK_LOCAL_ONLY) || (avlf & a_AMV_VLOOK_LOCAL));
1259 assert(!(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT) ||
1260 (avlf & a_AMV_VLOOK_I3VAL_NONEW));
1262 /* C99 */{
1263 struct a_amv_var **avpp, *lavp;
1265 avcp->avc_prime = a_AMV_HASH2PRIME(avcp->avc_hash);
1267 /* Optionally macro-`local' variables first */
1268 if(avlf & a_AMV_VLOOK_LOCAL){
1269 if(a_amv_lopts != NULL &&
1270 (avpp = *a_amv_lopts->as_amcap->amca_local_vars) != NULL){
1271 avpp += avcp->avc_prime;
1273 for(lavp = NULL, avp = *avpp; avp != NULL;
1274 lavp = avp, avp = avp->av_link)
1275 if(!strcmp(avp->av_name, avcp->avc_name)){
1276 /* Relink as head, hope it "sorts on usage" over time.
1277 * The code relies on this behaviour! */
1278 if(lavp != NULL){
1279 lavp->av_link = avp->av_link;
1280 avp->av_link = *avpp;
1281 *avpp = avp;
1283 goto jleave;
1287 if(avlf & a_AMV_VLOOK_LOCAL_ONLY)
1288 goto jerr;
1291 /* Global variable map */
1292 avpp = &a_amv_vars[avcp->avc_prime];
1294 for(lavp = NULL, avp = *avpp; avp != NULL; lavp = avp, avp = avp->av_link)
1295 if(!strcmp(avp->av_name, avcp->avc_name)){
1296 /* Relink as head, hope it "sorts on usage" over time.
1297 * The code relies on this behaviour! */
1298 if(lavp != NULL){
1299 lavp->av_link = avp->av_link;
1300 avp->av_link = *avpp;
1301 *avpp = avp;
1304 /* If this setting has been established via -S and we still have
1305 * not reached the _STARTED_CONFIG program state, it may have been
1306 * an explicit "clearance" that is to be treated as unset.
1307 * Because that is a special condition that (has been hacked in
1308 * later and) needs to be encapsulated in lower levels, but not
1309 * of interest if _set() or _clear() called us */
1310 switch(avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK){
1311 case a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET:
1312 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW)){
1313 avcp->avc_var = avp;
1314 avp = NULL;
1315 goto j_leave;
1317 /* FALLTHRU */
1318 default:
1319 break;
1321 goto jleave;
1325 /* If this is not an assembled variable we need to consider some special
1326 * initialization cases and eventually create the variable anew */
1327 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1328 /* Does it have an import-from-environment flag? */
1329 if(n_UNLIKELY((avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV)) != 0)){
1330 if(n_LIKELY((cp = getenv(avcp->avc_name)) != NULL)){
1331 /* May be better not to use that one, though? */
1332 /* TODO Outsource the tests into a _shared_ test function! */
1333 bool_t isempty, isbltin;
1335 isempty = (*cp == '\0' &&
1336 (avmp->avm_flags & a_AMV_VF_NOTEMPTY) != 0);
1337 isbltin = ((avmp->avm_flags & (a_AMV_VF_I3VAL | a_AMV_VF_DEFVAL)
1338 ) != 0);
1340 if(n_UNLIKELY(isempty)){
1341 n_err(_("Environment variable must not be empty: %s\n"),
1342 avcp->avc_name);
1343 if(!isbltin)
1344 goto jerr;
1345 }else if(n_LIKELY(*cp != '\0')){
1346 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1347 !a_amv_var_check_num(cp, FAL0))){
1348 n_err(_("Environment variable value not a number "
1349 "or out of range: %s\n"), avcp->avc_name);
1350 goto jerr;
1352 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1353 !a_amv_var_check_num(cp, TRU1))){
1354 n_err(_("Environment variable value not a number, "
1355 "negative or out of range: %s\n"), avcp->avc_name);
1356 goto jerr;
1358 goto jnewval;
1359 }else
1360 goto jnewval;
1364 /* A first-time init switch is to be handled now and here */
1365 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_I3VAL) != 0)){
1366 static struct a_amv_var_defval const **arr,
1367 *arr_base[a_AMV_VAR_I3VALS_CNT +1];
1369 if(arr == NULL){
1370 arr = &arr_base[0];
1371 arr[i = a_AMV_VAR_I3VALS_CNT] = NULL;
1372 while(i-- > 0)
1373 arr[i] = &a_amv_var_i3vals[i];
1376 for(i = 0; arr[i] != NULL; ++i)
1377 if(arr[i]->avdv_okey == avcp->avc_okey){
1378 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_1
1379 : arr[i]->avdv_value;
1380 /* Remove this entry, hope entire block becomes no-op asap */
1382 arr[i] = arr[i + 1];
1383 while(arr[i++] != NULL);
1385 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW))
1386 goto jnewval;
1387 if(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT)
1388 avp = (struct a_amv_var*)-1;
1389 goto jleave;
1393 /* */
1394 jdefval:
1395 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0){
1396 for(i = 0; i < a_AMV_VAR_DEFVALS_CNT; ++i)
1397 if(a_amv_var_defvals[i].avdv_okey == avcp->avc_okey){
1398 cp = (avmp->avm_flags & a_AMV_VF_BOOL) ? n_1
1399 : a_amv_var_defvals[i].avdv_value;
1400 goto jnewval;
1404 /* The virtual variables */
1405 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIRT) != 0)){
1406 for(i = 0; i < a_AMV_VAR_VIRTS_CNT; ++i)
1407 if(a_amv_var_virts[i].avv_okey == avcp->avc_okey){
1408 avp = n_UNCONST(a_amv_var_virts[i].avv_var);
1409 goto jleave;
1411 /* Not reached */
1415 jerr:
1416 avp = NULL;
1417 jleave:
1418 avcp->avc_var = avp;
1419 j_leave:
1420 NYD2_LEAVE;
1421 return (avp != NULL);
1423 jnewval:
1424 /* E.g., $TMPDIR may be set to non-existent, so we need to be able to catch
1425 * that and redirect to a possible default value */
1426 if((avmp->avm_flags & a_AMV_VF_VIP) &&
1427 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, cp)){
1428 #ifdef HAVE_SETENV
1429 if(avmp->avm_flags & (a_AMV_VF_IMPORT | a_AMV_VF_ENV))
1430 unsetenv(avcp->avc_name);
1431 #endif
1432 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)
1433 goto jdefval;
1434 goto jerr;
1435 }else{
1436 struct a_amv_var **avpp;
1437 size_t l;
1439 l = strlen(avcp->avc_name) +1;
1440 avcp->avc_var =
1441 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1442 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1443 *avpp = avp;
1444 avp->av_flags = avmp->avm_flags;
1445 avp->av_value = a_amv_var_copy(cp);
1446 memcpy(avp->av_name, avcp->avc_name, l);
1448 if(avp->av_flags & a_AMV_VF_ENV)
1449 a_amv_var__putenv(avcp, avp);
1450 if(avmp->avm_flags & a_AMV_VF_VIP)
1451 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, cp);
1452 goto jleave;
1456 static char const *
1457 a_amv_var_vsc_global(struct a_amv_var_carrier *avcp){
1458 char itoabuf[32];
1459 char const *rv;
1460 si32_t *ep;
1461 struct a_amv_var_map const *avmp;
1462 NYD2_ENTER;
1464 /* Not function local, TODO but lazy evaluted for now */
1465 if(avcp->avc_special_prop == a_AMV_VST_QM){
1466 avmp = &a_amv_var_map[a_AMV_VAR__QM_MAP_IDX];
1467 avcp->avc_okey = ok_v___qm;
1468 ep = &n_pstate_ex_no;
1469 }else{
1470 avmp = &a_amv_var_map[a_AMV_VAR__EM_MAP_IDX];
1471 avcp->avc_okey = ok_v___em;
1472 ep = &n_pstate_err_no;
1475 /* XXX Oh heaven, we are responsible to ensure that $?/! is up-to-date
1476 * TODO we could num=1 ok_v___[qe]m, but the thing is still a string
1477 * TODO and thus conversion takes places over and over again; also
1478 * TODO for now that would have to occur before we set _that_ value
1479 * TODO so let's special treat it until we store ints as such */
1480 switch(*ep){
1481 case 0: rv = n_0; break;
1482 case 1: rv = n_1; break;
1483 default:
1484 snprintf(itoabuf, sizeof itoabuf, "%d", *ep);
1485 rv = itoabuf;
1486 break;
1488 n_PS_ROOT_BLOCK(n_var_okset(avcp->avc_okey, (uintptr_t)rv));
1490 avcp->avc_hash = avmp->avm_hash;
1491 avcp->avc_map = avmp;
1492 rv = a_amv_var_lookup(avcp, a_AMV_VLOOK_NONE)
1493 ? avcp->avc_var->av_value : NULL;
1494 NYD2_LEAVE;
1495 return rv;
1498 static char const *
1499 a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp){
1500 char itoabuf[32];
1501 si32_t e;
1502 size_t i;
1503 char const *rv;
1504 NYD2_ENTER;
1506 i = strlen(rv = &avcp->avc_name[1]);
1508 /* ERR, ERRDOC, ERRNAME, plus *-NAME variants */
1509 if(rv[0] == 'E' && i >= 3 && rv[1] == 'R' && rv[2] == 'R'){
1510 if(i == 3){
1511 e = n_pstate_err_no;
1512 goto jeno;
1513 }else if(rv[3] == '-'){
1514 e = n_err_from_name(&rv[4]);
1515 jeno:
1516 switch(e){
1517 case 0: rv = n_0; break;
1518 case 1: rv = n_1; break;
1519 default:
1520 snprintf(itoabuf, sizeof itoabuf, "%d", e);
1521 rv = savestr(itoabuf); /* XXX yet, cannot do numbers */
1522 break;
1524 goto jleave;
1525 }else if(i >= 6){
1526 if(!memcmp(&rv[3], "DOC", 3)){
1527 rv += 6;
1528 switch(*rv){
1529 case '\0': e = n_pstate_err_no; break;
1530 case '-': e = n_err_from_name(&rv[1]); break;
1531 default: goto jerr;
1533 rv = n_err_to_doc(e);
1534 goto jleave;
1535 }else if(i >= 7 && !memcmp(&rv[3], "NAME", 4)){
1536 rv += 7;
1537 switch(*rv){
1538 case '\0': e = n_pstate_err_no; break;
1539 case '-': e = n_err_from_name(&rv[1]); break;
1540 default: goto jerr;
1542 rv = n_err_to_name(e);
1543 goto jleave;
1548 jerr:
1549 rv = NULL;
1550 jleave:
1551 NYD2_LEAVE;
1552 return rv;
1555 static char const *
1556 a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp){
1557 size_t i, j;
1558 ui16_t argc;
1559 char const *rv, **argv;
1560 NYD2_ENTER;
1562 rv = NULL;
1564 /* If in a macro/xy.. */
1565 if(a_amv_lopts != NULL){
1566 bool_t ismacky;
1567 struct a_amv_mac_call_args *amcap;
1569 amcap = a_amv_lopts->as_amcap;
1570 argc = amcap->amca_pospar.app_count;
1571 argv = amcap->amca_pospar.app_dat;
1572 argv += amcap->amca_pospar.app_idx;
1574 /* ..in a `call'ed macro only, to be exact. Or in a_AMV_MACKY_MACK */
1575 if(!(ismacky = (amcap->amca_amp == a_AMV_MACKY_MACK)) &&
1576 (amcap->amca_ps_hook_mask ||
1577 (assert(amcap->amca_amp != NULL),
1578 (amcap->amca_amp->am_flags & a_AMV_MF_TYPE_MASK
1579 ) == a_AMV_MF_ACCOUNT)))
1580 goto jleave;
1582 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1583 if(avcp->avc_special_prop > 0){
1584 if(argc >= avcp->avc_special_prop)
1585 rv = argv[avcp->avc_special_prop - 1];
1586 }else if(ismacky)
1587 rv = amcap->amca_name;
1588 else
1589 rv = (a_amv_lopts->as_up != NULL
1590 ? a_amv_lopts->as_up->as_amcap->amca_name : n_empty);
1591 goto jleave;
1593 /* MACKY_MACK doesn't know about [*@#] */
1594 /*else*/ if(ismacky){
1595 if(n_poption & n_PO_D_V)
1596 n_err(_("Cannot use $*/$@/$# in this context: %s\n"),
1597 n_shexp_quote_cp(avcp->avc_name, FAL0));
1598 goto jleave;
1600 }else{
1601 argc = a_amv_pospar.app_count;
1602 argv = a_amv_pospar.app_dat;
1603 argv += a_amv_pospar.app_idx;
1605 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1606 if(avcp->avc_special_prop > 0){
1607 if(argc >= avcp->avc_special_prop)
1608 rv = argv[avcp->avc_special_prop - 1];
1609 }else
1610 rv = n_progname;
1611 goto jleave;
1615 switch(avcp->avc_special_prop){ /* XXX OPTIMIZE */
1616 case a_AMV_VST_STAR:{
1617 char sep;
1619 sep = *ok_vlook(ifs);
1620 if(0){
1621 case a_AMV_VST_AT:
1622 sep = ' ';
1624 for(i = j = 0; i < argc; ++i)
1625 j += strlen(argv[i]) + 1;
1626 if(j == 0)
1627 rv = n_empty;
1628 else{
1629 char *cp;
1631 rv = cp = n_autorec_alloc(j);
1632 for(i = j = 0; i < argc; ++i){
1633 j = strlen(argv[i]);
1634 memcpy(cp, argv[i], j);
1635 cp += j;
1636 if(sep != '\0')
1637 *cp++ = sep;
1639 if(sep != '\0')
1640 --cp;
1641 *cp = '\0';
1643 }break;
1644 case a_AMV_VST_NOSIGN:{
1645 char *cp;
1647 rv = cp = n_autorec_alloc(sizeof("65535"));
1648 snprintf(cp, sizeof("65535"), "%hu", argc);
1649 }break;
1650 default:
1651 rv = n_empty;
1652 break;
1654 jleave:
1655 NYD2_LEAVE;
1656 return rv;
1659 static bool_t
1660 a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
1661 enum a_amv_var_setclr_flags avscf){
1662 struct a_amv_var *avp;
1663 char *oval;
1664 struct a_amv_var_map const *avmp;
1665 bool_t rv;
1666 NYD2_ENTER;
1668 if(value == NULL){
1669 rv = a_amv_var_clear(avcp, avscf);
1670 goto jleave;
1673 if((avmp = avcp->avc_map) != NULL){
1674 rv = FAL0;
1676 /* Validity checks */
1677 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_RDONLY) != 0 &&
1678 !(n_pstate & n_PS_ROOT))){
1679 value = N_("Variable is read-only: %s\n");
1680 goto jeavmp;
1682 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOTEMPTY) && *value == '\0')){
1683 value = N_("Variable must not be empty: %s\n");
1684 goto jeavmp;
1686 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1687 !a_amv_var_check_num(value, FAL0))){
1688 value = N_("Variable value not a number or out of range: %s\n");
1689 goto jeavmp;
1691 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1692 !a_amv_var_check_num(value, TRU1))){
1693 value = _("Variable value not a number, negative, "
1694 "or out of range: %s\n");
1695 goto jeavmp;
1697 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_IMPORT) != 0 &&
1698 !(n_psonce & n_PSO_STARTED) && !(n_pstate & n_PS_ROOT))){
1699 value = N_("Variable cannot be set in a resource file: %s\n");
1700 goto jeavmp;
1703 /* Any more complicated inter-dependency? */
1704 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1705 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, value))){
1706 value = N_("Assignment of variable aborted: %s\n");
1707 jeavmp:
1708 n_err(V_(value), avcp->avc_name);
1709 goto jleave;
1712 /* Transformations */
1713 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_LOWER)){
1714 char c;
1716 oval = savestr(value);
1717 value = oval;
1718 for(; (c = *oval) != '\0'; ++oval)
1719 *oval = lowerconv(c);
1722 /* Obsoletion warning */
1723 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_OBSOLETE) != 0))
1724 n_OBSOLETE2(_("obsoleted variable"), avcp->avc_name);
1727 /* Lookup possibly existing var. For */
1729 rv = TRU1;
1730 a_amv_var_lookup(avcp, (a_AMV_VLOOK_I3VAL_NONEW |
1731 ((avscf & a_AMV_VSETCLR_LOCAL)
1732 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0)));
1733 avp = avcp->avc_var;
1735 /* A `local' setting is never covered by `localopts' nor frozen */
1736 if(avscf & a_AMV_VSETCLR_LOCAL)
1737 goto jislocal;
1739 /* If this setting had been established via -S and we still have not reached
1740 * the _STARTED_CONFIG program state, silently ignore request! */
1741 if(n_UNLIKELY(avp != NULL) &&
1742 n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1743 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1744 if((n_pstate & n_PS_ROOT) ||
1745 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1746 (n_poption & n_PO_S_FLAG_TEMPORARY)))
1747 goto joval_and_go;
1748 if(n_poption & n_PO_D_VV)
1749 n_err(_("Temporarily frozen by -S, not `set'ing: %s\n"),
1750 avcp->avc_name);
1751 goto jleave;
1754 /* Otherwise, if -S freezing was an `unset' request, be very simple and
1755 * avoid tampering with that very special case we are not really prepared
1756 * for just one more line of code: throw the old thing away! */
1757 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET))
1758 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1759 else{
1760 assert(avp->av_value == n_empty);
1761 a_amv_vars[avcp->avc_prime] = avp->av_link;
1762 n_free(avp);
1763 avcp->avc_var = avp = NULL;
1767 /* Optionally cover by `localopts' */
1768 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1769 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1770 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1772 jislocal:
1773 if(avp != NULL)
1774 joval_and_go:
1775 oval = avp->av_value;
1776 else{
1777 size_t l;
1778 struct a_amv_var **avpp;
1780 if(avscf & a_AMV_VSETCLR_LOCAL){
1781 if((avpp = *a_amv_lopts->as_amcap->amca_local_vars) == NULL)
1782 avpp = *(a_amv_lopts->as_amcap->amca_local_vars =
1783 n_calloc(1, sizeof(*a_amv_lopts->as_amcap->amca_local_vars)));
1784 avpp += avcp->avc_prime;
1785 }else
1786 avpp = &a_amv_vars[avcp->avc_prime];
1788 l = strlen(avcp->avc_name) +1;
1789 avcp->avc_var = avp = n_calloc(1,
1790 n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1791 avp->av_link = *avpp;
1792 *avpp = avp;
1793 avp->av_flags = (avscf & a_AMV_VSETCLR_LOCAL)
1794 ? a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL
1795 : (avmp != NULL) ? avmp->avm_flags : 0;
1796 memcpy(avp->av_name, avcp->avc_name, l);
1797 oval = n_UNCONST(n_empty);
1800 if(avmp == NULL)
1801 avp->av_value = a_amv_var_copy(value);
1802 else{
1803 assert(!(avscf & a_AMV_VSETCLR_LOCAL));
1804 /* Via `set' etc. the user may give even boolean options non-boolean
1805 * values, ignore that and force boolean */
1806 if(!(avp->av_flags & a_AMV_VF_BOOL))
1807 avp->av_value = a_amv_var_copy(value);
1808 else{
1809 if(!(n_pstate & n_PS_ROOT) && (n_poption & n_PO_D_V) &&
1810 *value != '\0')
1811 n_err(_("Ignoring value of boolean variable: %s: %s\n"),
1812 avcp->avc_name, value);
1813 avp->av_value = n_UNCONST(n_1);
1817 /* A `local' setting can skip all the crude special things */
1818 if(!(avscf & a_AMV_VSETCLR_LOCAL)){
1819 if((avscf & a_AMV_VSETCLR_ENV) && !(avp->av_flags & a_AMV_VF_ENV))
1820 avp->av_flags |= a_AMV_VF_EXT_LINKED;
1821 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1822 rv = a_amv_var__putenv(avcp, avp);
1823 if(avp->av_flags & a_AMV_VF_VIP)
1824 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, value);
1826 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1827 if(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1828 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0)
1829 avp->av_flags |= a_AMV_VF_EXT_FROZEN;
1832 a_amv_var_free(oval);
1833 jleave:
1834 NYD2_LEAVE;
1835 return rv;
1838 static bool_t
1839 a_amv_var__putenv(struct a_amv_var_carrier *avcp, struct a_amv_var *avp){
1840 #ifndef HAVE_SETENV
1841 char *cp;
1842 #endif
1843 bool_t rv;
1844 NYD2_ENTER;
1846 #ifdef HAVE_SETENV
1847 rv = (setenv(avcp->avc_name, avp->av_value, 1) == 0);
1848 #else
1849 cp = sstrdup(savecatsep(avcp->avc_name, '=', avp->av_value));
1851 if((rv = (putenv(cp) == 0))){
1852 char *ocp;
1854 ocp = avp->av_env;
1855 avp->av_env = cp;
1856 cp = ocp;
1859 if(cp != NULL)
1860 n_free(cp);
1861 #endif
1862 NYD2_LEAVE;
1863 return rv;
1866 static bool_t
1867 a_amv_var_clear(struct a_amv_var_carrier *avcp,
1868 enum a_amv_var_setclr_flags avscf){
1869 struct a_amv_var **avpp, *avp;
1870 struct a_amv_var_map const *avmp;
1871 bool_t rv;
1872 NYD2_ENTER;
1874 rv = FAL0;
1876 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1877 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NODEL) != 0 &&
1878 !(n_pstate & n_PS_ROOT))){
1879 n_err(_("Variable may not be unset: %s\n"), avcp->avc_name);
1880 goto jleave;
1882 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1883 !a_amv_var_check_vips(a_AMV_VIP_CLEAR, avcp->avc_okey, NULL))){
1884 n_err(_("Clearance of variable aborted: %s\n"), avcp->avc_name);
1885 goto jleave;
1889 rv = TRU1;
1891 if(n_UNLIKELY(!a_amv_var_lookup(avcp,
1892 (((avscf & a_AMV_VSETCLR_LOCAL)
1893 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0) |
1894 a_AMV_VLOOK_I3VAL_NONEW | a_AMV_VLOOK_I3VAL_NONEW_REPORT)))){
1895 assert(avcp->avc_var == NULL);
1896 /* This may be a clearance request from the command line, via -S, and we
1897 * need to keep track of that! Unfortunately we are not prepared for
1898 * this, really, so we need to create a fake entry that is known and
1899 * handled correctly by the lowermost variable layer!
1900 * However, all this cannot happen for plain unset of `local' variables */
1901 if(avscf & a_AMV_VSETCLR_LOCAL)
1902 goto jleave;
1903 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
1904 (n_poption & n_PO_S_FLAG_TEMPORARY)) Jfreeze:{
1905 size_t l;
1907 l = strlen(avcp->avc_name) +1;
1908 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1909 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1910 *avpp = avp;
1911 avp->av_value = n_UNCONST(n_empty); /* Sth. covered by _var_free()! */
1912 avp->av_flags = (avmp != NULL ? avmp->avm_flags : 0) |
1913 a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET;
1914 memcpy(avp->av_name, avcp->avc_name, l);
1916 if((avscf & a_AMV_VSETCLR_ENV) ||
1917 (avmp != NULL && (avmp->avm_flags & a_AMV_VF_ENV)))
1918 a_amv_var__clearenv(avcp->avc_name, NULL);
1919 }else if(avscf & a_AMV_VSETCLR_ENV){
1920 jforce_env:
1921 if(!(rv = a_amv_var__clearenv(avcp->avc_name, NULL)))
1922 goto jerr_env_unset;
1923 }else{
1924 /* TODO "cannot unset undefined variable" not echoed in "ROBOT" state,
1925 * TODO should only be like that with "ignerr"! */
1926 jerr_env_unset:
1927 if(!(n_pstate & (n_PS_ROOT | n_PS_ROBOT)) && (n_poption & n_PO_D_V))
1928 n_err(_("Cannot unset undefined variable: %s\n"), avcp->avc_name);
1930 goto jleave;
1931 }else if((avp = avcp->avc_var) == (struct a_amv_var*)-1){
1932 /* Clearance request from command line, via -S? As above.. */
1933 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1934 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0))
1935 goto Jfreeze;
1936 avcp->avc_var = NULL;
1937 if(avscf & a_AMV_VSETCLR_ENV)
1938 goto jforce_env;
1939 goto jleave;
1941 assert(avcp->avc_var != NULL);
1943 /* `local' variables bypass "frozen" checks and `localopts' coverage etc. */
1944 if(avp->av_flags & a_AMV_VF_EXT_LOCAL)
1945 goto jdefault_path;
1947 /* If this setting has been established via -S and we still have not reached
1948 * the _STARTED_CONFIG program state, silently ignore request!
1949 * XXX All this is very complicated for the tenth of a second */
1950 /*else*/ if(n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1951 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1952 if((n_pstate & n_PS_ROOT) ||
1953 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1954 (n_poption & n_PO_S_FLAG_TEMPORARY))){
1955 /* Be aware this may turn a set into an unset! */
1956 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET)){
1957 if(avp->av_flags & a_AMV_VF_DEFVAL)
1958 goto jdefault_path;
1959 a_amv_var_free(avp->av_value);
1960 avp->av_flags |= a_AMV_VF_EXT_FROZEN_UNSET;
1961 avp->av_value = n_UNCONST(n_empty); /* _var_free() covered */
1962 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1963 goto jforce_env;
1965 goto jleave;
1967 if(n_poption & n_PO_D_VV)
1968 n_err(_("Temporarily frozen by -S, not `unset'ting: %s\n"),
1969 avcp->avc_name);
1970 goto jleave;
1972 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1975 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1976 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1977 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1979 jdefault_path:
1980 assert(avp == avcp->avc_var);
1981 avcp->avc_var = NULL;
1982 avpp = &((avp->av_flags & a_AMV_VF_EXT_LOCAL)
1983 ? *a_amv_lopts->as_amcap->amca_local_vars : a_amv_vars
1984 )[avcp->avc_prime];
1985 assert(*avpp == avp); /* (always listhead after lookup()) */
1986 *avpp = (*avpp)->av_link;
1988 if(avp->av_flags & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1989 rv = a_amv_var__clearenv(avp->av_name, avp);
1990 a_amv_var_free(avp->av_value);
1991 n_free(avp);
1993 /* XXX Fun part, extremely simple-minded for now: if this variable has
1994 * XXX a default value, immediately reinstantiate it! TODO Heh? */
1995 if(n_UNLIKELY(avmp != NULL && (avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)){
1996 a_amv_var_lookup(avcp, a_AMV_VLOOK_I3VAL_NONEW);
1997 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
1998 (n_poption & n_PO_S_FLAG_TEMPORARY))
1999 avp->av_flags |= a_AMV_VF_EXT_FROZEN;
2001 jleave:
2002 NYD2_LEAVE;
2003 return rv;
2006 static bool_t
2007 a_amv_var__clearenv(char const *name, struct a_amv_var *avp){
2008 extern char **environ;
2009 char **ecpp;
2010 bool_t rv;
2011 NYD2_ENTER;
2012 n_UNUSED(avp);
2014 rv = FAL0;
2015 ecpp = environ;
2017 #ifndef HAVE_SETENV
2018 if(avp != NULL && avp->av_env != NULL){
2019 for(; *ecpp != NULL; ++ecpp)
2020 if(*ecpp == avp->av_env){
2022 ecpp[0] = ecpp[1];
2023 while(*ecpp++ != NULL);
2024 n_free(avp->av_env);
2025 avp->av_env = NULL;
2026 rv = TRU1;
2027 break;
2029 }else
2030 #endif
2032 size_t l;
2034 if((l = strlen(name)) > 0){
2035 for(; *ecpp != NULL; ++ecpp)
2036 if(!strncmp(*ecpp, name, l) && (*ecpp)[l] == '='){
2037 #ifdef HAVE_SETENV
2038 unsetenv(name);
2039 #else
2041 ecpp[0] = ecpp[1];
2042 while(*ecpp++ != NULL);
2043 #endif
2044 rv = TRU1;
2045 break;
2049 NYD2_LEAVE;
2050 return rv;
2053 static void
2054 a_amv_var_show_all(void){
2055 struct n_string msg, *msgp;
2056 FILE *fp;
2057 size_t no, i;
2058 struct a_amv_var *avp;
2059 char const **vacp, **cap;
2060 NYD2_ENTER;
2062 if((fp = Ftmp(NULL, "setlist", OF_RDWR | OF_UNLINK | OF_REGISTER)) == NULL){
2063 n_perr(_("`set' list: cannot create temporary file"), 0);
2064 goto jleave;
2067 /* We need to instantiate first-time-inits and default values here, so that
2068 * they will be regular members of our _vars[] table */
2069 for(i = a_AMV_VAR_I3VALS_CNT; i-- > 0;)
2070 n_var_oklook(a_amv_var_i3vals[i].avdv_okey);
2071 for(i = a_AMV_VAR_DEFVALS_CNT; i-- > 0;)
2072 n_var_oklook(a_amv_var_defvals[i].avdv_okey);
2074 for(no = i = 0; i < a_AMV_PRIME; ++i)
2075 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2076 ++no;
2077 no += a_AMV_VAR_VIRTS_CNT;
2079 vacp = n_autorec_alloc(no * sizeof(*vacp));
2081 for(cap = vacp, i = 0; i < a_AMV_PRIME; ++i)
2082 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2083 *cap++ = avp->av_name;
2084 for(i = a_AMV_VAR_VIRTS_CNT; i-- > 0;)
2085 *cap++ = a_amv_var_virts[i].avv_var->av_name;
2087 if(no > 1)
2088 qsort(vacp, no, sizeof *vacp, &a_amv_var__show_cmp);
2090 msgp = &msg;
2091 msgp = n_string_reserve(n_string_creat(msgp), 80);
2092 for(i = 0, cap = vacp; no != 0; ++cap, --no)
2093 i += a_amv_var_show(*cap, fp, msgp);
2094 n_string_gut(&msg);
2096 page_or_print(fp, i);
2097 Fclose(fp);
2098 jleave:
2099 NYD2_LEAVE;
2102 static int
2103 a_amv_var__show_cmp(void const *s1, void const *s2){
2104 int rv;
2105 NYD2_ENTER;
2107 rv = strcmp(*(char**)n_UNCONST(s1), *(char**)n_UNCONST(s2));
2108 NYD2_LEAVE;
2109 return rv;
2112 static size_t
2113 a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp){
2114 struct a_amv_var_carrier avc;
2115 char const *quote;
2116 struct a_amv_var *avp;
2117 bool_t isset;
2118 size_t i;
2119 NYD2_ENTER;
2121 msgp = n_string_trunc(msgp, 0);
2122 i = 0;
2124 a_amv_var_revlookup(&avc, name);
2125 isset = a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
2126 avp = avc.avc_var;
2128 if(n_poption & n_PO_D_V){
2129 if(avc.avc_map == NULL){
2130 msgp = n_string_push_cp(msgp, "#assembled variable with value");
2131 i = 1;
2132 }else{
2133 struct{
2134 ui16_t flag;
2135 char msg[22];
2136 } const tbase[] = {
2137 {a_AMV_VF_CHAIN, "variable chain"},
2138 {a_AMV_VF_VIRT, "virtual"},
2139 {a_AMV_VF_RDONLY, "read-only"},
2140 {a_AMV_VF_NODEL, "nodelete"},
2141 {a_AMV_VF_I3VAL, "initial-value"},
2142 {a_AMV_VF_DEFVAL, "default-value"},
2143 {a_AMV_VF_IMPORT, "import-environ-first\0"}, /* assert NUL in max */
2144 {a_AMV_VF_ENV, "sync-environ"},
2145 {a_AMV_VF_NOLOPTS, "no-localopts"},
2146 {a_AMV_VF_NOTEMPTY, "notempty"},
2147 {a_AMV_VF_NUM, "number"},
2148 {a_AMV_VF_POSNUM, "positive-number"},
2149 {a_AMV_VF_OBSOLETE, "obsoleted"},
2150 }, *tp;
2152 for(tp = tbase; PTRCMP(tp, <, &tbase[n_NELEM(tbase)]); ++tp)
2153 if(isset ? (avp->av_flags & tp->flag)
2154 : (avc.avc_map->avm_flags & tp->flag)){
2155 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2156 msgp = n_string_push_cp(msgp, tp->msg);
2160 if(avp != NULL && (avp->av_flags & a_AMV_VF_EXT_FROZEN)){
2161 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2162 msgp = n_string_push_cp(msgp, "(un)?set via -S");
2165 if(i > 0)
2166 msgp = n_string_push_cp(msgp, "\n ");
2169 /* (Read-only variables are generally shown via comments..) */
2170 if(!isset || (avp->av_flags & a_AMV_VF_RDONLY)){
2171 msgp = n_string_push_c(msgp, n_ns[0]);
2172 if(!isset){
2173 if(avc.avc_map != NULL && (avc.avc_map->avm_flags & a_AMV_VF_BOOL))
2174 msgp = n_string_push_cp(msgp, "boolean; ");
2175 msgp = n_string_push_cp(msgp, "variable not set: ");
2176 msgp = n_string_push_cp(msgp, n_shexp_quote_cp(name, FAL0));
2177 goto jleave;
2181 n_UNINIT(quote, NULL);
2182 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2183 quote = n_shexp_quote_cp(avp->av_value, TRU1);
2184 if(strcmp(quote, avp->av_value))
2185 msgp = n_string_push_cp(msgp, "wysh ");
2186 }else if(n_poption & n_PO_D_V)
2187 msgp = n_string_push_cp(msgp, "wysh "); /* (for shell-style comment) */
2189 if(avp->av_flags & a_AMV_VF_EXT_LINKED)
2190 msgp = n_string_push_cp(msgp, "environ ");
2191 msgp = n_string_push_cp(msgp, "set ");
2192 msgp = n_string_push_cp(msgp, name);
2194 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2195 msgp = n_string_push_c(msgp, '=');
2196 msgp = n_string_push_cp(msgp, quote);
2197 }else if(n_poption & n_PO_D_V)
2198 msgp = n_string_push_cp(msgp, " #boolean");
2200 jleave:
2201 msgp = n_string_push_c(msgp, '\n');
2202 fputs(n_string_cp(msgp), fp);
2203 NYD2_ENTER;
2204 return (i > 0 ? 2 : 1);
2207 static bool_t
2208 a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf){
2209 char *cp, *cp2, *varbuf, c;
2210 size_t errs;
2211 NYD2_ENTER;
2213 errs = 0;
2214 jouter:
2215 while((cp = *ap++) != NULL){
2216 /* Isolate key */
2217 cp2 = varbuf = n_autorec_alloc(strlen(cp) +1);
2219 for(; (c = *cp) != '=' && c != '\0'; ++cp){
2220 if(cntrlchar(c) || spacechar(c)){
2221 n_err(_("Variable name with control or space character ignored: "
2222 "%s\n"), ap[-1]);
2223 ++errs;
2224 goto jouter;
2226 *cp2++ = c;
2228 *cp2 = '\0';
2229 if(c == '\0')
2230 cp = n_UNCONST(n_empty);
2231 else
2232 ++cp;
2234 if(varbuf == cp2){
2235 n_err(_("Empty variable name ignored\n"));
2236 ++errs;
2237 }else{
2238 struct a_amv_var_carrier avc;
2239 bool_t isunset;
2241 if((isunset = (varbuf[0] == 'n' && varbuf[1] == 'o')))
2242 varbuf = &varbuf[2];
2244 a_amv_var_revlookup(&avc, varbuf);
2246 if((avscf & a_AMV_VSETCLR_LOCAL) && avc.avc_map != NULL){
2247 if(n_poption & n_PO_D_V)
2248 n_err(_("Builtin variable not overwritten by `local': %s\n"),
2249 varbuf);
2250 ++errs;
2251 }else if(isunset)
2252 errs += !a_amv_var_clear(&avc, avscf);
2253 else
2254 errs += !a_amv_var_set(&avc, cp, avscf);
2257 NYD2_LEAVE;
2258 return (errs == 0);
2261 FL int
2262 c_define(void *v){
2263 int rv;
2264 char **args;
2265 NYD_ENTER;
2267 rv = 1;
2269 if((args = v)[0] == NULL){
2270 rv = (a_amv_mac_show(a_AMV_MF_NONE) == FAL0);
2271 goto jleave;
2274 if(args[1] == NULL || args[1][0] != '{' || args[1][1] != '\0' ||
2275 args[2] != NULL){
2276 n_err(_("Synopsis: define: <name> {\n"));
2277 goto jleave;
2280 rv = (a_amv_mac_def(args[0], a_AMV_MF_NONE) == FAL0);
2281 jleave:
2282 NYD_LEAVE;
2283 return rv;
2286 FL int
2287 c_undefine(void *v){
2288 int rv;
2289 char **args;
2290 NYD_ENTER;
2292 rv = 0;
2293 args = v;
2295 rv |= !a_amv_mac_undef(*args, a_AMV_MF_NONE);
2296 while(*++args != NULL);
2297 NYD_LEAVE;
2298 return rv;
2301 FL int
2302 c_call(void *vp){
2303 int rv;
2304 NYD_ENTER;
2306 rv = a_amv_mac_call(vp, FAL0);
2307 NYD_LEAVE;
2308 return rv;
2311 FL int
2312 c_call_if(void *vp){
2313 int rv;
2314 NYD_ENTER;
2316 rv = a_amv_mac_call(vp, TRU1);
2317 NYD_LEAVE;
2318 return rv;
2321 FL int
2322 c_account(void *v){
2323 struct a_amv_mac_call_args *amcap;
2324 struct a_amv_mac *amp;
2325 char **args;
2326 int rv, i, oqf, nqf;
2327 NYD_ENTER;
2329 rv = 1;
2331 if((args = v)[0] == NULL){
2332 rv = (a_amv_mac_show(a_AMV_MF_ACCOUNT) == FAL0);
2333 goto jleave;
2336 if(args[1] && args[1][0] == '{' && args[1][1] == '\0'){
2337 if(args[2] != NULL){
2338 n_err(_("Synopsis: account: <name> {\n"));
2339 goto jleave;
2341 if(!asccasecmp(args[0], ACCOUNT_NULL)){
2342 n_err(_("`account': cannot use reserved name: %s\n"),
2343 ACCOUNT_NULL);
2344 goto jleave;
2346 rv = (a_amv_mac_def(args[0], a_AMV_MF_ACCOUNT) == FAL0);
2347 goto jleave;
2350 if(n_pstate & n_PS_HOOK_MASK){
2351 n_err(_("`account': cannot change account from within a hook\n"));
2352 goto jleave;
2355 save_mbox_for_possible_quitstuff();
2357 amp = NULL;
2358 if(asccasecmp(args[0], ACCOUNT_NULL) != 0 &&
2359 (amp = a_amv_mac_lookup(args[0], NULL, a_AMV_MF_ACCOUNT)) == NULL){
2360 n_err(_("`account': account does not exist: %s\n"), args[0]);
2361 goto jleave;
2364 oqf = savequitflags();
2366 /* Shutdown the active account */
2367 if(a_amv_acc_curr != NULL){
2368 char const *cp;
2369 char *var;
2371 /* Is there a cleanup hook? */
2372 var = savecat("on-account-cleanup-", a_amv_acc_curr->am_name);
2373 if((cp = n_var_vlook(var, FAL0)) != NULL ||
2374 (cp = ok_vlook(on_account_cleanup)) != NULL){
2375 struct a_amv_mac *amphook;
2377 if((amphook = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) != NULL){
2378 amcap = n_lofi_alloc(sizeof *amcap);
2379 memset(amcap, 0, sizeof *amcap);
2380 amcap->amca_name = cp;
2381 amcap->amca_amp = amphook;
2382 amcap->amca_unroller = &a_amv_acc_curr->am_lopts;
2383 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2384 amcap->amca_no_xcall = TRU1;
2385 n_pstate |= n_PS_HOOK;
2386 rv = a_amv_mac_exec(amcap);
2387 n_pstate &= ~n_PS_HOOK_MASK;
2388 }else
2389 n_err(_("*on-account-leave* hook %s does not exist\n"),
2390 n_shexp_quote_cp(cp, FAL0));
2393 /* `localopts'? */
2394 if(a_amv_acc_curr->am_lopts != NULL)
2395 a_amv_lopts_unroll(&a_amv_acc_curr->am_lopts);
2397 /* For accounts this lingers */
2398 --a_amv_acc_curr->am_refcnt;
2399 if(a_amv_acc_curr->am_flags & a_AMV_MF_DELETE)
2400 a_amv_mac_free(a_amv_acc_curr);
2403 a_amv_acc_curr = amp;
2405 /* And switch to any non-"null" account */
2406 if(amp != NULL){
2407 assert(amp->am_lopts == NULL);
2408 amcap = n_lofi_alloc(sizeof *amcap);
2409 memset(amcap, 0, sizeof *amcap);
2410 amcap->amca_name = amp->am_name;
2411 amcap->amca_amp = amp;
2412 amcap->amca_unroller = &amp->am_lopts;
2413 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2414 amcap->amca_no_xcall = TRU1;
2415 ++amp->am_refcnt; /* We may not run 0 to avoid being deleted! */
2416 if(!a_amv_mac_exec(amcap)){
2417 /* XXX account switch incomplete, unroll? */
2418 n_err(_("`account': failed to switch to account: %s\n"), amp->am_name);
2419 goto jleave;
2423 n_PS_ROOT_BLOCK((amp != NULL ? ok_vset(account, amp->am_name)
2424 : ok_vclear(account)));
2426 if(n_psonce & n_PSO_STARTED){
2427 assert(!(n_pstate & n_PS_HOOK_MASK));
2428 nqf = savequitflags(); /* TODO obsolete (leave -> void -> new box!) */
2429 restorequitflags(oqf);
2430 if((i = setfile("%", 0)) < 0)
2431 goto jleave;
2432 temporary_folder_hook_check(FAL0);
2433 if(i > 0 && !ok_blook(emptystart))
2434 goto jleave;
2435 n_folder_announce(n_ANNOUNCE_CHANGE);
2436 restorequitflags(nqf);
2438 rv = 0;
2439 jleave:
2440 NYD_LEAVE;
2441 return rv;
2444 FL int
2445 c_unaccount(void *v){
2446 int rv;
2447 char **args;
2448 NYD_ENTER;
2450 rv = 0;
2451 args = v;
2453 rv |= !a_amv_mac_undef(*args, a_AMV_MF_ACCOUNT);
2454 while(*++args != NULL);
2455 NYD_LEAVE;
2456 return rv;
2459 FL int
2460 c_localopts(void *vp){
2461 enum a_amv_loflags alf, alm;
2462 char const **argv;
2463 int rv;
2464 NYD_ENTER;
2466 rv = 1;
2468 if(a_amv_lopts == NULL){
2469 n_err(_("Cannot use `localopts' in this context\n"));
2470 goto jleave;
2473 if((argv = vp)[1] == NULL || is_asccaseprefix((++argv)[-1], "scope"))
2474 alf = alm = a_AMV_LF_SCOPE;
2475 else if(is_asccaseprefix(argv[-1], "call"))
2476 alf = a_AMV_LF_CALL, alm = a_AMV_LF_CALL_MASK;
2477 else if(is_asccaseprefix(argv[-1], "call-fixate"))
2478 alf = a_AMV_LF_CALL_FIXATE, alm = a_AMV_LF_CALL_MASK;
2479 else{
2480 n_err(_("Synopsis: localopts: [<scope|call|call-fixate>] <boolean>\n"));
2481 goto jleave;
2484 if(alf == a_AMV_LF_SCOPE &&
2485 (a_amv_lopts->as_loflags & a_AMV_LF_SCOPE_FIXATE)){
2486 if(n_poption & n_PO_D_V)
2487 n_err(_("Cannot turn off `localopts', setting is fixated\n"));
2488 goto jleave;
2491 a_amv_lopts->as_loflags &= ~alm;
2492 if(boolify(*argv, UIZ_MAX, FAL0) > 0)
2493 a_amv_lopts->as_loflags |= alf;
2494 rv = 0;
2495 jleave:
2496 NYD_LEAVE;
2497 return rv;
2500 FL int
2501 c_shift(void *vp){ /* xxx move to bottom, not in macro part! */
2502 struct a_amv_pospar *appp;
2503 ui16_t i;
2504 int rv;
2505 NYD_ENTER;
2507 rv = 1;
2509 if((vp = *(char**)vp) == NULL)
2510 i = 1;
2511 else{
2512 si16_t sib;
2514 if((n_idec_si16_cp(&sib, vp, 10, NULL
2515 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2516 ) != n_IDEC_STATE_CONSUMED || sib < 0){
2517 n_err(_("`shift': invalid argument: %s\n"), vp);
2518 goto jleave;
2520 i = (ui16_t)sib;
2523 /* If in in a macro/xy */
2524 if(a_amv_lopts != NULL){
2525 struct a_amv_mac const *amp;
2526 struct a_amv_mac_call_args *amcap;
2528 /* Explicitly do allow `vpospar' created things! */
2529 amp = (amcap = a_amv_lopts->as_amcap)->amca_amp;
2530 if((amp == NULL || amcap->amca_ps_hook_mask ||
2531 (amp->am_flags & a_AMV_MF_TYPE_MASK) == a_AMV_MF_ACCOUNT) &&
2532 amcap->amca_pospar.app_not_heap){
2533 n_err(_("Cannot use `shift' in `account's or hook macros etc.\n"));
2534 goto jleave;
2536 appp = &amcap->amca_pospar;
2537 }else
2538 appp = &a_amv_pospar;
2540 if(i > appp->app_count){
2541 n_err(_("`shift': cannot shift %hu of %hu parameters\n"),
2542 i, appp->app_count);
2543 goto jleave;
2544 }else{
2545 appp->app_idx += i;
2546 appp->app_count -= i;
2547 rv = 0;
2549 jleave:
2550 NYD_LEAVE;
2551 return rv;
2554 FL int
2555 c_return(void *vp){ /* TODO the exit status should be m_si64! */
2556 int rv;
2557 NYD_ENTER;
2559 if(a_amv_lopts != NULL){
2560 char const **argv;
2562 n_go_input_force_eof();
2563 n_pstate_err_no = n_ERR_NONE;
2564 rv = 0;
2566 if((argv = vp)[0] != NULL){
2567 si32_t i;
2569 if((n_idec_si32_cp(&i, argv[0], 10, NULL
2570 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2571 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2572 rv = (int)i;
2573 else{
2574 n_err(_("`return': return value argument is invalid: %s\n"),
2575 argv[0]);
2576 n_pstate_err_no = n_ERR_INVAL;
2577 rv = 1;
2580 if(argv[1] != NULL){
2581 if((n_idec_si32_cp(&i, argv[1], 10, NULL
2582 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2583 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2584 n_pstate_err_no = i;
2585 else{
2586 n_err(_("`return': error number argument is invalid: %s\n"),
2587 argv[1]);
2588 n_pstate_err_no = n_ERR_INVAL;
2589 rv = 1;
2593 }else{
2594 n_err(_("Can only use `return' in a macro\n"));
2595 n_pstate_err_no = n_ERR_OPNOTSUPP;
2596 rv = 1;
2598 NYD_LEAVE;
2599 return rv;
2602 FL bool_t
2603 temporary_folder_hook_check(bool_t nmail){ /* TODO temporary, v15: drop */
2604 struct a_amv_mac_call_args *amcap;
2605 struct a_amv_mac *amp;
2606 size_t len;
2607 char const *cp;
2608 char *var;
2609 bool_t rv;
2610 NYD_ENTER;
2612 rv = TRU1;
2613 var = n_autorec_alloc(len = strlen(mailname) + sizeof("folder-hook-") -1 +1);
2615 /* First try the fully resolved path */
2616 snprintf(var, len, "folder-hook-%s", mailname);
2617 if((cp = n_var_vlook(var, FAL0)) != NULL)
2618 goto jmac;
2620 /* If we are under *folder*, try the usual +NAME syntax, too */
2621 if(displayname[0] == '+'){
2622 char *x;
2624 for(x = &mailname[len]; x != mailname; --x)
2625 if(x[-1] == '/'){
2626 snprintf(var, len, "folder-hook-+%s", x);
2627 if((cp = n_var_vlook(var, FAL0)) != NULL)
2628 goto jmac;
2629 break;
2633 /* Plain *folder-hook* is our last try */
2634 if((cp = ok_vlook(folder_hook)) == NULL)
2635 goto jleave;
2637 jmac:
2638 if((amp = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) == NULL){
2639 n_err(_("Cannot call *folder-hook* for %s: macro does not exist: %s\n"),
2640 n_shexp_quote_cp(displayname, FAL0), cp);
2641 rv = FAL0;
2642 goto jleave;
2645 amcap = n_lofi_alloc(sizeof *amcap);
2646 memset(amcap, 0, sizeof *amcap);
2647 amcap->amca_name = cp;
2648 amcap->amca_amp = amp;
2649 n_pstate &= ~n_PS_HOOK_MASK;
2650 if(nmail){
2651 amcap->amca_unroller = NULL;
2652 n_pstate |= n_PS_HOOK_NEWMAIL;
2653 }else{
2654 amcap->amca_unroller = &a_amv_folder_hook_lopts;
2655 n_pstate |= n_PS_HOOK;
2657 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2658 amcap->amca_ps_hook_mask = TRU1;
2659 amcap->amca_no_xcall = TRU1;
2660 rv = a_amv_mac_exec(amcap);
2661 n_pstate &= ~n_PS_HOOK_MASK;
2663 jleave:
2664 NYD_LEAVE;
2665 return rv;
2668 FL void
2669 temporary_folder_hook_unroll(void){ /* XXX intermediate hack */
2670 NYD_ENTER;
2671 if(a_amv_folder_hook_lopts != NULL){
2672 void *save = a_amv_lopts;
2674 a_amv_lopts = NULL;
2675 a_amv_lopts_unroll(&a_amv_folder_hook_lopts);
2676 assert(a_amv_folder_hook_lopts == NULL);
2677 a_amv_lopts = save;
2679 NYD_LEAVE;
2682 FL void
2683 temporary_compose_mode_hook_call(char const *macname,
2684 void (*hook_pre)(void *), void *hook_arg){
2685 /* TODO compose_mode_hook_call() temporary, v15: generalize; see a_GO_SPLICE
2686 * TODO comment in go.c for the right way of doing things! */
2687 static struct a_amv_lostack *cmh_losp;
2688 struct a_amv_mac_call_args *amcap;
2689 struct a_amv_mac *amp;
2690 NYD_ENTER;
2692 amp = NULL;
2694 if(macname == (char*)-1){
2695 a_amv_mac__finalize(cmh_losp);
2696 cmh_losp = NULL;
2697 }else if(macname != NULL &&
2698 (amp = a_amv_mac_lookup(macname, NULL, a_AMV_MF_NONE)) == NULL)
2699 n_err(_("Cannot call *on-compose-**: macro does not exist: %s\n"),
2700 macname);
2701 else{
2702 amcap = n_lofi_alloc(sizeof *amcap);
2703 memset(amcap, 0, sizeof *amcap);
2704 amcap->amca_name = (macname != NULL) ? macname
2705 : "*on-compose-splice(-shell)?*";
2706 amcap->amca_amp = amp;
2707 amcap->amca_unroller = &a_amv_compose_lopts;
2708 amcap->amca_hook_pre = hook_pre;
2709 amcap->amca_hook_arg = hook_arg;
2710 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2711 amcap->amca_ps_hook_mask = TRU1;
2712 amcap->amca_no_xcall = TRU1;
2713 n_pstate &= ~n_PS_HOOK_MASK;
2714 n_pstate |= n_PS_HOOK;
2715 if(macname != NULL)
2716 a_amv_mac_exec(amcap);
2717 else{
2718 cmh_losp = n_lofi_alloc(sizeof *cmh_losp);
2719 memset(cmh_losp, 0, sizeof *cmh_losp);
2720 cmh_losp->as_global_saved = a_amv_lopts;
2721 cmh_losp->as_lopts = *(cmh_losp->as_amcap = amcap)->amca_unroller;
2722 cmh_losp->as_loflags = a_AMV_LF_SCOPE_FIXATE;
2723 a_amv_lopts = cmh_losp;
2726 NYD_LEAVE;
2729 FL void
2730 temporary_compose_mode_hook_unroll(void){ /* XXX intermediate hack */
2731 NYD_ENTER;
2732 if(a_amv_compose_lopts != NULL){
2733 void *save = a_amv_lopts;
2735 a_amv_lopts = NULL;
2736 a_amv_lopts_unroll(&a_amv_compose_lopts);
2737 assert(a_amv_compose_lopts == NULL);
2738 a_amv_lopts = save;
2740 NYD_LEAVE;
2743 FL bool_t
2744 n_var_is_user_writable(char const *name){
2745 struct a_amv_var_carrier avc;
2746 struct a_amv_var_map const *avmp;
2747 bool_t rv;
2748 NYD_ENTER;
2750 a_amv_var_revlookup(&avc, name);
2751 if((avmp = avc.avc_map) == NULL)
2752 rv = TRU1;
2753 else
2754 rv = ((avmp->avm_flags & (a_AMV_VF_BOOL | a_AMV_VF_RDONLY)) == 0);
2755 NYD_LEAVE;
2756 return rv;
2759 FL char *
2760 n_var_oklook(enum okeys okey){
2761 struct a_amv_var_carrier avc;
2762 char *rv;
2763 struct a_amv_var_map const *avmp;
2764 NYD_ENTER;
2766 avc.avc_map = avmp = &a_amv_var_map[okey];
2767 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2768 avc.avc_hash = avmp->avm_hash;
2769 avc.avc_okey = okey;
2771 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE))
2772 rv = avc.avc_var->av_value;
2773 else
2774 rv = NULL;
2775 NYD_LEAVE;
2776 return rv;
2779 FL bool_t
2780 n_var_okset(enum okeys okey, uintptr_t val){
2781 struct a_amv_var_carrier avc;
2782 bool_t ok;
2783 struct a_amv_var_map const *avmp;
2784 NYD_ENTER;
2786 avc.avc_map = avmp = &a_amv_var_map[okey];
2787 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2788 avc.avc_hash = avmp->avm_hash;
2789 avc.avc_okey = okey;
2791 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
2792 a_AMV_VSETCLR_NONE);
2793 NYD_LEAVE;
2794 return ok;
2797 FL bool_t
2798 n_var_okclear(enum okeys okey){
2799 struct a_amv_var_carrier avc;
2800 bool_t rv;
2801 struct a_amv_var_map const *avmp;
2802 NYD_ENTER;
2804 avc.avc_map = avmp = &a_amv_var_map[okey];
2805 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2806 avc.avc_hash = avmp->avm_hash;
2807 avc.avc_okey = okey;
2809 rv = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
2810 NYD_LEAVE;
2811 return rv;
2814 FL char const *
2815 n_var_vlook(char const *vokey, bool_t try_getenv){
2816 struct a_amv_var_carrier avc;
2817 char const *rv;
2818 NYD_ENTER;
2820 a_amv_var_revlookup(&avc, vokey);
2822 switch((enum a_amv_var_special_category)avc.avc_special_cat){
2823 default: /* silence CC */
2824 case a_AMV_VSC_NONE:
2825 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_LOCAL))
2826 rv = avc.avc_var->av_value;
2827 /* Only check the environment for something that is otherwise unknown */
2828 else if(try_getenv && avc.avc_map == NULL)
2829 rv = getenv(vokey);
2830 else
2831 rv = NULL;
2832 break;
2833 case a_AMV_VSC_GLOBAL:
2834 rv = a_amv_var_vsc_global(&avc);
2835 break;
2836 case a_AMV_VSC_MULTIPLEX:
2837 rv = a_amv_var_vsc_multiplex(&avc);
2838 break;
2839 case a_AMV_VSC_POSPAR:
2840 case a_AMV_VSC_POSPAR_ENV:
2841 rv = a_amv_var_vsc_pospar(&avc);
2842 break;
2844 NYD_LEAVE;
2845 return rv;
2848 FL bool_t
2849 n_var_vexplode(void const **cookie){
2850 struct a_amv_pospar *appp;
2851 NYD_ENTER;
2853 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
2854 : &a_amv_pospar;
2855 *cookie = (appp->app_count > 0) ? &appp->app_dat[appp->app_idx] : NULL;
2856 NYD_LEAVE;
2857 return (*cookie != NULL);
2860 FL bool_t
2861 n_var_vset(char const *vokey, uintptr_t val){
2862 struct a_amv_var_carrier avc;
2863 bool_t ok;
2864 NYD_ENTER;
2866 a_amv_var_revlookup(&avc, vokey);
2868 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
2869 a_AMV_VSETCLR_NONE);
2870 NYD_LEAVE;
2871 return ok;
2874 FL bool_t
2875 n_var_vclear(char const *vokey){
2876 struct a_amv_var_carrier avc;
2877 bool_t ok;
2878 NYD_ENTER;
2880 a_amv_var_revlookup(&avc, vokey);
2882 ok = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
2883 NYD_LEAVE;
2884 return ok;
2887 #ifdef HAVE_SOCKETS
2888 FL char *
2889 n_var_xoklook(enum okeys okey, struct url const *urlp,
2890 enum okey_xlook_mode oxm){
2891 struct a_amv_var_carrier avc;
2892 struct str const *us;
2893 size_t nlen;
2894 char *nbuf, *rv;
2895 struct a_amv_var_map const *avmp;
2896 NYD_ENTER;
2898 assert(oxm & (OXM_PLAIN | OXM_H_P | OXM_U_H_P));
2900 /* For simplicity: allow this case too */
2901 if(!(oxm & (OXM_H_P | OXM_U_H_P))){
2902 nbuf = NULL;
2903 goto jplain;
2906 avc.avc_map = avmp = &a_amv_var_map[okey];
2907 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2908 avc.avc_okey = okey;
2910 us = (oxm & OXM_U_H_P) ? &urlp->url_u_h_p : &urlp->url_h_p;
2911 nlen = strlen(avc.avc_name);
2912 nbuf = n_lofi_alloc(nlen + 1 + us->l +1);
2913 memcpy(nbuf, avc.avc_name, nlen);
2914 nbuf[nlen++] = '-';
2916 /* One of .url_u_h_p and .url_h_p we test in here */
2917 memcpy(nbuf + nlen, us->s, us->l +1);
2918 avc.avc_name = a_amv_var_canonify(nbuf);
2919 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2920 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_LOCAL))
2921 goto jvar;
2923 /* The second */
2924 if(oxm & OXM_H_P){
2925 us = &urlp->url_h_p;
2926 memcpy(nbuf + nlen, us->s, us->l +1);
2927 avc.avc_name = a_amv_var_canonify(nbuf);
2928 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
2929 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_LOCAL)){
2930 jvar:
2931 rv = avc.avc_var->av_value;
2932 goto jleave;
2936 jplain:
2937 rv = (oxm & OXM_PLAIN) ? n_var_oklook(okey) : NULL;
2938 jleave:
2939 if(nbuf != NULL)
2940 n_lofi_free(nbuf);
2941 NYD_LEAVE;
2942 return rv;
2944 #endif /* HAVE_SOCKETS */
2946 FL int
2947 c_set(void *vp){
2948 int err;
2949 char **ap;
2950 NYD_ENTER;
2952 if(*(ap = vp) == NULL){
2953 a_amv_var_show_all();
2954 err = 0;
2955 }else{
2956 enum a_amv_var_setclr_flags avscf;
2958 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
2959 avscf = a_AMV_VSETCLR_NONE;
2960 else{
2961 if(a_amv_lopts == NULL){
2962 n_err(_("`set': cannot use `local' in this context\n"));
2963 err = 1;
2964 goto jleave;
2966 avscf = a_AMV_VSETCLR_LOCAL;
2968 err = !a_amv_var_c_set(ap, avscf);
2970 jleave:
2971 NYD_LEAVE;
2972 return err;
2975 FL int
2976 c_unset(void *vp){
2977 struct a_amv_var_carrier avc;
2978 char **ap;
2979 int err;
2980 enum a_amv_var_setclr_flags avscf;
2981 NYD_ENTER;
2983 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
2984 avscf = a_AMV_VSETCLR_NONE;
2985 else{
2986 if(a_amv_lopts == NULL){
2987 n_err(_("`unset': cannot use `local' in this context\n"));
2988 err = 1;
2989 goto jleave;
2991 avscf = a_AMV_VSETCLR_LOCAL;
2994 for(err = 0, ap = vp; *ap != NULL; ++ap){
2995 a_amv_var_revlookup(&avc, *ap);
2997 err |= !a_amv_var_clear(&avc, avscf);
2999 jleave:
3000 NYD_LEAVE;
3001 return err;
3004 FL int
3005 c_varshow(void *v){
3006 char **ap;
3007 NYD_ENTER;
3009 if(*(ap = v) == NULL)
3010 v = NULL;
3011 else{
3012 struct n_string msg, *msgp = &msg;
3014 msgp = n_string_creat(msgp);
3015 for(; *ap != NULL; ++ap)
3016 a_amv_var_show(*ap, n_stdout, msgp);
3017 n_string_gut(msgp);
3019 NYD_LEAVE;
3020 return (v == NULL ? !STOP : !OKAY); /* xxx 1:bad 0:good -- do some */
3023 FL int
3024 c_varedit(void *v){
3025 struct a_amv_var_carrier avc;
3026 FILE *of, *nf;
3027 char *val, **argv;
3028 int err;
3029 sighandler_type sigint;
3030 NYD_ENTER;
3032 sigint = safe_signal(SIGINT, SIG_IGN);
3034 for(err = 0, argv = v; *argv != NULL; ++argv){
3035 memset(&avc, 0, sizeof avc);
3037 a_amv_var_revlookup(&avc, *argv);
3039 if(avc.avc_map != NULL){
3040 if(avc.avc_map->avm_flags & a_AMV_VF_BOOL){
3041 n_err(_("`varedit': cannot edit boolean variable: %s\n"),
3042 avc.avc_name);
3043 continue;
3045 if(avc.avc_map->avm_flags & a_AMV_VF_RDONLY){
3046 n_err(_("`varedit': cannot edit read-only variable: %s\n"),
3047 avc.avc_name);
3048 continue;
3052 a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
3054 if((of = Ftmp(NULL, "varedit", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
3055 NULL){
3056 n_perr(_("`varedit': cannot create temporary file"), 0);
3057 err = 1;
3058 break;
3059 }else if(avc.avc_var != NULL && *(val = avc.avc_var->av_value) != '\0' &&
3060 sizeof *val != fwrite(val, strlen(val), sizeof *val, of)){
3061 n_perr(_("`varedit' failed to write old value to temporary file"), 0);
3062 Fclose(of);
3063 err = 1;
3064 continue;
3067 fflush_rewind(of);
3068 nf = run_editor(of, (off_t)-1, 'e', FAL0, NULL, NULL, SEND_MBOX, sigint);
3069 Fclose(of);
3071 if(nf != NULL){
3072 int c;
3073 char const *varres;
3074 off_t l;
3076 l = fsize(nf);
3077 if(UICMP(64, l, >=, UIZ_MAX -42)){
3078 n_err(_("`varedit': not enough memory to store variable: %s\n"),
3079 avc.avc_name);
3080 varres = n_empty;
3081 err = 1;
3082 }else{
3083 varres = val = n_autorec_alloc(l +1);
3084 for(; l > 0 && (c = getc(nf)) != EOF; --l)
3085 *val++ = c;
3086 *val++ = '\0';
3087 if(l != 0){
3088 n_err(_("`varedit': I/O while reading new value of: %s\n"),
3089 avc.avc_name);
3090 err = 1;
3094 if(!a_amv_var_set(&avc, varres, a_AMV_VSETCLR_NONE))
3095 err = 1;
3097 Fclose(nf);
3098 }else{
3099 n_err(_("`varedit': cannot start $EDITOR\n"));
3100 err = 1;
3101 break;
3105 safe_signal(SIGINT, sigint);
3106 NYD_LEAVE;
3107 return err;
3110 FL int
3111 c_environ(void *v){
3112 struct a_amv_var_carrier avc;
3113 int err;
3114 char **ap;
3115 bool_t islnk;
3116 NYD_ENTER;
3118 if((islnk = is_asccaseprefix(*(ap = v), "link")) ||
3119 is_asccaseprefix(*ap, "unlink")){
3120 for(err = 0; *++ap != NULL;){
3121 a_amv_var_revlookup(&avc, *ap);
3123 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE) && (islnk ||
3124 (avc.avc_var->av_flags & a_AMV_VF_EXT_LINKED))){
3125 if(!islnk){
3126 avc.avc_var->av_flags &= ~a_AMV_VF_EXT_LINKED;
3127 continue;
3128 }else if(avc.avc_var->av_flags &
3129 (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)){
3130 if(n_poption & n_PO_D_V)
3131 n_err(_("`environ': link: already established: %s\n"), *ap);
3132 continue;
3134 avc.avc_var->av_flags |= a_AMV_VF_EXT_LINKED;
3135 if(!(avc.avc_var->av_flags & a_AMV_VF_ENV))
3136 a_amv_var__putenv(&avc, avc.avc_var);
3137 }else if(!islnk){
3138 n_err(_("`environ': unlink: no link established: %s\n"), *ap);
3139 err = 1;
3140 }else{
3141 char const *evp = getenv(*ap);
3143 if(evp != NULL)
3144 err |= !a_amv_var_set(&avc, evp, a_AMV_VSETCLR_ENV);
3145 else{
3146 n_err(_("`environ': link: cannot link to non-existent: %s\n"),
3147 *ap);
3148 err = 1;
3152 }else if(is_asccaseprefix(*ap, "set"))
3153 err = !a_amv_var_c_set(++ap, a_AMV_VSETCLR_ENV);
3154 else if(is_asccaseprefix(*ap, "unset")){
3155 for(err = 0; *++ap != NULL;){
3156 a_amv_var_revlookup(&avc, *ap);
3158 if(!a_amv_var_clear(&avc, a_AMV_VSETCLR_ENV))
3159 err = 1;
3161 }else{
3162 n_err(_("Synopsis: environ: <link|set|unset> <variable>...\n"));
3163 err = 1;
3165 NYD_LEAVE;
3166 return err;
3169 FL int
3170 c_vexpr(void *v){ /* TODO POSIX expr(1) comp. exit status; overly complicat. */
3171 size_t i;
3172 enum n_idec_state ids;
3173 si64_t lhv, rhv;
3174 char op, varbuf[64 + 64 / 8 +1];
3175 char const **argv, *varname, *varres, *cp;
3176 enum{
3177 a_ERR = 1u<<0,
3178 a_SOFTOVERFLOW = 1u<<1,
3179 a_ISNUM = 1u<<2,
3180 a_ISDECIMAL = 1u<<3, /* Print only decimal result */
3181 a_SATURATED = 1u<<4,
3182 a_ICASE = 1u<<5,
3183 a_UNSIGNED_OP = 1u<<6, /* Unsigned right shift (share bit ok) */
3184 a_TMP = 1u<<30
3185 } f;
3186 NYD_ENTER;
3188 f = a_ERR;
3189 argv = v;
3190 varname = (n_pstate & n_PS_ARGMOD_VPUT) ? *argv++ : NULL;
3191 n_UNINIT(varres, n_empty);
3193 if((cp = argv[0])[0] == '\0')
3194 goto jesubcmd;
3196 if(cp[1] == '\0'){
3197 op = cp[0];
3198 jnumop:
3199 f |= a_ISNUM;
3200 switch(op){
3201 case '=':
3202 case '~':
3203 if(argv[1] == NULL || argv[2] != NULL)
3204 goto jesynopsis;
3206 if(*(cp = *++argv) == '\0')
3207 lhv = 0;
3208 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
3209 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3210 ) != n_IDEC_STATE_CONSUMED){
3211 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3212 goto jenum_range;
3213 f |= a_SOFTOVERFLOW;
3214 break;
3216 if(op == '~')
3217 lhv = ~lhv;
3218 break;
3220 case '+':
3221 case '-':
3222 case '*':
3223 case '/':
3224 case '%':
3225 case '|':
3226 case '&':
3227 case '^':
3228 case '<':
3229 case '>':
3230 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3231 goto jesynopsis;
3232 else{
3233 char xop;
3235 if(*(cp = *++argv) == '\0')
3236 lhv = 0;
3237 else if(((ids = n_idec_si64_cp(&lhv, cp, 0, NULL)
3238 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3239 ) != n_IDEC_STATE_CONSUMED){
3240 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3241 goto jenum_range;
3242 f |= a_SOFTOVERFLOW;
3243 break;
3246 if(*(cp = *++argv) == '\0')
3247 rhv = 0;
3248 else if(((ids = n_idec_si64_cp(&rhv, cp, 0, NULL)
3249 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3250 ) != n_IDEC_STATE_CONSUMED){
3251 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3252 goto jenum_range;
3253 f |= a_SOFTOVERFLOW;
3254 lhv = rhv;
3255 break;
3258 xop = op;
3259 jnumop_again:
3260 switch(xop){
3261 case '+':
3262 if(rhv < 0){
3263 if(rhv != SI64_MIN){
3264 rhv = -rhv;
3265 xop = '-';
3266 goto jnumop_again;
3267 }else if(lhv < 0)
3268 goto jenum_plusminus;
3269 else if(lhv == 0){
3270 lhv = rhv;
3271 break;
3273 }else if(SI64_MAX - rhv < lhv)
3274 goto jenum_plusminus;
3275 lhv += rhv;
3276 break;
3277 case '-':
3278 if(rhv < 0){
3279 if(rhv != SI64_MIN){
3280 rhv = -rhv;
3281 xop = '+';
3282 goto jnumop_again;
3283 }else if(lhv > 0)
3284 goto jenum_plusminus;
3285 else if(lhv == 0){
3286 lhv = rhv;
3287 break;
3289 }else if(SI64_MIN + rhv > lhv){
3290 jenum_plusminus:
3291 if(!(f & a_SATURATED))
3292 goto jenum_overflow;
3293 f |= a_SOFTOVERFLOW;
3294 lhv = (lhv < 0 || xop == '-') ? SI64_MIN : SI64_MAX;
3295 break;
3297 lhv -= rhv;
3298 break;
3299 case '*':
3300 /* Will the result be positive? */
3301 if((lhv < 0) == (rhv < 0)){
3302 if(lhv > 0){
3303 lhv = -lhv;
3304 rhv = -rhv;
3306 if(rhv != 0 && lhv != 0 && SI64_MAX / rhv > lhv){
3307 if(!(f & a_SATURATED))
3308 goto jenum_overflow;
3309 f |= a_SOFTOVERFLOW;
3310 lhv = SI64_MAX;
3311 }else
3312 lhv *= rhv;
3313 }else{
3314 if(rhv > 0){
3315 if(lhv != 0 && SI64_MIN / lhv < rhv){
3316 if(!(f & a_SATURATED))
3317 goto jenum_overflow;
3318 f |= a_SOFTOVERFLOW;
3319 lhv = SI64_MIN;
3320 }else
3321 lhv *= rhv;
3322 }else{
3323 if(rhv != 0 && lhv != 0 && SI64_MIN / rhv < lhv){
3324 if(!(f & a_SATURATED))
3325 goto jenum_overflow;
3326 f |= a_SOFTOVERFLOW;
3327 lhv = SI64_MIN;
3328 }else
3329 lhv *= rhv;
3332 break;
3333 case '/':
3334 if(rhv == 0){
3335 if(!(f & a_SATURATED))
3336 goto jenum_range;
3337 f |= a_SOFTOVERFLOW;
3338 lhv = SI64_MAX;
3339 }else
3340 lhv /= rhv;
3341 break;
3342 case '%':
3343 if(rhv == 0){
3344 if(!(f & a_SATURATED))
3345 goto jenum_range;
3346 f |= a_SOFTOVERFLOW;
3347 lhv = SI64_MAX;
3348 }else
3349 lhv %= rhv;
3350 break;
3351 case '|':
3352 lhv |= rhv;
3353 break;
3354 case '&':
3355 lhv &= rhv;
3356 break;
3357 case '^':
3358 lhv ^= rhv;
3359 break;
3360 case '<':
3361 case '>':
3362 if(!(f & a_TMP)){
3363 argv -= 2;
3364 goto jesubcmd;
3366 if(rhv > 63){ /* xxx 63? */
3367 if(!(f & a_SATURATED))
3368 goto jenum_overflow;
3369 rhv = 63;
3371 if(op == '<')
3372 lhv <<= (ui8_t)rhv;
3373 else if(f & a_UNSIGNED_OP)
3374 lhv = (ui64_t)lhv >> (ui8_t)rhv;
3375 else
3376 lhv >>= (ui8_t)rhv;
3377 break;
3380 break;
3381 default:
3382 goto jesubcmd;
3384 }else if(cp[0] == '<'){
3385 if(*++cp != '<')
3386 goto jesubcmd;
3387 if(*++cp == '@'){
3388 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3389 f |= a_SATURATED;
3390 ++cp;
3392 if(*cp != '\0')
3393 goto jesubcmd;
3394 f |= a_TMP;
3395 op = '<';
3396 goto jnumop;
3397 }else if(cp[0] == '>'){
3398 if(*++cp != '>')
3399 goto jesubcmd;
3400 if(*++cp == '>'){
3401 f |= a_UNSIGNED_OP;
3402 ++cp;
3404 if(*cp == '@'){
3405 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3406 f |= a_SATURATED;
3407 ++cp;
3409 if(*cp != '\0')
3410 goto jesubcmd;
3411 f |= a_TMP;
3412 op = '>';
3413 goto jnumop;
3414 }else if(cp[2] == '\0' && cp[1] == '@'){
3415 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix, not suffix"));
3416 f |= a_SATURATED;
3417 op = cp[0];
3418 goto jnumop;
3419 }else if(cp[0] == '@'){
3420 f |= a_SATURATED;
3421 op = *++cp;
3422 if(*++cp != '\0'){
3423 if(op != *cp)
3424 goto jesubcmd;
3425 switch(op){
3426 case '<':
3427 if(*++cp != '\0')
3428 goto jesubcmd;
3429 f |= a_TMP;
3430 break;
3431 case '>':
3432 if(*++cp != '\0'){
3433 if(*cp != '>' || *++cp != '\0')
3434 goto jesubcmd;
3435 f |= a_UNSIGNED_OP;
3437 f |= a_TMP;
3438 break;
3439 default:
3440 goto jesubcmd;
3443 goto jnumop;
3444 }else if(is_asccaseprefix(cp, "length")){
3445 f |= a_ISNUM | a_ISDECIMAL;
3446 if(argv[1] == NULL || argv[2] != NULL)
3447 goto jesynopsis;
3449 i = strlen(*++argv);
3450 if(UICMP(64, i, >, SI64_MAX))
3451 goto jestr_overflow;
3452 lhv = (si64_t)i;
3453 }else if(is_asccaseprefix(cp, "hash")){
3454 f |= a_ISNUM | a_ISDECIMAL;
3455 if(argv[1] == NULL || argv[2] != NULL)
3456 goto jesynopsis;
3458 i = n_torek_hash(*++argv);
3459 lhv = (si64_t)i;
3460 }else if(is_asccaseprefix(cp, "find")){
3461 f |= a_ISNUM | a_ISDECIMAL;
3462 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3463 goto jesynopsis;
3465 if((cp = strstr(argv[1], argv[2])) == NULL)
3466 goto jestr_nodata;
3467 i = PTR2SIZE(cp - argv[1]);
3468 if(UICMP(64, i, >, SI64_MAX))
3469 goto jestr_overflow;
3470 lhv = (si64_t)i;
3471 }else if(is_asccaseprefix(cp, "ifind")){
3472 f |= a_ISNUM | a_ISDECIMAL;
3473 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3474 goto jesynopsis;
3476 if((cp = asccasestr(argv[1], argv[2])) == NULL)
3477 goto jestr_nodata;
3478 i = PTR2SIZE(cp - argv[1]);
3479 if(UICMP(64, i, >, SI64_MAX))
3480 goto jestr_overflow;
3481 lhv = (si64_t)i;
3482 }else if(is_asccaseprefix(cp, "substring")){
3483 if(argv[1] == NULL || argv[2] == NULL)
3484 goto jesynopsis;
3485 if(argv[3] != NULL && argv[4] != NULL)
3486 goto jesynopsis;
3488 i = strlen(varres = *++argv);
3490 if(*(cp = *++argv) == '\0')
3491 lhv = 0;
3492 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3493 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3494 ) != n_IDEC_STATE_CONSUMED)
3495 goto jestr_numrange;
3496 if(lhv < 0){
3497 if(UICMP(64, i, <, -lhv))
3498 goto jesubstring_off;
3499 lhv = i + lhv;
3501 if(UICMP(64, i, >=, lhv)){
3502 i -= lhv;
3503 varres += lhv;
3504 }else{
3505 jesubstring_off:
3506 if(n_poption & n_PO_D_V)
3507 n_err(_("`vexpr': substring: offset argument too large: %s\n"),
3508 n_shexp_quote_cp(argv[-1], FAL0));
3509 f |= a_SOFTOVERFLOW;
3512 if(argv[1] != NULL){
3513 if(*(cp = *++argv) == '\0')
3514 lhv = 0;
3515 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3516 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3517 ) != n_IDEC_STATE_CONSUMED)
3518 goto jestr_numrange;
3519 if(lhv < 0){
3520 if(UICMP(64, i, <, -lhv))
3521 goto jesubstring_len;
3522 lhv = i + lhv;
3524 if(UICMP(64, i, >=, lhv)){
3525 if(UICMP(64, i, !=, lhv))
3526 varres = savestrbuf(varres, (size_t)lhv);
3527 }else{
3528 jesubstring_len:
3529 if(n_poption & n_PO_D_V)
3530 n_err(_("`vexpr': substring: length argument too large: %s\n"),
3531 n_shexp_quote_cp(argv[-2], FAL0));
3532 f |= a_SOFTOVERFLOW;
3535 }else if(is_asccaseprefix(cp, "trim")) Jtrim: {
3536 struct str trim;
3537 enum n_str_trim_flags stf;
3539 if(argv[1] == NULL || argv[2] != NULL)
3540 goto jesynopsis;
3542 if(is_asccaseprefix(cp, "trim-front"))
3543 stf = n_STR_TRIM_FRONT;
3544 else if(is_asccaseprefix(cp, "trim-end"))
3545 stf = n_STR_TRIM_END;
3546 else
3547 stf = n_STR_TRIM_BOTH;
3549 trim.l = strlen(trim.s = n_UNCONST(argv[1]));
3550 (void)n_str_trim(&trim, stf);
3551 varres = savestrbuf(trim.s, trim.l);
3552 }else if(is_asccaseprefix(cp, "trim-front"))
3553 goto Jtrim;
3554 else if(is_asccaseprefix(cp, "trim-end"))
3555 goto Jtrim;
3556 else if(is_asccaseprefix(cp, "random")){
3557 if(argv[1] == NULL || argv[2] != NULL)
3558 goto jesynopsis;
3560 if((n_idec_si64_cp(&lhv, argv[1], 0, NULL
3561 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3562 ) != n_IDEC_STATE_CONSUMED || lhv < 0 || lhv > PATH_MAX)
3563 goto jestr_numrange;
3564 if(lhv == 0)
3565 lhv = NAME_MAX;
3566 varres = n_random_create_cp((size_t)lhv, NULL);
3567 }else if(is_asccaseprefix(cp, "file-expand")){
3568 if(argv[1] == NULL || argv[2] != NULL)
3569 goto jesynopsis;
3571 if((varres = fexpand(argv[1], FEXP_NVAR | FEXP_NOPROTO)) == NULL)
3572 goto jestr_nodata;
3573 }else if(is_asccaseprefix(cp, "makeprint")){
3574 struct str sin, sout;
3576 if(argv[1] == NULL || argv[2] != NULL)
3577 goto jesynopsis;
3579 /* XXX using strlen for `vexpr makeprint' is wrong for UTF-16 */
3580 sin.l = strlen(sin.s = n_UNCONST(argv[1]));
3581 makeprint(&sin, &sout);
3582 varres = savestrbuf(sout.s, sout.l);
3583 n_free(sout.s);
3584 /* TODO `vexpr': (wide) string length, find, etc!! */
3585 #ifdef HAVE_REGEX
3586 }else if(is_asccaseprefix(cp, "regex")) Jregex:{
3587 regmatch_t rema[1 + n_VEXPR_REGEX_MAX];
3588 regex_t re;
3589 int reflrv;
3591 f |= a_ISNUM | a_ISDECIMAL;
3592 if(argv[1] == NULL || argv[2] == NULL ||
3593 (argv[3] != NULL && argv[4] != NULL))
3594 goto jesynopsis;
3596 reflrv = REG_EXTENDED;
3597 if(f & a_ICASE)
3598 reflrv |= REG_ICASE;
3599 if((reflrv = regcomp(&re, argv[2], reflrv))){
3600 n_err(_("`vexpr': invalid regular expression: %s: %s\n"),
3601 n_shexp_quote_cp(argv[2], FAL0), n_regex_err_to_doc(NULL, reflrv));
3602 assert(f & a_ERR);
3603 n_pstate_err_no = n_ERR_INVAL;
3604 goto jestr;
3606 reflrv = regexec(&re, argv[1], n_NELEM(rema), rema, 0);
3607 regfree(&re);
3608 if(reflrv == REG_NOMATCH)
3609 goto jestr_nodata;
3611 /* Search only? Else replace, which is a bit */
3612 if(argv[3] == NULL){
3613 if(UICMP(64, rema[0].rm_so, >, SI64_MAX))
3614 goto jestr_overflow;
3615 lhv = (si64_t)rema[0].rm_so;
3616 }else{
3617 /* We need to setup some kind of pseudo macro environment for this */
3618 struct a_amv_lostack los;
3619 struct a_amv_mac_call_args amca;
3620 char const **reargv;
3622 memset(&amca, 0, sizeof amca);
3623 amca.amca_name = savestrbuf(&argv[1][rema[0].rm_so],
3624 rema[0].rm_eo - rema[0].rm_so);
3625 amca.amca_amp = a_AMV_MACKY_MACK;
3626 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3628 amca.amca_pospar.app_count = (ui32_t)i - 1;
3629 amca.amca_pospar.app_not_heap = TRU1;
3630 amca.amca_pospar.app_dat =
3631 reargv = n_autorec_alloc(sizeof(char*) * i);
3632 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3633 *reargv++ = savestrbuf(&argv[1][rema[i].rm_so],
3634 rema[i].rm_eo - rema[i].rm_so);
3635 *reargv = NULL;
3637 memset(&los, 0, sizeof los);
3638 hold_all_sigs(); /* TODO DISLIKE! */
3639 los.as_global_saved = a_amv_lopts;
3640 los.as_amcap = &amca;
3641 los.as_up = los.as_global_saved;
3642 a_amv_lopts = &los;
3644 /* C99 */{
3645 struct str templ;
3646 struct n_string s_b;
3647 enum n_shexp_state shs;
3649 templ.s = n_UNCONST(argv[3]);
3650 templ.l = UIZ_MAX;
3651 shs = n_shexp_parse_token((n_SHEXP_PARSE_LOG |
3652 n_SHEXP_PARSE_IGNORE_EMPTY | n_SHEXP_PARSE_QUOTE_AUTO_FIXED |
3653 n_SHEXP_PARSE_QUOTE_AUTO_DSQ),
3654 n_string_creat_auto(&s_b), &templ, NULL);
3655 if((shs & (n_SHEXP_STATE_ERR_MASK | n_SHEXP_STATE_STOP)
3656 ) == n_SHEXP_STATE_STOP){
3657 varres = n_string_cp(&s_b);
3658 n_string_drop_ownership(&s_b);
3659 }else
3660 varres = NULL;
3663 a_amv_lopts = los.as_global_saved;
3664 rele_all_sigs(); /* TODO DISLIKE! */
3666 if(varres == NULL)
3667 goto jestr_nodata;
3668 f &= ~(a_ISNUM | a_ISDECIMAL);
3670 }else if(is_asccaseprefix(argv[0], "iregex")){
3671 f |= a_ICASE;
3672 goto Jregex;
3673 #endif /* HAVE_REGEX */
3674 }else
3675 goto jesubcmd;
3677 n_pstate_err_no = (f & a_SOFTOVERFLOW) ? n_ERR_OVERFLOW : n_ERR_NONE;
3678 f &= ~a_ERR;
3680 /* Generate the variable value content for numerics.
3681 * Anticipate in our handling below! (Don't do needless work) */
3682 jleave:
3683 if((f & a_ISNUM) && ((f & a_ISDECIMAL) || varname != NULL)){
3684 snprintf(varbuf, sizeof varbuf, "%" PRId64 , lhv);
3685 varres = varbuf;
3688 if(varname == NULL){
3689 /* If there was no error and we are printing a numeric result, print some
3690 * more bases for the fun of it */
3691 if((f & (a_ERR | a_ISNUM | a_ISDECIMAL)) == a_ISNUM){
3692 size_t j;
3694 for(j = 1, i = 0; i < 64; ++i){
3695 varbuf[63 + 64 / 8 -j - i] = (lhv & ((ui64_t)1 << i)) ? '1' : '0';
3696 if((i & 7) == 7 && i != 63){
3697 ++j;
3698 varbuf[63 + 64 / 8 -j - i] = ' ';
3701 varbuf[64 + 64 / 8 -1] = '\0';
3702 if(fprintf(n_stdout,
3703 "%s\n0%" PRIo64 " | 0x%" PRIX64 " | %" PRId64 "\n",
3704 varbuf, lhv, lhv, lhv) < 0){
3705 n_pstate_err_no = n_err_no;
3706 f |= a_ERR;
3708 }else if(varres != NULL && fprintf(n_stdout, "%s\n", varres) < 0){
3709 n_pstate_err_no = n_err_no;
3710 f |= a_ERR;
3712 }else if(!n_var_vset(varname, (uintptr_t)varres)){
3713 n_pstate_err_no = n_ERR_NOTSUP;
3714 f |= a_ERR;
3716 NYD_LEAVE;
3717 return (f & a_ERR) ? 1 : 0;
3719 jerr:
3720 f = a_ERR | a_ISNUM;
3721 lhv = -1;
3722 goto jleave;
3723 jesubcmd:
3724 n_err(_("`vexpr': invalid subcommand: %s\n"),
3725 n_shexp_quote_cp(*argv, FAL0));
3726 n_pstate_err_no = n_ERR_INVAL;
3727 goto jerr;
3728 jesynopsis:
3729 n_err(_("Synopsis: vexpr: <operator> <:argument:>\n"));
3730 n_pstate_err_no = n_ERR_INVAL;
3731 goto jerr;
3733 jenum_range:
3734 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3735 n_shexp_quote_cp(*argv, FAL0));
3736 n_pstate_err_no = n_ERR_RANGE;
3737 goto jerr;
3738 jenum_overflow:
3739 n_err(_("`vexpr': expression overflows datatype: %" PRId64 " %c %" PRId64
3740 "\n"), lhv, op, rhv);
3741 n_pstate_err_no = n_ERR_OVERFLOW;
3742 goto jerr;
3744 jestr_numrange:
3745 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3746 n_shexp_quote_cp(*argv, FAL0));
3747 n_pstate_err_no = n_ERR_RANGE;
3748 goto jestr;
3749 jestr_overflow:
3750 n_err(_("`vexpr': string length or offset overflows datatype\n"));
3751 n_pstate_err_no = n_ERR_OVERFLOW;
3752 goto jestr;
3753 jestr_nodata:
3754 n_pstate_err_no = n_ERR_NODATA;
3755 /* FALLTHRU */
3756 jestr:
3757 varres = n_empty;
3758 f &= ~a_ISNUM;
3759 f |= a_ERR;
3760 goto jleave;
3763 FL int
3764 c_vpospar(void *v){
3765 struct n_cmd_arg *cap;
3766 size_t i;
3767 struct a_amv_pospar *appp;
3768 enum{
3769 a_NONE = 0,
3770 a_ERR = 1u<<0,
3771 a_SET = 1u<<1,
3772 a_CLEAR = 1u<<2,
3773 a_QUOTE = 1u<<3
3774 } f;
3775 char const *varres;
3776 struct n_cmd_arg_ctx *cacp;
3777 NYD_ENTER;
3779 n_pstate_err_no = n_ERR_NONE;
3780 n_UNINIT(varres, n_empty);
3781 cacp = v;
3782 cap = cacp->cac_arg;
3784 if(is_asccaseprefix(cap->ca_arg.ca_str.s, "set"))
3785 f = a_SET;
3786 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "clear"))
3787 f = a_CLEAR;
3788 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "quote"))
3789 f = a_QUOTE;
3790 else{
3791 n_err(_("`vpospar': invalid subcommand: %s\n"),
3792 n_shexp_quote_cp(cap->ca_arg.ca_str.s, FAL0));
3793 n_pstate_err_no = n_ERR_INVAL;
3794 f = a_ERR;
3795 goto jleave;
3797 --cacp->cac_no;
3799 if((f & (a_CLEAR | a_QUOTE)) && cap->ca_next != NULL){
3800 n_err(_("`vpospar': `%s': takes no argument\n"), cap->ca_arg.ca_str.s);
3801 n_pstate_err_no = n_ERR_INVAL;
3802 f = a_ERR;
3803 goto jleave;
3806 cap = cap->ca_next;
3808 /* If in a macro, we need to overwrite the local instead of global argv */
3809 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
3810 : &a_amv_pospar;
3812 if(f & (a_SET | a_CLEAR)){
3813 if(cacp->cac_vput != NULL)
3814 n_err(_("`vpospar': `vput' only supported for `quote' subcommand\n"));
3815 if(!appp->app_not_heap && appp->app_maxcount > 0){
3816 for(i = appp->app_maxcount; i-- != 0;)
3817 n_free(n_UNCONST(appp->app_dat[i]));
3818 n_free(appp->app_dat);
3820 memset(appp, 0, sizeof *appp);
3822 if(f & a_SET){
3823 if((i = cacp->cac_no) > a_AMV_POSPAR_MAX){
3824 n_err(_("`vpospar': overflow: %" PRIuZ " arguments!\n"), i);
3825 n_pstate_err_no = n_ERR_OVERFLOW;
3826 f = a_ERR;
3827 goto jleave;
3830 memset(appp, 0, sizeof *appp);
3831 if(i > 0){
3832 appp->app_maxcount = appp->app_count = (ui16_t)i;
3833 /* XXX Optimize: store it all in one chunk! */
3834 ++i;
3835 i *= sizeof *appp->app_dat;
3836 appp->app_dat = n_alloc(i);
3838 for(i = 0; cap != NULL; ++i, cap = cap->ca_next){
3839 appp->app_dat[i] = n_alloc(cap->ca_arg.ca_str.l +1);
3840 memcpy(n_UNCONST(appp->app_dat[i]), cap->ca_arg.ca_str.s,
3841 cap->ca_arg.ca_str.l +1);
3844 appp->app_dat[i] = NULL;
3847 }else{
3848 if(appp->app_count == 0)
3849 varres = n_empty;
3850 else{
3851 struct str in;
3852 struct n_string s, *sp;
3853 char sep1, sep2;
3855 sp = n_string_creat_auto(&s);
3857 sep1 = *ok_vlook(ifs);
3858 sep2 = *ok_vlook(ifs_ws);
3859 if(sep1 == sep2)
3860 sep2 = '\0';
3861 if(sep1 == '\0')
3862 sep1 = ' ';
3864 for(i = 0; i < appp->app_count; ++i){
3865 if(sp->s_len){
3866 if(!n_string_can_swallow(sp, 2))
3867 goto jeover;
3868 sp = n_string_push_c(sp, sep1);
3869 if(sep2 != '\0')
3870 sp = n_string_push_c(sp, sep2);
3872 in.l = strlen(in.s = n_UNCONST(appp->app_dat[i + appp->app_idx]));
3874 if(!n_string_can_swallow(sp, in.l)){
3875 jeover:
3876 n_err(_("`vpospar': overflow: string too long!\n"));
3877 n_pstate_err_no = n_ERR_OVERFLOW;
3878 f = a_ERR;
3879 goto jleave;
3881 sp = n_shexp_quote(sp, &in, TRU1);
3884 varres = n_string_cp(sp);
3887 if(cacp->cac_vput == NULL){
3888 if(fprintf(n_stdout, "%s\n", varres) < 0){
3889 n_pstate_err_no = n_err_no;
3890 f |= a_ERR;
3892 }else if(!n_var_vset(cacp->cac_vput, (uintptr_t)varres)){
3893 n_pstate_err_no = n_ERR_NOTSUP;
3894 f |= a_ERR;
3897 jleave:
3898 NYD_LEAVE;
3899 return (f & a_ERR) ? 1 : 0;
3902 /* s-it-mode */