a_sendout_attach_msg(): oops, did not generate Content-ID:!
[s-mailx.git] / accmacvar.c
blobdbadec09637660a5aa35326878d28e33cb933079
1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
2 *@ Account, macro and variable handling; `vexpr' and `vpospar'.
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 (which is highly related..)
6 *@ - update the manual!
7 *@ TODO . `localopts' should act like an automatic permanent `scope' command
8 *@ TODO modifier! We need an OnScopeLeaveEvent, then.
9 *@ TODO Also see the a_GO_SPLICE comment in go.c.
10 *@ TODO . Optimize: with the dynamic hashmaps, and the object based approach
11 *@ TODO it should become possible to strip down the implementation again.
12 *@ TODO E.g., FREEZE is much too complicated: use an overlay object ptr,
13 *@ TODO UNLIKELY() it, and add a OnProgramStartupCompletedEvent to
14 *@ TODO incorporate what it tracks, then drop it. Etc.
15 *@ TODO Global -> Scope -> Local, all "overlay" objects.
17 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
18 * Copyright (c) 2012 - 2018 Steffen (Daode) Nurpmeso <steffen@sdaoden.eu>.
19 * SPDX-License-Identifier: BSD-3-Clause
22 * Copyright (c) 1980, 1993
23 * The Regents of the University of California. All rights reserved.
25 * Redistribution and use in source and binary forms, with or without
26 * modification, are permitted provided that the following conditions
27 * are met:
28 * 1. Redistributions of source code must retain the above copyright
29 * notice, this list of conditions and the following disclaimer.
30 * 2. Redistributions in binary form must reproduce the above copyright
31 * notice, this list of conditions and the following disclaimer in the
32 * documentation and/or other materials provided with the distribution.
33 * 3. Neither the name of the University nor the names of its contributors
34 * may be used to endorse or promote products derived from this software
35 * without specific prior written permission.
37 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
38 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
39 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
40 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
41 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
42 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
43 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
44 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
45 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
46 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
47 * SUCH DAMAGE.
49 #undef n_FILE
50 #define n_FILE accmacvar
52 #ifndef HAVE_AMALGAMATION
53 # include "nail.h"
54 #endif
56 #if !defined HAVE_SETENV && !defined HAVE_PUTENV
57 # error Exactly one of HAVE_SETENV and HAVE_PUTENV
58 #endif
60 /* Positional parameter maximum (macro arguments, `vexpr' "splitifs") */
61 #define a_AMV_POSPAR_MAX SI16_MAX
63 /* Special "pseudo macro" that stabs you from the back */
64 #define a_AMV_MACKY_MACK ((struct a_amv_mac*)-1)
66 /* Note: changing the hash function must be reflected in `vexpr' "hash",
67 * because that is used by the hashtable creator scripts! */
68 #define a_AMV_PRIME HSHSIZE
69 #define a_AMV_NAME2HASH(N) n_torek_hash(N)
70 #define a_AMV_HASH2PRIME(H) ((H) % a_AMV_PRIME)
72 enum a_amv_mac_flags{
73 a_AMV_MF_NONE = 0,
74 a_AMV_MF_ACCOUNT = 1u<<0, /* This macro is an `account' */
75 a_AMV_MF_TYPE_MASK = a_AMV_MF_ACCOUNT,
76 a_AMV_MF_UNDEF = 1u<<1, /* Unlink after lookup */
77 a_AMV_MF_DELETE = 1u<<7, /* Delete in progress, free once refcnt==0 */
78 a_AMV_MF__MAX = 0xFFu
81 enum a_amv_loflags{
82 a_AMV_LF_NONE = 0,
83 a_AMV_LF_SCOPE = 1u<<0, /* Current scope `localopts' on */
84 a_AMV_LF_SCOPE_FIXATE = 1u<<1, /* Ditto, but fixated */
85 a_AMV_LF_SCOPE_MASK = a_AMV_LF_SCOPE | a_AMV_LF_SCOPE_FIXATE,
86 a_AMV_LF_CALL = 1u<<2, /* `localopts' on for `call'ed scopes */
87 a_AMV_LF_CALL_FIXATE = 1u<<3, /* Ditto, but fixated */
88 a_AMV_LF_CALL_MASK = a_AMV_LF_CALL | a_AMV_LF_CALL_FIXATE,
89 a_AMV_LF_CALL_TO_SCOPE_SHIFT = 2
92 /* make-okey-map.pl ensures that _VIRT implies _RDONLY and _NODEL, and that
93 * _IMPORT implies _ENV; it doesn't verify anything...
94 * More description at nail.h:enum okeys */
95 enum a_amv_var_flags{
96 a_AMV_VF_NONE = 0,
98 /* The basic set of flags, also present in struct a_amv_var_map.avm_flags */
99 a_AMV_VF_BOOL = 1u<<0, /* ok_b_* */
100 a_AMV_VF_CHAIN = 1u<<1, /* Has -HOST and/or -USER@HOST variants */
101 a_AMV_VF_VIRT = 1u<<2, /* "Stateless" automatic variable */
102 a_AMV_VF_VIP = 1u<<3, /* Wants _var_check_vips() evaluation */
103 a_AMV_VF_RDONLY = 1u<<4, /* May not be set by user */
104 a_AMV_VF_NODEL = 1u<<5, /* May not be deleted */
105 a_AMV_VF_I3VAL = 1u<<6, /* Has an initial value */
106 a_AMV_VF_DEFVAL = 1u<<7, /* Has a default value */
107 a_AMV_VF_IMPORT = 1u<<8, /* Import ONLY from env (pre n_PSO_STARTED) */
108 a_AMV_VF_ENV = 1u<<9, /* Update environment on change */
109 a_AMV_VF_NOLOPTS = 1u<<10, /* May not be tracked by `localopts' */
110 a_AMV_VF_NOTEMPTY = 1u<<11, /* May not be assigned an empty value */
111 /* TODO _VF_NUM, _VF_POSNUM: we also need 64-bit limit numbers! */
112 a_AMV_VF_NUM = 1u<<12, /* Value must be a 32-bit number */
113 a_AMV_VF_POSNUM = 1u<<13, /* Value must be positive 32-bit number */
114 a_AMV_VF_LOWER = 1u<<14, /* Values will be stored in lowercase version */
115 a_AMV_VF_OBSOLETE = 1u<<15, /* Is obsolete? */
116 a_AMV_VF__MASK = (1u<<(15+1)) - 1,
118 /* Extended flags, not part of struct a_amv_var_map.avm_flags */
119 /* This flag indicates the instance is actually a variant of a _VF_CHAIN,
120 * it thus uses the a_amv_var_map of the base variable, but it is not the
121 * base itself and therefore care must be taken */
122 a_AMV_VF_EXT_CHAIN = 1u<<22,
123 a_AMV_VF_EXT_LOCAL = 1u<<23, /* `local' */
124 a_AMV_VF_EXT_LINKED = 1u<<24, /* `environ' link'ed */
125 a_AMV_VF_EXT_FROZEN = 1u<<25, /* Has been set by -S,.. */
126 a_AMV_VF_EXT_FROZEN_UNSET = 1u<<26, /* ..and was used to unset a variable */
127 a_AMV_VF_EXT__FROZEN_MASK = a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET,
128 a_AMV_VF_EXT__MASK = (1u<<(26+1)) - 1
131 enum a_amv_var_lookup_flags{
132 a_AMV_VLOOK_NONE = 0,
133 a_AMV_VLOOK_LOCAL = 1u<<0, /* Query `local' layer first */
134 a_AMV_VLOOK_LOCAL_ONLY = 1u<<1, /* MUST be a `local' variable */
135 /* Do not allocate new var for _I3VAL, see _var_lookup() for more */
136 a_AMV_VLOOK_I3VAL_NONEW = 1u<<2,
137 a_AMV_VLOOK_I3VAL_NONEW_REPORT = 1u<<3
140 enum a_amv_var_setclr_flags{
141 a_AMV_VSETCLR_NONE = 0,
142 a_AMV_VSETCLR_LOCAL = 1u<<0, /* Use `local' variables only */
143 a_AMV_VSETCLR_ENV = 1u<<1 /* `environ' or otherwise environ */
146 /* We support some special parameter names for one(+)-letter variable names;
147 * note these have counterparts in the code that manages shell expansion!
148 * All these special variables are solely backed by n_var_vlook(), and besides
149 * there is only a_amv_var_revlookup() which knows about them */
150 enum a_amv_var_special_category{
151 a_AMV_VSC_NONE, /* Normal variable, no special treatment */
152 a_AMV_VSC_GLOBAL, /* ${[?!]} are specially mapped, but global */
153 a_AMV_VSC_MULTIPLEX, /* ${^.*} circumflex accent multiplexer */
154 a_AMV_VSC_POSPAR, /* ${[1-9][0-9]*} positional parameters */
155 a_AMV_VSC_POSPAR_ENV /* ${[*@#]} positional parameter support variables */
158 enum a_amv_var_special_type{
159 /* _VSC_GLOBAL */
160 a_AMV_VST_QM, /* ? */
161 a_AMV_VST_EM, /* ! */
162 /* _VSC_MULTIPLEX */
163 /* This is special in that it is a multiplex indicator, the ^ is followed by
164 * a normal variable */
165 a_AMV_VST_CACC, /* ^ (circumflex accent) */
166 /* _VSC_POSPAR_ENV */
167 a_AMV_VST_STAR, /* * */
168 a_AMV_VST_AT, /* @ */
169 a_AMV_VST_NOSIGN /* # */
172 enum a_amv_var_vip_mode{
173 a_AMV_VIP_SET_PRE,
174 a_AMV_VIP_SET_POST,
175 a_AMV_VIP_CLEAR
178 struct a_amv_pospar{
179 ui16_t app_maxcount; /* == slots in .app_dat */
180 ui16_t app_count; /* Maximum a_AMV_POSPAR_MAX */
181 ui16_t app_idx; /* `shift' moves this one, decs .app_count */
182 bool_t app_not_heap; /* .app_dat stuff not dynamically allocated */
183 ui8_t app__dummy[1];
184 char const **app_dat; /* NULL terminated (for "$@" explosion support) */
186 n_CTA(a_AMV_POSPAR_MAX <= SI16_MAX, "Limit exceeds datatype capabilities");
188 struct a_amv_mac{
189 struct a_amv_mac *am_next;
190 ui32_t am_maxlen; /* of any line in .am_line_dat */
191 ui32_t am_line_cnt; /* of *.am_line_dat (but NULL terminated) */
192 struct a_amv_mac_line **am_line_dat; /* TODO use deque? */
193 struct a_amv_var *am_lopts; /* `localopts' unroll list */
194 ui32_t am_refcnt; /* 0-based for `un{account,define}' purposes */
195 ui8_t am_flags; /* enum a_amv_mac_flags */
196 char am_name[n_VFIELD_SIZE(3)]; /* of this macro */
198 n_CTA(a_AMV_MF__MAX <= UI8_MAX, "Enumeration excesses storage datatype");
200 struct a_amv_mac_line{
201 ui32_t aml_len;
202 ui32_t aml_prespc; /* Number of leading SPACEs, for display purposes */
203 char aml_dat[n_VFIELD_SIZE(0)];
206 struct a_amv_mac_call_args{
207 char const *amca_name; /* For MACKY_MACK, this is *0*! */
208 struct a_amv_mac *amca_amp; /* "const", but for am_refcnt */
209 struct a_amv_var **amca_unroller;
210 void (*amca_hook_pre)(void *);
211 void *amca_hook_arg;
212 ui8_t amca_loflags;
213 bool_t amca_ps_hook_mask;
214 bool_t amca_no_xcall; /* We want n_GO_INPUT_NO_XCALL for this */
215 ui8_t amca__pad[5];
216 struct a_amv_var *(*amca_local_vars)[a_AMV_PRIME]; /* `local's, or NULL */
217 struct a_amv_pospar amca_pospar;
220 struct a_amv_lostack{
221 struct a_amv_lostack *as_global_saved; /* Saved global XXX due to jump */
222 struct a_amv_mac_call_args *as_amcap;
223 struct a_amv_lostack *as_up; /* Outer context */
224 struct a_amv_var *as_lopts;
225 ui8_t as_loflags; /* enum a_amv_mac_loflags */
226 ui8_t avs__pad[7];
229 struct a_amv_var{
230 struct a_amv_var *av_link;
231 char *av_value;
232 #ifdef HAVE_PUTENV
233 char *av_env; /* Actively managed putenv(3) memory, or NULL */
234 #endif
235 ui32_t av_flags; /* enum a_amv_var_flags inclusive extended bits */
236 char av_name[n_VFIELD_SIZE(4)];
238 n_CTA(a_AMV_VF_EXT__MASK <= UI32_MAX, "Enumeration excesses storage datatype");
240 /* After inclusion of gen-okeys.h we assert keyoff fits in 16-bit */
241 struct a_amv_var_map{
242 ui32_t avm_hash;
243 ui16_t avm_keyoff;
244 ui16_t avm_flags; /* enum a_amv_var_flags without extended bits */
246 n_CTA(a_AMV_VF__MASK <= UI16_MAX, "Enumeration excesses storage datatype");
248 /* XXX Since there is no indicator character used for variable chains, we just
249 * XXX cannot do better than using some s....y detection.
250 * The length of avcmb_prefix is highly hardwired with make-okey-map.pl etc. */
251 struct a_amv_var_chain_map_bsrch{
252 char avcmb_prefix[4];
253 ui16_t avcmb_chain_map_off;
254 ui16_t avcmb_chain_map_eokey; /* This is an enum okey */
257 /* Use 16-bit for enum okeys, which should always be sufficient; all around
258 * here we use 32-bit for it instead, but that owed to faster access (?) */
259 struct a_amv_var_chain_map{
260 ui16_t avcm_keyoff;
261 ui16_t avcm_okey;
263 n_CTA(n_OKEYS_MAX <= UI16_MAX, "Enumeration excesses storage datatype");
265 struct a_amv_var_virt{
266 ui32_t avv_okey;
267 ui8_t avv__dummy[4];
268 struct a_amv_var const *avv_var;
271 struct a_amv_var_defval{
272 ui32_t avdv_okey;
273 ui8_t avdv__pad[4];
274 char const *avdv_value; /* Only for !BOOL (otherwise plain existence) */
277 struct a_amv_var_carrier{
278 char const *avc_name;
279 ui32_t avc_hash;
280 ui32_t avc_prime;
281 struct a_amv_var *avc_var;
282 struct a_amv_var_map const *avc_map;
283 enum okeys avc_okey;
284 bool_t avc_is_chain_variant; /* Base is a chain, this a variant thereof */
285 ui8_t avc_special_cat;
286 /* Numerical parameter name if .avc_special_cat=a_AMV_VSC_POSPAR,
287 * otherwise a enum a_amv_var_special_type */
288 ui16_t avc_special_prop;
291 /* Include constant make-okey-map.pl output, and the generated version data */
292 #include <gen-version.h>
293 #include <gen-okeys.h>
295 /* As promised above, CTAs to protect our structures */
296 n_CTA(a_AMV_VAR_NAME_KEY_MAXOFF <= UI16_MAX,
297 "Enumeration excesses storage datatype");
299 /* The currently active account */
300 static struct a_amv_mac *a_amv_acc_curr;
302 static struct a_amv_mac *a_amv_macs[a_AMV_PRIME]; /* TODO dynamically spaced */
304 /* Unroll list of currently running macro stack */
305 static struct a_amv_lostack *a_amv_lopts;
307 static struct a_amv_var *a_amv_vars[a_AMV_PRIME]; /* TODO dynamically spaced */
309 /* Global (i.e., non-local) a_AMV_VSC_POSPAR stack */
310 static struct a_amv_pospar a_amv_pospar;
312 /* TODO We really deserve localopts support for *folder-hook*s, so hack it in
313 * TODO today via a static lostack, it should be a field in mailbox, once that
314 * TODO is a real multi-instance object */
315 static struct a_amv_var *a_amv_folder_hook_lopts;
317 /* TODO Rather ditto (except for storage -> cmd_ctx), compose hooks */
318 static struct a_amv_var *a_amv_compose_lopts;
320 /* Lookup for macros/accounts: if newamp is not NULL it will be linked in the
321 * map, if _MF_UNDEF is set a possibly existing entry will be removed (first).
322 * Returns NULL if a lookup failed, or if newamp was set, the found entry in
323 * plain lookup cases or when _UNDEF was performed on a currently active entry
324 * (the entry will have been unlinked, and the _MF_DELETE will be honoured once
325 * the reference count reaches 0), and (*)-1 if an _UNDEF was performed */
326 static struct a_amv_mac *a_amv_mac_lookup(char const *name,
327 struct a_amv_mac *newamp, enum a_amv_mac_flags amf);
329 /* `call', `call_if' (and `xcall' via go.c -> c_call()) */
330 static int a_amv_mac_call(void *v, bool_t silent_nexist);
332 /* Execute a macro; amcap must reside in LOFI memory */
333 static bool_t a_amv_mac_exec(struct a_amv_mac_call_args *amcap);
335 static void a_amv_mac__finalize(void *vp);
337 /* User display helpers */
338 static bool_t a_amv_mac_show(enum a_amv_mac_flags amf);
340 /* _def() returns error for faulty definitions and already existing * names,
341 * _undef() returns error if a named thing doesn't exist */
342 static bool_t a_amv_mac_def(char const *name, enum a_amv_mac_flags amf);
343 static bool_t a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf);
345 /* */
346 static void a_amv_mac_free(struct a_amv_mac *amp);
348 /* Update replay-log */
349 static void a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
350 struct a_amv_var *oavp);
351 static void a_amv_lopts_unroll(struct a_amv_var **avpp);
353 /* Special cased value string allocation */
354 static char *a_amv_var_copy(char const *str);
355 static void a_amv_var_free(char *cp);
357 /* Check for special housekeeping. _VIP_SET_POST and _VIP_CLEAR do not fail
358 * (or propagate errors), _VIP_SET_PRE may and should case abortion */
359 static bool_t a_amv_var_check_vips(enum a_amv_var_vip_mode avvm,
360 enum okeys okey, char const **val);
362 /* _VF_NUM / _VF_POSNUM */
363 static bool_t a_amv_var_check_num(char const *val, bool_t posnum);
365 /* Try to reverse lookup a name to an enum okeys mapping, zeroing avcp.
366 * Updates .avc_name and .avc_hash; .avc_map is NULL if none found.
367 * We may try_harder to identify name: it may be an extended chain.
368 * That test only is actually performed by the latter(, then) */
369 static bool_t a_amv_var_revlookup(struct a_amv_var_carrier *avcp,
370 char const *name, bool_t try_harder);
371 static bool_t a_amv_var_revlookup_chain(struct a_amv_var_carrier *avcp,
372 char const *name);
374 /* Lookup a variable from .avc_(map|name|hash), return whether it was found.
375 * Sets .avc_prime; .avc_var is NULL if not found.
376 * Here it is where we care for _I3VAL and _DEFVAL.
377 * An _I3VAL will be "consumed" as necessary anyway, but it won't be used to
378 * create a new variable if _VLOOK_I3VAL_NONEW is set; if
379 * _VLOOK_I3VAL_NONEW_REPORT is set then we set .avc_var to -1 and return true
380 * if that was the case, otherwise we'll return FAL0, then! */
381 static bool_t a_amv_var_lookup(struct a_amv_var_carrier *avcp,
382 enum a_amv_var_lookup_flags avlf);
384 /* Lookup functions for special category variables, _pospar drives all
385 * positional parameter etc. special categories */
386 static char const *a_amv_var_vsc_global(struct a_amv_var_carrier *avcp);
387 static char const *a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp);
388 static char const *a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp);
390 /* Set var from .avc_(map|name|hash), return success */
391 static bool_t a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
392 enum a_amv_var_setclr_flags avscf);
394 static bool_t a_amv_var__putenv(struct a_amv_var_carrier *avcp,
395 struct a_amv_var *avp);
397 /* Clear var from .avc_(map|name|hash); sets .avc_var=NULL, return success */
398 static bool_t a_amv_var_clear(struct a_amv_var_carrier *avcp,
399 enum a_amv_var_setclr_flags avscf);
401 static bool_t a_amv_var__clearenv(char const *name, struct a_amv_var *avp);
403 /* List all variables */
404 static void a_amv_var_show_all(void);
406 static int a_amv_var__show_cmp(void const *s1, void const *s2);
408 /* Actually do print one, return number of lines written */
409 static size_t a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp);
411 /* Shared c_set() and c_environ():set impl, return success */
412 static bool_t a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf);
414 static struct a_amv_mac *
415 a_amv_mac_lookup(char const *name, struct a_amv_mac *newamp,
416 enum a_amv_mac_flags amf){
417 struct a_amv_mac *amp, **ampp;
418 ui32_t h;
419 enum a_amv_mac_flags save_amf;
420 NYD2_ENTER;
422 save_amf = amf;
423 amf &= a_AMV_MF_TYPE_MASK;
424 h = a_AMV_NAME2HASH(name);
425 h = a_AMV_HASH2PRIME(h);
426 ampp = &a_amv_macs[h];
428 for(amp = *ampp; amp != NULL; ampp = &(*ampp)->am_next, amp = amp->am_next){
429 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf &&
430 !strcmp(amp->am_name, name)){
431 if(n_LIKELY((save_amf & a_AMV_MF_UNDEF) == 0))
432 goto jleave;
434 *ampp = amp->am_next;
436 if(amp->am_refcnt > 0){
437 amp->am_flags |= a_AMV_MF_DELETE;
438 if(n_poption & n_PO_D_V)
439 n_err(_("Delayed deletion of currently active %s: %s\n"),
440 (amp->am_flags & a_AMV_MF_ACCOUNT ? "account" : "define"),
441 name);
442 }else{
443 a_amv_mac_free(amp);
444 amp = (struct a_amv_mac*)-1;
446 break;
450 if(newamp != NULL){
451 ampp = &a_amv_macs[h];
452 newamp->am_next = *ampp;
453 *ampp = newamp;
454 amp = NULL;
456 jleave:
457 NYD2_LEAVE;
458 return amp;
461 static int
462 a_amv_mac_call(void *v, bool_t silent_nexist){
463 struct a_amv_mac *amp;
464 int rv;
465 char const *name;
466 struct n_cmd_arg_ctx *cacp;
467 NYD_ENTER;
469 cacp = v;
471 if(cacp->cac_no == 0){
472 n_err(_("Synopsis: call(_if)?: name [:<arguments>:]\n"));
473 n_pstate_err_no = n_ERR_INVAL;
474 rv = 1;
475 goto jleave;
478 name = cacp->cac_arg->ca_arg.ca_str.s;
480 if(n_UNLIKELY(cacp->cac_no > a_AMV_POSPAR_MAX)){
481 n_err(_("Too many arguments to macro `call': %s\n"), name);
482 n_pstate_err_no = n_ERR_OVERFLOW;
483 rv = 1;
484 }else if(n_UNLIKELY((amp = a_amv_mac_lookup(name, NULL, a_AMV_MF_NONE)
485 ) == NULL)){
486 if(!silent_nexist)
487 n_err(_("Undefined macro called: %s\n"), n_shexp_quote_cp(name, FAL0));
488 n_pstate_err_no = n_ERR_NOENT;
489 rv = 1;
490 }else{
491 char const **argv;
492 struct a_amv_mac_call_args *amcap;
493 size_t argc;
495 argc = cacp->cac_no + 1;
496 amcap = n_lofi_alloc(sizeof *amcap + (argc * sizeof *argv));
497 argv = (void*)&amcap[1];
499 for(argc = 0; (cacp->cac_arg = cacp->cac_arg->ca_next) != NULL; ++argc)
500 argv[argc] = cacp->cac_arg->ca_arg.ca_str.s;
501 argv[argc] = NULL;
503 memset(amcap, 0, sizeof *amcap);
504 amcap->amca_name = name;
505 amcap->amca_amp = amp;
506 if(a_amv_lopts != NULL)
507 amcap->amca_loflags = (a_amv_lopts->as_loflags & a_AMV_LF_CALL_MASK
508 ) >> a_AMV_LF_CALL_TO_SCOPE_SHIFT;
509 if(argc > 0){
510 amcap->amca_pospar.app_count = (ui16_t)argc;
511 amcap->amca_pospar.app_not_heap = TRU1;
512 amcap->amca_pospar.app_dat = argv;
515 (void)a_amv_mac_exec(amcap);
516 rv = n_pstate_ex_no;
518 jleave:
519 NYD_LEAVE;
520 return rv;
523 static bool_t
524 a_amv_mac_exec(struct a_amv_mac_call_args *amcap){
525 struct a_amv_lostack *losp;
526 struct a_amv_mac_line **amlp;
527 char **args_base, **args;
528 struct a_amv_mac *amp;
529 bool_t rv;
530 NYD2_ENTER;
532 amp = amcap->amca_amp;
533 assert(amp != NULL && amp != a_AMV_MACKY_MACK);
534 ++amp->am_refcnt;
535 /* XXX Unfortunately we yet need to dup the macro lines! :( */
536 args_base = args = n_alloc(sizeof(*args) * (amp->am_line_cnt +1));
537 for(amlp = amp->am_line_dat; *amlp != NULL; ++amlp)
538 *(args++) = sbufdup((*amlp)->aml_dat, (*amlp)->aml_len);
539 *args = NULL;
541 losp = n_lofi_alloc(sizeof *losp);
542 losp->as_global_saved = a_amv_lopts;
543 if((losp->as_amcap = amcap)->amca_unroller == NULL){
544 losp->as_up = losp->as_global_saved;
545 losp->as_lopts = NULL;
546 }else{
547 losp->as_up = NULL;
548 losp->as_lopts = *amcap->amca_unroller;
550 losp->as_loflags = amcap->amca_loflags;
552 a_amv_lopts = losp;
553 if(amcap->amca_hook_pre != NULL)
554 n_PS_ROOT_BLOCK((*amcap->amca_hook_pre)(amcap->amca_hook_arg));
555 rv = n_go_macro((n_GO_INPUT_NONE |
556 (amcap->amca_no_xcall ? n_GO_INPUT_NO_XCALL : 0)),
557 amp->am_name, args_base, &a_amv_mac__finalize, losp);
558 NYD2_LEAVE;
559 return rv;
562 static void
563 a_amv_mac__finalize(void *vp){
564 struct a_amv_mac *amp;
565 struct a_amv_mac_call_args *amcap;
566 struct a_amv_lostack *losp;
567 NYD2_ENTER;
569 losp = vp;
570 a_amv_lopts = losp->as_global_saved;
572 amcap = losp->as_amcap;
574 /* Delete positional parameter stack */
575 if(!amcap->amca_pospar.app_not_heap && amcap->amca_pospar.app_maxcount > 0){
576 ui16_t i;
578 for(i = amcap->amca_pospar.app_maxcount; i-- != 0;)
579 n_free(n_UNCONST(amcap->amca_pospar.app_dat[i]));
580 n_free(amcap->amca_pospar.app_dat);
583 /* `local' variable hashmap. These have no environment map, never */
584 if(amcap->amca_local_vars != NULL){
585 struct a_amv_var **avpp_base, **avpp, *avp;
587 for(avpp_base = *amcap->amca_local_vars, avpp = &avpp_base[a_AMV_PRIME];
588 avpp-- != avpp_base;)
589 while((avp = *avpp)){
590 assert((avp->av_flags & (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL)) ==
591 (a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL));
592 assert(!(avp->av_flags &
593 ((a_AMV_VF__MASK | a_AMV_VF_EXT__MASK) &
594 ~(a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL))));
595 *avpp = avp->av_link;
596 a_amv_var_free(avp->av_value);
597 n_free(avp);
599 n_free(avpp_base);
602 /* Unroll `localopts', if applicable */
603 if(amcap->amca_unroller == NULL){
604 if(losp->as_lopts != NULL)
605 a_amv_lopts_unroll(&losp->as_lopts);
606 }else
607 *amcap->amca_unroller = losp->as_lopts;
609 if(amcap->amca_ps_hook_mask)
610 n_pstate &= ~n_PS_HOOK_MASK;
612 if((amp = amcap->amca_amp) != a_AMV_MACKY_MACK && amp != NULL &&
613 --amp->am_refcnt == 0 && (amp->am_flags & a_AMV_MF_DELETE))
614 a_amv_mac_free(amp);
616 n_lofi_free(losp);
617 n_lofi_free(amcap);
618 NYD2_LEAVE;
621 static bool_t
622 a_amv_mac_show(enum a_amv_mac_flags amf){
623 size_t lc, mc, ti, i;
624 char const *typestr;
625 FILE *fp;
626 bool_t rv;
627 NYD2_ENTER;
629 rv = FAL0;
631 if((fp = Ftmp(NULL, "deflist", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
632 NULL){
633 n_perr(_("`define' or `account' list: cannot create temporary file"), 0);
634 goto jleave;
637 amf &= a_AMV_MF_TYPE_MASK;
638 typestr = (amf & a_AMV_MF_ACCOUNT) ? "account" : "define";
640 for(lc = mc = ti = 0; ti < a_AMV_PRIME; ++ti){
641 struct a_amv_mac *amp;
643 for(amp = a_amv_macs[ti]; amp != NULL; amp = amp->am_next){
644 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
645 struct a_amv_mac_line **amlpp;
647 if(++mc > 1){
648 putc('\n', fp);
649 ++lc;
651 ++lc;
652 fprintf(fp, "%s %s {\n", typestr, amp->am_name);
653 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++lc, ++amlpp){
654 for(i = (*amlpp)->aml_prespc; i > 0; --i)
655 putc(' ', fp);
656 fputs((*amlpp)->aml_dat, fp);
657 putc('\n', fp);
659 fputs("}\n", fp);
660 ++lc;
664 if(mc > 0)
665 page_or_print(fp, lc);
667 rv = (ferror(fp) == 0);
668 Fclose(fp);
669 jleave:
670 NYD2_LEAVE;
671 return rv;
674 static bool_t
675 a_amv_mac_def(char const *name, enum a_amv_mac_flags amf){
676 struct str line;
677 ui32_t line_cnt, maxlen;
678 struct linelist{
679 struct linelist *ll_next;
680 struct a_amv_mac_line *ll_amlp;
681 } *llp, *ll_head, *ll_tail;
682 union {size_t s; int i; ui32_t ui; size_t l;} n;
683 struct a_amv_mac *amp;
684 bool_t rv;
685 NYD2_ENTER;
687 memset(&line, 0, sizeof line);
688 rv = FAL0;
689 amp = NULL;
691 /* TODO We should have our input state machine which emits Line events,
692 * TODO and hook different consumers dependent on our content, as stated
693 * TODO in i think lex_input: like this local macros etc. would become
694 * TODO possible from the input side */
695 /* Read in the lines which form the macro content */
696 for(ll_tail = ll_head = NULL, line_cnt = maxlen = 0;;){
697 ui32_t leaspc;
698 char *cp;
700 n.i = n_go_input(n_GO_INPUT_CTX_DEFAULT | n_GO_INPUT_NL_ESC, n_empty,
701 &line.s, &line.l, NULL, NULL);
702 if(n.ui == 0)
703 continue;
704 if(n.i < 0){
705 n_err(_("Unterminated %s definition: %s\n"),
706 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
707 goto jerr;
710 /* Trim WS, remember amount of leading spaces for display purposes */
711 for(cp = line.s, leaspc = 0; n.ui > 0; ++cp, --n.ui)
712 if(*cp == '\t')
713 leaspc = (leaspc + 8u) & ~7u;
714 else if(*cp == ' ')
715 ++leaspc;
716 else
717 break;
718 for(; n.ui > 0 && spacechar(cp[n.ui - 1]); --n.ui)
720 if(n.ui == 0)
721 continue;
723 maxlen = n_MAX(maxlen, n.ui);
724 cp[n.ui++] = '\0';
726 /* Is is the closing brace? */
727 if(*cp == '}')
728 break;
730 if(n_LIKELY(++line_cnt < UI32_MAX)){
731 struct a_amv_mac_line *amlp;
733 llp = n_autorec_alloc(sizeof *llp);
734 if(ll_head == NULL)
735 ll_head = llp;
736 else
737 ll_tail->ll_next = llp;
738 ll_tail = llp;
739 llp->ll_next = NULL;
740 llp->ll_amlp = amlp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac_line,
741 aml_dat) + n.ui);
742 amlp->aml_len = n.ui -1;
743 amlp->aml_prespc = leaspc;
744 memcpy(amlp->aml_dat, cp, n.ui);
745 }else{
746 n_err(_("Too much content in %s definition: %s\n"),
747 (amf & a_AMV_MF_ACCOUNT ? "account" : "macro"), name);
748 goto jerr;
752 /* Create the new macro */
753 n.s = strlen(name) +1;
754 amp = n_alloc(n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name) + n.s);
755 memset(amp, 0, n_VSTRUCT_SIZEOF(struct a_amv_mac, am_name));
756 amp->am_maxlen = maxlen;
757 amp->am_line_cnt = line_cnt;
758 amp->am_flags = amf;
759 memcpy(amp->am_name, name, n.s);
760 /* C99 */{
761 struct a_amv_mac_line **amlpp;
763 amp->am_line_dat = amlpp = n_alloc(sizeof(*amlpp) * ++line_cnt);
764 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
765 *amlpp++ = llp->ll_amlp;
766 *amlpp = NULL;
769 /* Create entry, replace a yet existing one as necessary */
770 a_amv_mac_lookup(name, amp, amf | a_AMV_MF_UNDEF);
771 rv = TRU1;
772 jleave:
773 if(line.s != NULL)
774 n_free(line.s);
775 NYD2_LEAVE;
776 return rv;
778 jerr:
779 for(llp = ll_head; llp != NULL; llp = llp->ll_next)
780 n_free(llp->ll_amlp);
782 * if(amp != NULL){
783 * n_free(amp->am_line_dat);
784 * n_free(amp);
785 *}*/
786 goto jleave;
789 static bool_t
790 a_amv_mac_undef(char const *name, enum a_amv_mac_flags amf){
791 struct a_amv_mac *amp;
792 bool_t rv;
793 NYD2_ENTER;
795 rv = TRU1;
797 if(n_LIKELY(name[0] != '*' || name[1] != '\0')){
798 if((amp = a_amv_mac_lookup(name, NULL, amf | a_AMV_MF_UNDEF)) == NULL){
799 n_err(_("%s not defined: %s\n"),
800 (amf & a_AMV_MF_ACCOUNT ? "Account" : "Macro"), name);
801 rv = FAL0;
803 }else{
804 struct a_amv_mac **ampp, *lamp;
806 for(ampp = a_amv_macs; PTRCMP(ampp, <, &a_amv_macs[n_NELEM(a_amv_macs)]);
807 ++ampp)
808 for(lamp = NULL, amp = *ampp; amp != NULL;){
809 if((amp->am_flags & a_AMV_MF_TYPE_MASK) == amf){
810 /* xxx Expensive but rare: be simple */
811 a_amv_mac_lookup(amp->am_name, NULL, amf | a_AMV_MF_UNDEF);
812 amp = (lamp == NULL) ? *ampp : lamp->am_next;
813 }else{
814 lamp = amp;
815 amp = amp->am_next;
819 NYD2_LEAVE;
820 return rv;
823 static void
824 a_amv_mac_free(struct a_amv_mac *amp){
825 struct a_amv_mac_line **amlpp;
826 NYD2_ENTER;
828 for(amlpp = amp->am_line_dat; *amlpp != NULL; ++amlpp)
829 n_free(*amlpp);
830 n_free(amp->am_line_dat);
831 n_free(amp);
832 NYD2_LEAVE;
835 static void
836 a_amv_lopts_add(struct a_amv_lostack *alp, char const *name,
837 struct a_amv_var *oavp){
838 struct a_amv_var *avp;
839 size_t nl, vl;
840 NYD2_ENTER;
842 /* Propagate unrolling up the stack, as necessary */
843 assert(alp != NULL);
844 for(;;){
845 if(alp->as_loflags & a_AMV_LF_SCOPE_MASK)
846 break;
847 if((alp = alp->as_up) == NULL)
848 goto jleave;
851 /* Check whether this variable is handled yet XXX Boost: ID check etc.!! */
852 for(avp = alp->as_lopts; avp != NULL; avp = avp->av_link)
853 if(!strcmp(avp->av_name, name))
854 goto jleave;
856 nl = strlen(name) +1;
857 vl = (oavp != NULL) ? strlen(oavp->av_value) +1 : 0;
858 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + nl + vl);
859 avp->av_link = alp->as_lopts;
860 alp->as_lopts = avp;
861 if(vl != 0){
862 avp->av_value = &avp->av_name[nl];
863 avp->av_flags = oavp->av_flags;
864 memcpy(avp->av_value, oavp->av_value, vl);
866 memcpy(avp->av_name, name, nl);
867 jleave:
868 NYD2_LEAVE;
871 static void
872 a_amv_lopts_unroll(struct a_amv_var **avpp){
873 struct a_amv_lostack *save_alp;
874 struct a_amv_var *x, *avp;
875 NYD2_ENTER;
877 avp = *avpp;
878 *avpp = NULL;
880 save_alp = a_amv_lopts;
881 a_amv_lopts = NULL;
882 while(avp != NULL){
883 x = avp;
884 avp = avp->av_link;
885 n_PS_ROOT_BLOCK(n_var_vset(x->av_name, (uintptr_t)x->av_value));
886 n_free(x);
888 a_amv_lopts = save_alp;
889 NYD2_LEAVE;
892 static char *
893 a_amv_var_copy(char const *str){
894 char *news;
895 size_t len;
896 NYD2_ENTER;
898 if(*str == '\0')
899 news = n_UNCONST(n_empty);
900 else if(str[1] == '\0'){
901 if(str[0] == '1')
902 news = n_UNCONST(n_1);
903 else if(str[0] == '0')
904 news = n_UNCONST(n_0);
905 else
906 goto jheap;
907 }else if(str[2] == '\0' && str[0] == '-' && str[1] == '1')
908 news = n_UNCONST(n_m1);
909 else{
910 jheap:
911 len = strlen(str) +1;
912 news = n_alloc(len);
913 memcpy(news, str, len);
915 NYD2_LEAVE;
916 return news;
919 static void
920 a_amv_var_free(char *cp){
921 NYD2_ENTER;
922 if(cp[0] != '\0' && cp != n_0 && cp != n_1 && cp != n_m1)
923 n_free(cp);
924 NYD2_LEAVE;
927 static bool_t
928 a_amv_var_check_vips(enum a_amv_var_vip_mode avvm, enum okeys okey,
929 char const **val){
930 char const *emsg;
931 bool_t ok;
932 NYD2_ENTER;
934 ok = TRU1;
936 if(avvm == a_AMV_VIP_SET_PRE){
937 switch(okey){
938 default:
939 break;
940 case ok_v_charset_7bit:
941 case ok_v_charset_8bit:
942 case ok_v_charset_unknown_8bit:
943 case ok_v_ttycharset:
944 if((*val = n_iconv_normalize_name(*val)) == NULL)
945 ok = FAL0;
946 break;
947 case ok_v_customhdr:{
948 char const *vp;
949 char *buf;
950 struct n_header_field *hflp, **hflpp, *hfp;
951 NYD_ENTER;
953 buf = savestr(*val);
954 hflp = NULL;
955 hflpp = &hflp;
957 while((vp = n_strsep_esc(&buf, ',', TRU1)) != NULL){
958 if(!n_header_add_custom(hflpp, vp, TRU1)){
959 emsg = N_("Invalid *customhdr* entry: %s\n");
960 ok = FAL0;
961 break;
963 hflpp = &(*hflpp)->hf_next;
966 hflpp = ok ? &n_customhdr_list : &hflp;
967 while((hfp = *hflpp) != NULL){
968 *hflpp = hfp->hf_next;
969 n_free(hfp);
971 if(!ok)
972 goto jerr;
973 n_customhdr_list = hflp;
974 }break;
975 case ok_v_from:
976 case ok_v_sender:{
977 struct name *np;
979 if((np = lextract(*val, GEXTRA | GFULL)) == NULL){
980 jefrom:
981 emsg = N_("*from* / *sender*: invalid address(es): %s\n");
982 goto jerr;
983 }else if(okey == ok_v_sender && np->n_flink != NULL){
984 emsg = N_("*sender*: may not contain multiple addresses: %s\n");
985 goto jerr;
986 }else for(; np != NULL; np = np->n_flink)
987 if(is_addr_invalid(np, EACM_STRICT | EACM_NOLOG | EACM_NONAME))
988 goto jefrom;
989 }break;
990 case ok_v_HOME:
991 /* Note this gets called from main.c during initialization, and they
992 * simply set this to pw_dir as a fallback: don't verify _that_ call.
993 * See main.c! */
994 if(!(n_pstate & n_PS_ROOT) && !n_is_dir(*val, TRUM1)){
995 emsg = N_("$HOME is not a directory or not accessible: %s\n");
996 goto jerr;
998 break;
999 case ok_v_hostname:
1000 case ok_v_smtp_hostname:
1001 #ifdef HAVE_IDNA
1002 if(**val != '\0'){
1003 struct n_string cnv;
1005 n_string_creat_auto(&cnv);
1006 if(!n_idna_to_ascii(&cnv, *val, UIZ_MAX)){
1007 /*n_string_gut(&res);*/
1008 emsg = N_("*hostname*/*smtp_hostname*: "
1009 "IDNA encoding failed: %s\n");
1010 goto jerr;
1012 *val = n_string_cp(&cnv);
1013 /*n_string_drop_ownership(&cnv);*/
1015 #endif
1016 break;
1017 case ok_v_quote_chars:{
1018 char c;
1019 char const *cp;
1021 for(cp = *val; (c = *cp++) != '\0';)
1022 if(!asciichar(c) || blankspacechar(c)){
1023 ok = FAL0;
1024 break;
1026 }break;
1027 case ok_v_sendcharsets:{
1028 struct n_string s, *sp = &s;
1029 char *csv, *cp;
1031 sp = n_string_creat_auto(sp);
1032 csv = savestr(*val);
1034 while((cp = n_strsep(&csv, ',', TRU1)) != NULL){
1035 if((cp = n_iconv_normalize_name(cp)) == NULL){
1036 ok = FAL0;
1037 break;
1039 if(sp->s_len > 0)
1040 sp = n_string_push_c(sp, ',');
1041 sp = n_string_push_cp(sp, cp);
1044 *val = n_string_cp(sp);
1045 /* n_string_drop_ownership(sp); */
1046 }break;
1047 case ok_v_TMPDIR:
1048 if(!n_is_dir(*val, TRU1)){
1049 emsg = N_("$TMPDIR is not a directory or not accessible: %s\n");
1050 goto jerr;
1052 break;
1053 case ok_v_umask:
1054 if(**val != '\0'){
1055 ui64_t uib;
1057 n_idec_ui64_cp(&uib, *val, 0, NULL);
1058 if(uib & ~0777u){ /* (is valid _VF_POSNUM) */
1059 emsg = N_("Invalid *umask* setting: %s\n");
1060 goto jerr;
1063 break;
1065 }else if(avvm == a_AMV_VIP_SET_POST){
1066 switch(okey){
1067 default:
1068 break;
1069 case ok_b_ask:
1070 ok_bset(asksub);
1071 break;
1072 case ok_b_debug:
1073 n_poption |= n_PO_DEBUG;
1074 break;
1075 case ok_v_HOME:
1076 /* Invalidate any resolved folder then, too
1077 * FALLTHRU */
1078 case ok_v_folder:
1079 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1080 break;
1081 case ok_v_ifs:{
1082 char *x_b, *x, c;
1083 char const *cp;
1085 cp = *val;
1086 x_b = x = n_autorec_alloc(strlen(cp) +1);
1087 while((c = *cp++) != '\0')
1088 if(spacechar(c))
1089 *x++ = c;
1090 *x = '\0';
1091 n_PS_ROOT_BLOCK(ok_vset(ifs_ws, x_b));
1092 }break;
1093 #ifdef HAVE_SETLOCALE
1094 case ok_v_LANG:
1095 case ok_v_LC_ALL:
1096 case ok_v_LC_CTYPE:
1097 n_locale_init();
1098 break;
1099 #endif
1100 case ok_b_memdebug:
1101 n_poption |= n_PO_MEMDEBUG;
1102 break;
1103 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1104 if(!(n_pstate & n_PS_ROOT))
1105 n_PS_ROOT_BLOCK(ok_bset(posix));
1106 break;
1107 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1108 if(!(n_pstate & n_PS_ROOT))
1109 n_PS_ROOT_BLOCK(ok_bset(POSIXLY_CORRECT));
1110 break;
1111 case ok_b_skipemptybody:
1112 n_poption |= n_PO_E_FLAG;
1113 break;
1114 case ok_b_typescript_mode:
1115 ok_bset(colour_disable);
1116 ok_bset(line_editor_disable);
1117 if(!(n_psonce & n_PSO_STARTED))
1118 ok_bset(termcap_disable);
1119 break;
1120 case ok_v_umask:
1121 if(**val != '\0'){
1122 ui64_t uib;
1124 n_idec_ui64_cp(&uib, *val, 0, NULL);
1125 umask((mode_t)uib);
1127 break;
1128 case ok_b_verbose:
1129 n_poption |= (n_poption & n_PO_VERB) ? n_PO_VERBVERB : n_PO_VERB;
1130 break;
1132 }else{
1133 switch(okey){
1134 default:
1135 break;
1136 case ok_b_ask:
1137 ok_bclear(asksub);
1138 break;
1139 case ok_b_debug:
1140 n_poption &= ~n_PO_DEBUG;
1141 break;
1142 case ok_v_customhdr:{
1143 struct n_header_field *hfp;
1145 while((hfp = n_customhdr_list) != NULL){
1146 n_customhdr_list = hfp->hf_next;
1147 n_free(hfp);
1149 }break;
1150 case ok_v_HOME:
1151 /* Invalidate any resolved folder then, too
1152 * FALLTHRU */
1153 case ok_v_folder:
1154 n_PS_ROOT_BLOCK(ok_vclear(folder_resolved));
1155 break;
1156 case ok_b_memdebug:
1157 n_poption &= ~n_PO_MEMDEBUG;
1158 break;
1159 case ok_b_POSIXLY_CORRECT: /* <-> *posix* */
1160 if(!(n_pstate & n_PS_ROOT))
1161 n_PS_ROOT_BLOCK(ok_bclear(posix));
1162 break;
1163 case ok_b_posix: /* <-> $POSIXLY_CORRECT */
1164 if(!(n_pstate & n_PS_ROOT))
1165 n_PS_ROOT_BLOCK(ok_bclear(POSIXLY_CORRECT));
1166 break;
1167 case ok_b_skipemptybody:
1168 n_poption &= ~n_PO_E_FLAG;
1169 break;
1170 case ok_b_verbose:
1171 n_poption &= ~(n_PO_VERB | n_PO_VERBVERB);
1172 break;
1176 jleave:
1177 NYD2_LEAVE;
1178 return ok;
1179 jerr:
1180 n_err(V_(emsg), n_shexp_quote_cp(*val, FAL0));
1181 ok = FAL0;
1182 goto jleave;
1185 static bool_t
1186 a_amv_var_check_num(char const *val, bool_t posnum){
1187 /* TODO The internal/environment variables which are num= or posnum= should
1188 * TODO gain special lookup functions, or the return should be void* and
1189 * TODO castable to integer; i.e. no more strtoX() should be needed.
1190 * TODO I.e., the result of this function should instead be stored */
1191 bool_t rv;
1192 NYD2_ENTER;
1194 rv = TRU1;
1196 if(*val != '\0'){ /* Would be _VF_NOTEMPTY if not allowed */
1197 ui64_t uib;
1198 enum n_idec_state ids;
1200 ids = n_idec_cp(&uib, val, 0,
1201 (n_IDEC_MODE_LIMIT_32BIT |
1202 (posnum ? n_IDEC_MODE_SIGNED_TYPE : n_IDEC_MODE_NONE)), NULL);
1203 if((ids & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
1204 ) != n_IDEC_STATE_CONSUMED)
1205 rv = FAL0;
1206 /* TODO Unless we store integers we need to look and forbid, because
1207 * TODO callee may not be able to swallow, e.g., "-1" */
1208 if(posnum && (ids & n_IDEC_STATE_SEEN_MINUS))
1209 rv = FAL0;
1211 NYD2_LEAVE;
1212 return rv;
1215 static bool_t
1216 a_amv_var_revlookup(struct a_amv_var_carrier *avcp, char const *name,
1217 bool_t try_harder){
1218 ui32_t hash, i, j;
1219 struct a_amv_var_map const *avmp;
1220 char c;
1221 NYD2_ENTER;
1223 memset(avcp, 0, sizeof *avcp); /* XXX overkill, just set chain */
1225 /* It may be a special a.k.a. macro-local or one-letter parameter */
1226 c = name[0];
1227 if(n_UNLIKELY(digitchar(c))){
1228 /* (Inline dec. atoi, ugh) */
1229 for(j = (ui8_t)c - '0', i = 1;; ++i){
1230 c = name[i];
1231 if(c == '\0')
1232 break;
1233 if(!digitchar(c))
1234 goto jno_special_param;
1235 j = j * 10 + (ui8_t)c - '0';
1237 if(j <= a_AMV_POSPAR_MAX){
1238 avcp->avc_special_cat = a_AMV_VSC_POSPAR;
1239 goto jspecial_param;
1241 }else if(n_UNLIKELY(name[1] == '\0')){
1242 switch(c){
1243 case '?':
1244 case '!':
1245 avcp->avc_special_cat = a_AMV_VSC_GLOBAL;
1246 j = (c == '?') ? a_AMV_VST_QM : a_AMV_VST_EM;
1247 goto jspecial_param;
1248 case '^':
1249 goto jmultiplex;
1250 case '*':
1251 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1252 j = a_AMV_VST_STAR;
1253 goto jspecial_param;
1254 case '@':
1255 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1256 j = a_AMV_VST_AT;
1257 goto jspecial_param;
1258 case '#':
1259 avcp->avc_special_cat = a_AMV_VSC_POSPAR_ENV;
1260 j = a_AMV_VST_NOSIGN;
1261 goto jspecial_param;
1262 default:
1263 break;
1265 }else if(c == '^'){
1266 jmultiplex:
1267 avcp->avc_special_cat = a_AMV_VSC_MULTIPLEX;
1268 j = a_AMV_VST_CACC;
1269 goto jspecial_param;
1272 /* This is nothing special, but a plain variable */
1273 jno_special_param:
1274 assert(a_AMV_VSC_NONE == 0);/*avcp->avc_special_cat = a_AMV_VSC_NONE;*/
1275 avcp->avc_name = name;
1276 avcp->avc_hash = hash = a_AMV_NAME2HASH(name);
1278 /* Is it a known okey? Walk over the hashtable */
1279 for(i = hash % a_AMV_VAR_REV_PRIME, j = 0; j <= a_AMV_VAR_REV_LONGEST; ++j){
1280 ui32_t x;
1282 if((x = a_amv_var_revmap[i]) == a_AMV_VAR_REV_ILL)
1283 break;
1285 avmp = &a_amv_var_map[x];
1286 if(avmp->avm_hash == hash &&
1287 !strcmp(&a_amv_var_names[avmp->avm_keyoff], name)){
1288 avcp->avc_map = avmp;
1289 avcp->avc_okey = (enum okeys)x;
1290 goto jleave;
1293 if(++i == a_AMV_VAR_REV_PRIME){
1294 #ifdef a_AMV_VAR_REV_WRAPAROUND
1295 i = 0;
1296 #else
1297 break;
1298 #endif
1302 /* Not a known key, but it may be a chain extension of one.
1303 * We possibly wanna know for a variety of reasons */
1304 if(try_harder && a_amv_var_revlookup_chain(avcp, name))
1305 goto jleave;
1307 assert(avcp->avc_map == NULL);/*avcp->avc_map = NULL;*/
1308 avcp = NULL;
1309 jleave:
1310 assert(avcp == NULL || avcp->avc_map != NULL ||
1311 avcp->avc_special_cat == a_AMV_VSC_NONE);
1312 NYD2_LEAVE;
1313 return (avcp != NULL);
1315 /* All these are mapped to *--special-param* */
1316 jspecial_param:
1317 avcp->avc_name = name;
1318 avcp->avc_special_prop = (ui16_t)j;
1319 avmp = &a_amv_var_map[a_AMV_VAR__SPECIAL_PARAM_MAP_IDX];
1320 avcp->avc_hash = avmp->avm_hash;
1321 avcp->avc_map = avmp;
1322 avcp->avc_okey = ok_v___special_param;
1323 goto jleave;
1326 static bool_t
1327 a_amv_var_revlookup_chain(struct a_amv_var_carrier *avcp, char const *name){
1328 uiz_t i;
1329 struct a_amv_var_chain_map_bsrch const *avcmbp, *avcmbp_x;
1330 NYD_ENTER;
1332 if(strlen(name) <
1333 n_SIZEOF_FIELD(struct a_amv_var_chain_map_bsrch, avcmb_prefix)){
1334 avcp = NULL;
1335 goto jleave;
1338 avcmbp = &a_amv_var_chain_map_bsrch[0];
1339 i = a_AMV_VAR_CHAIN_MAP_BSRCH_CNT - 0;
1341 int cres;
1342 avcmbp_x = &avcmbp[i >> 1];
1343 cres = memcmp(name, avcmbp_x->avcmb_prefix,
1344 n_SIZEOF_FIELD(struct a_amv_var_chain_map_bsrch, avcmb_prefix));
1345 if(cres != 0){
1346 /* Go right instead? */
1347 if(cres > 0){
1348 avcmbp = ++avcmbp_x;
1349 --i;
1351 }else{
1352 /* Once the binary search found the right prefix we have to use
1353 * a linear walk from then on, because there is no "trigger"
1354 * character: anything could be something free-form or
1355 * a chain-extension, we just do not know. Unfortunately.
1356 * Luckily cramping the walk to a small window is possible */
1357 struct a_amv_var_chain_map const *avcmp, *avcmp_hit;
1359 avcmp = &a_amv_var_chain_map[avcmbp_x->avcmb_chain_map_off];
1360 avcmp_hit = NULL;
1362 char c;
1363 char const *cp, *ncp;
1365 cp = &a_amv_var_names[avcmp->avcm_keyoff +
1366 n_SIZEOF_FIELD(struct a_amv_var_chain_map_bsrch,
1367 avcmb_prefix)];
1368 ncp = &name[n_SIZEOF_FIELD(struct a_amv_var_chain_map_bsrch,
1369 avcmb_prefix)];
1370 for(;; ++ncp, ++cp)
1371 if(*ncp != (c = *cp) || c == '\0')
1372 break;
1373 /* Is it a chain extension of this key? */
1374 if(c == '\0' && *ncp == '-')
1375 avcmp_hit = avcmp;
1376 else if(avcmp_hit != NULL)
1377 break;
1378 }while((avcmp++)->avcm_okey < avcmbp_x->avcmb_chain_map_eokey);
1380 if(avcmp_hit != NULL){
1381 avcp->avc_map = &a_amv_var_map[avcp->avc_okey =
1382 (enum okeys)avcmp_hit->avcm_okey];
1383 avcp->avc_is_chain_variant = TRU1;
1384 goto jleave;
1386 break;
1388 }while((i >>= 1) > 0);
1390 avcp = NULL;
1391 jleave:
1392 NYD_LEAVE;
1393 return (avcp != NULL);
1396 static bool_t
1397 a_amv_var_lookup(struct a_amv_var_carrier *avcp,
1398 enum a_amv_var_lookup_flags avlf){
1399 size_t i;
1400 char const *cp;
1401 ui32_t f;
1402 struct a_amv_var_map const *avmp;
1403 struct a_amv_var *avp;
1404 NYD2_ENTER;
1406 assert(!(avlf & a_AMV_VLOOK_LOCAL_ONLY) || (avlf & a_AMV_VLOOK_LOCAL));
1407 assert(!(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT) ||
1408 (avlf & a_AMV_VLOOK_I3VAL_NONEW));
1410 /* C99 */{
1411 struct a_amv_var **avpp, *lavp;
1413 avcp->avc_prime = a_AMV_HASH2PRIME(avcp->avc_hash);
1415 /* Optionally macro-`local' variables first */
1416 if(avlf & a_AMV_VLOOK_LOCAL){
1417 if(a_amv_lopts != NULL &&
1418 (avpp = *a_amv_lopts->as_amcap->amca_local_vars) != NULL){
1419 avpp += avcp->avc_prime;
1421 for(lavp = NULL, avp = *avpp; avp != NULL;
1422 lavp = avp, avp = avp->av_link)
1423 if(!strcmp(avp->av_name, avcp->avc_name)){
1424 /* Relink as head, hope it "sorts on usage" over time.
1425 * The code relies on this behaviour! */
1426 if(lavp != NULL){
1427 lavp->av_link = avp->av_link;
1428 avp->av_link = *avpp;
1429 *avpp = avp;
1431 goto jleave;
1435 if(avlf & a_AMV_VLOOK_LOCAL_ONLY)
1436 goto jerr;
1439 /* Global variable map */
1440 avpp = &a_amv_vars[avcp->avc_prime];
1442 for(lavp = NULL, avp = *avpp; avp != NULL; lavp = avp, avp = avp->av_link)
1443 if(!strcmp(avp->av_name, avcp->avc_name)){
1444 /* Relink as head, hope it "sorts on usage" over time.
1445 * The code relies on this behaviour! */
1446 if(lavp != NULL){
1447 lavp->av_link = avp->av_link;
1448 avp->av_link = *avpp;
1449 *avpp = avp;
1452 /* If this setting has been established via -S and we still have
1453 * not reached the _STARTED_CONFIG program state, it may have been
1454 * an explicit "clearance" that is to be treated as unset.
1455 * Because that is a special condition that (has been hacked in
1456 * later and) needs to be encapsulated in lower levels, but not
1457 * of interest if _set() or _clear() called us */
1458 switch(avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK){
1459 case a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET:
1460 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW)){
1461 avcp->avc_var = avp;
1462 avp = NULL;
1463 goto j_leave;
1465 /* FALLTHRU */
1466 default:
1467 break;
1469 goto jleave;
1473 /* If this is not an assembled variable we need to consider some special
1474 * initialization cases and eventually create the variable anew */
1475 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
1476 f = avmp->avm_flags;
1478 /* Does it have an import-from-environment flag? */
1479 if(n_UNLIKELY((f & (a_AMV_VF_IMPORT | a_AMV_VF_ENV)) != 0)){
1480 if(n_LIKELY((cp = getenv(avcp->avc_name)) != NULL)){
1481 /* May be better not to use that one, though? */
1482 /* TODO Outsource the tests into a _shared_ test function! */
1483 bool_t isempty, isbltin;
1485 isempty = (*cp == '\0' && (f & a_AMV_VF_NOTEMPTY) != 0);
1486 isbltin = ((f & (a_AMV_VF_I3VAL | a_AMV_VF_DEFVAL)) != 0);
1488 if(n_UNLIKELY(isempty)){
1489 n_err(_("Environment variable must not be empty: %s\n"),
1490 avcp->avc_name);
1491 if(!isbltin)
1492 goto jerr;
1493 }else if(n_LIKELY(*cp != '\0')){
1494 if(n_UNLIKELY((f & a_AMV_VF_NUM) &&
1495 !a_amv_var_check_num(cp, FAL0))){
1496 n_err(_("Environment variable value not a number "
1497 "or out of range: %s\n"), avcp->avc_name);
1498 goto jerr;
1500 if(n_UNLIKELY((f & a_AMV_VF_POSNUM) &&
1501 !a_amv_var_check_num(cp, TRU1))){
1502 n_err(_("Environment variable value not a number, "
1503 "negative or out of range: %s\n"), avcp->avc_name);
1504 goto jerr;
1506 goto jnewval;
1507 }else
1508 goto jnewval;
1512 /* A first-time init switch is to be handled now and here */
1513 if(n_UNLIKELY((f & a_AMV_VF_I3VAL) != 0)){
1514 static struct a_amv_var_defval const **arr,
1515 *arr_base[a_AMV_VAR_I3VALS_CNT +1];
1517 if(arr == NULL){
1518 arr = &arr_base[0];
1519 arr[i = a_AMV_VAR_I3VALS_CNT] = NULL;
1520 while(i-- > 0)
1521 arr[i] = &a_amv_var_i3vals[i];
1524 for(i = 0; arr[i] != NULL; ++i)
1525 if(arr[i]->avdv_okey == avcp->avc_okey){
1526 cp = (f & a_AMV_VF_BOOL) ? n_1 : arr[i]->avdv_value;
1527 /* Remove this entry, hope entire block becomes no-op asap */
1529 arr[i] = arr[i + 1];
1530 while(arr[i++] != NULL);
1532 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW))
1533 goto jnewval;
1534 if(avlf & a_AMV_VLOOK_I3VAL_NONEW_REPORT)
1535 avp = (struct a_amv_var*)-1;
1536 goto jleave;
1540 /* */
1541 jdefval:
1542 if(n_UNLIKELY(f & a_AMV_VF_DEFVAL) != 0){
1543 for(i = 0; i < a_AMV_VAR_DEFVALS_CNT; ++i)
1544 if(a_amv_var_defvals[i].avdv_okey == avcp->avc_okey){
1545 cp = (f & a_AMV_VF_BOOL) ? n_1
1546 : a_amv_var_defvals[i].avdv_value;
1547 goto jnewval;
1551 /* The virtual variables */
1552 if(n_UNLIKELY((f & a_AMV_VF_VIRT) != 0)){
1553 for(i = 0; i < a_AMV_VAR_VIRTS_CNT; ++i)
1554 if(a_amv_var_virts[i].avv_okey == avcp->avc_okey){
1555 avp = n_UNCONST(a_amv_var_virts[i].avv_var);
1556 goto jleave;
1558 /* Not reached */
1562 jerr:
1563 avp = NULL;
1564 jleave:
1565 avcp->avc_var = avp;
1566 j_leave:
1567 if(!(avlf & a_AMV_VLOOK_I3VAL_NONEW) && (n_poption & n_PO_D_VV) &&
1568 avp != (struct a_amv_var*)-1 && avcp->avc_okey != ok_v_log_prefix){
1569 /* I18N: Variable "name" is set to "value" */
1570 n_err(_("*%s* is %s\n"),
1571 n_shexp_quote_cp(avcp->avc_name, FAL0),
1572 (avp == NULL ? _("not set")
1573 : ((avp->av_flags & a_AMV_VF_BOOL) ? _("boolean set")
1574 : n_shexp_quote_cp(avp->av_value, FAL0))));
1576 NYD2_LEAVE;
1577 return (avp != NULL);
1579 jnewval:
1580 assert(avmp != NULL);
1581 assert(f == avmp->avm_flags);
1582 /* E.g., $TMPDIR may be set to non-existent, so we need to be able to catch
1583 * that and redirect to a possible default value */
1584 if((f & a_AMV_VF_VIP) &&
1585 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, &cp)){
1586 #ifdef HAVE_SETENV
1587 if(f & (a_AMV_VF_IMPORT | a_AMV_VF_ENV))
1588 unsetenv(avcp->avc_name);
1589 #endif
1590 if(n_UNLIKELY(f & a_AMV_VF_DEFVAL) != 0)
1591 goto jdefval;
1592 goto jerr;
1593 }else{
1594 struct a_amv_var **avpp;
1595 size_t l;
1597 l = strlen(avcp->avc_name) +1;
1598 avcp->avc_var =
1599 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1600 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
1601 *avpp = avp;
1602 assert(!avcp->avc_is_chain_variant);
1603 avp->av_flags = f;
1604 avp->av_value = a_amv_var_copy(cp);
1605 memcpy(avp->av_name, avcp->avc_name, l);
1607 if(f & a_AMV_VF_ENV)
1608 a_amv_var__putenv(avcp, avp);
1609 if(f & a_AMV_VF_VIP)
1610 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, &cp);
1611 goto jleave;
1615 static char const *
1616 a_amv_var_vsc_global(struct a_amv_var_carrier *avcp){
1617 char iencbuf[n_IENC_BUFFER_SIZE];
1618 char const *rv;
1619 si32_t *ep;
1620 struct a_amv_var_map const *avmp;
1621 NYD2_ENTER;
1623 /* Not function local, TODO but lazy evaluted for now */
1624 if(avcp->avc_special_prop == a_AMV_VST_QM){
1625 avmp = &a_amv_var_map[a_AMV_VAR__QM_MAP_IDX];
1626 avcp->avc_okey = ok_v___qm;
1627 ep = &n_pstate_ex_no;
1628 }else{
1629 avmp = &a_amv_var_map[a_AMV_VAR__EM_MAP_IDX];
1630 avcp->avc_okey = ok_v___em;
1631 ep = &n_pstate_err_no;
1634 /* XXX Oh heaven, we are responsible to ensure that $?/! is up-to-date
1635 * TODO we could num=1 ok_v___[qe]m, but the thing is still a string
1636 * TODO and thus conversion takes places over and over again; also
1637 * TODO for now that would have to occur before we set _that_ value
1638 * TODO so let's special treat it until we store ints as such */
1639 switch(*ep){
1640 case 0: rv = n_0; break;
1641 case 1: rv = n_1; break;
1642 default:
1643 rv = n_ienc_buf(iencbuf, *ep, 10, n_IENC_MODE_SIGNED_TYPE);
1644 break;
1646 n_PS_ROOT_BLOCK(n_var_okset(avcp->avc_okey, (uintptr_t)rv));
1648 avcp->avc_hash = avmp->avm_hash;
1649 avcp->avc_map = avmp;
1650 rv = a_amv_var_lookup(avcp, a_AMV_VLOOK_NONE)
1651 ? avcp->avc_var->av_value : NULL;
1652 NYD2_LEAVE;
1653 return rv;
1656 static char const *
1657 a_amv_var_vsc_multiplex(struct a_amv_var_carrier *avcp){
1658 char iencbuf[n_IENC_BUFFER_SIZE];
1659 si32_t e;
1660 size_t i;
1661 char const *rv;
1662 NYD2_ENTER;
1664 i = strlen(rv = &avcp->avc_name[1]);
1666 /* ERR, ERRDOC, ERRNAME, plus *-NAME variants */
1667 if(rv[0] == 'E' && i >= 3 && rv[1] == 'R' && rv[2] == 'R'){
1668 if(i == 3){
1669 e = n_pstate_err_no;
1670 goto jeno;
1671 }else if(rv[3] == '-'){
1672 e = n_err_from_name(&rv[4]);
1673 jeno:
1674 switch(e){
1675 case 0: rv = n_0; break;
1676 case 1: rv = n_1; break;
1677 default:
1678 /* XXX Need to convert number to string yet */
1679 rv = savestr(n_ienc_buf(iencbuf, e, 10, n_IENC_MODE_SIGNED_TYPE));
1680 break;
1682 goto jleave;
1683 }else if(i >= 6){
1684 if(!memcmp(&rv[3], "DOC", 3)){
1685 rv += 6;
1686 switch(*rv){
1687 case '\0': e = n_pstate_err_no; break;
1688 case '-': e = n_err_from_name(&rv[1]); break;
1689 default: goto jerr;
1691 rv = n_err_to_doc(e);
1692 goto jleave;
1693 }else if(i >= 7 && !memcmp(&rv[3], "NAME", 4)){
1694 rv += 7;
1695 switch(*rv){
1696 case '\0': e = n_pstate_err_no; break;
1697 case '-': e = n_err_from_name(&rv[1]); break;
1698 default: goto jerr;
1700 rv = n_err_to_name(e);
1701 goto jleave;
1706 jerr:
1707 rv = NULL;
1708 jleave:
1709 NYD2_LEAVE;
1710 return rv;
1713 static char const *
1714 a_amv_var_vsc_pospar(struct a_amv_var_carrier *avcp){
1715 size_t i, j;
1716 ui16_t argc;
1717 char const *rv, **argv;
1718 NYD2_ENTER;
1720 rv = NULL;
1722 /* If in a macro/xy.. */
1723 if(a_amv_lopts != NULL){
1724 bool_t ismacky;
1725 struct a_amv_mac_call_args *amcap;
1727 amcap = a_amv_lopts->as_amcap;
1728 argc = amcap->amca_pospar.app_count;
1729 argv = amcap->amca_pospar.app_dat;
1730 argv += amcap->amca_pospar.app_idx;
1732 /* ..in a `call'ed macro only, to be exact. Or in a_AMV_MACKY_MACK */
1733 if(!(ismacky = (amcap->amca_amp == a_AMV_MACKY_MACK)) &&
1734 (amcap->amca_ps_hook_mask ||
1735 (assert(amcap->amca_amp != NULL),
1736 (amcap->amca_amp->am_flags & a_AMV_MF_TYPE_MASK
1737 ) == a_AMV_MF_ACCOUNT)))
1738 goto jleave;
1740 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1741 if(avcp->avc_special_prop > 0){
1742 if(argc >= avcp->avc_special_prop)
1743 rv = argv[avcp->avc_special_prop - 1];
1744 }else if(ismacky)
1745 rv = amcap->amca_name;
1746 else
1747 rv = (a_amv_lopts->as_up != NULL
1748 ? a_amv_lopts->as_up->as_amcap->amca_name : n_empty);
1749 goto jleave;
1751 /* MACKY_MACK doesn't know about [*@#] */
1752 /*else*/ if(ismacky){
1753 if(n_poption & n_PO_D_V)
1754 n_err(_("Cannot use $*/$@/$# in this context: %s\n"),
1755 n_shexp_quote_cp(avcp->avc_name, FAL0));
1756 goto jleave;
1758 }else{
1759 argc = a_amv_pospar.app_count;
1760 argv = a_amv_pospar.app_dat;
1761 argv += a_amv_pospar.app_idx;
1763 if(avcp->avc_special_cat == a_AMV_VSC_POSPAR){
1764 if(avcp->avc_special_prop > 0){
1765 if(argc >= avcp->avc_special_prop)
1766 rv = argv[avcp->avc_special_prop - 1];
1767 }else
1768 rv = n_progname;
1769 goto jleave;
1773 switch(avcp->avc_special_prop){ /* XXX OPTIMIZE */
1774 case a_AMV_VST_STAR:{
1775 char sep;
1777 sep = *ok_vlook(ifs);
1778 if(0){
1779 case a_AMV_VST_AT:
1780 sep = ' ';
1782 for(i = j = 0; i < argc; ++i)
1783 j += strlen(argv[i]) + 1;
1784 if(j == 0)
1785 rv = n_empty;
1786 else{
1787 char *cp;
1789 rv = cp = n_autorec_alloc(j);
1790 for(i = j = 0; i < argc; ++i){
1791 j = strlen(argv[i]);
1792 memcpy(cp, argv[i], j);
1793 cp += j;
1794 if(sep != '\0')
1795 *cp++ = sep;
1797 if(sep != '\0')
1798 --cp;
1799 *cp = '\0';
1801 }break;
1802 case a_AMV_VST_NOSIGN:{
1803 char iencbuf[n_IENC_BUFFER_SIZE];
1805 rv = savestr(n_ienc_buf(iencbuf, argc, 10, n_IENC_MODE_NONE));
1806 }break;
1807 default:
1808 rv = n_empty;
1809 break;
1811 jleave:
1812 NYD2_LEAVE;
1813 return rv;
1816 static bool_t
1817 a_amv_var_set(struct a_amv_var_carrier *avcp, char const *value,
1818 enum a_amv_var_setclr_flags avscf){
1819 struct a_amv_var *avp;
1820 char *oval;
1821 struct a_amv_var_map const *avmp;
1822 bool_t rv;
1823 NYD2_ENTER;
1825 if(value == NULL){
1826 rv = a_amv_var_clear(avcp, avscf);
1827 goto jleave;
1830 if((avmp = avcp->avc_map) != NULL){
1831 rv = FAL0;
1833 /* Validity checks */
1834 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_RDONLY) != 0 &&
1835 !(n_pstate & n_PS_ROOT))){
1836 value = N_("Variable is read-only: %s\n");
1837 goto jeavmp;
1839 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NOTEMPTY) && *value == '\0')){
1840 value = N_("Variable must not be empty: %s\n");
1841 goto jeavmp;
1843 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_NUM) &&
1844 !a_amv_var_check_num(value, FAL0))){
1845 value = N_("Variable value not a number or out of range: %s\n");
1846 goto jeavmp;
1848 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_POSNUM) &&
1849 !a_amv_var_check_num(value, TRU1))){
1850 value = _("Variable value not a number, negative, "
1851 "or out of range: %s\n");
1852 goto jeavmp;
1855 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_IMPORT) != 0 &&
1856 !(n_psonce & n_PSO_STARTED) && !(n_pstate & n_PS_ROOT))){
1857 value = N_("Variable cannot be set in a resource file: %s\n");
1858 goto jeavmp;
1861 /* Any more complicated inter-dependency? */
1862 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_VIP) != 0 &&
1863 !a_amv_var_check_vips(a_AMV_VIP_SET_PRE, avcp->avc_okey, &value))){
1864 value = N_("Assignment of variable aborted: %s\n");
1865 jeavmp:
1866 n_err(V_(value), avcp->avc_name);
1867 goto jleave;
1870 /* Transformations */
1871 if(n_UNLIKELY(avmp->avm_flags & a_AMV_VF_LOWER)){
1872 char c;
1874 oval = savestr(value);
1875 value = oval;
1876 for(; (c = *oval) != '\0'; ++oval)
1877 *oval = lowerconv(c);
1880 /* Obsoletion warning */
1881 if(n_UNLIKELY((avmp->avm_flags & a_AMV_VF_OBSOLETE) != 0))
1882 n_OBSOLETE2(_("variable superseded or obsoleted"), avcp->avc_name);
1885 /* Lookup possibly existing var. For */
1887 rv = TRU1;
1888 a_amv_var_lookup(avcp, (a_AMV_VLOOK_I3VAL_NONEW |
1889 ((avscf & a_AMV_VSETCLR_LOCAL)
1890 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0)));
1891 avp = avcp->avc_var;
1893 /* A `local' setting is never covered by `localopts' nor frozen */
1894 if(avscf & a_AMV_VSETCLR_LOCAL)
1895 goto jislocal;
1897 /* If this setting had been established via -S and we still have not reached
1898 * the _STARTED_CONFIG program state, silently ignore request! */
1899 if(n_UNLIKELY(avp != NULL) &&
1900 n_UNLIKELY((avp->av_flags & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
1901 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
1902 if((n_pstate & n_PS_ROOT) ||
1903 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
1904 (n_poption & n_PO_S_FLAG_TEMPORARY)))
1905 goto joval_and_go;
1906 if(n_poption & n_PO_D_VV)
1907 n_err(_("Temporarily frozen by -S, not `set'ing: %s\n"),
1908 avcp->avc_name);
1909 goto jleave;
1912 /* Otherwise, if -S freezing was an `unset' request, be very simple and
1913 * avoid tampering with that very special case we are not really prepared
1914 * for just one more line of code: throw the old thing away! */
1915 if(!(avp->av_flags & a_AMV_VF_EXT_FROZEN_UNSET))
1916 avp->av_flags &= ~a_AMV_VF_EXT__FROZEN_MASK;
1917 else{
1918 assert(avp->av_value == n_empty);
1919 assert(a_amv_vars[avcp->avc_prime] == avp);
1920 a_amv_vars[avcp->avc_prime] = avp->av_link;
1921 n_free(avp);
1922 avcp->avc_var = avp = NULL;
1926 /* Optionally cover by `localopts' */
1927 if(n_UNLIKELY(a_amv_lopts != NULL) &&
1928 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
1929 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
1931 jislocal:
1932 if(avp != NULL)
1933 joval_and_go:
1934 oval = avp->av_value;
1935 else{
1936 size_t l;
1937 struct a_amv_var **avpp;
1939 if(avscf & a_AMV_VSETCLR_LOCAL){
1940 if((avpp = *a_amv_lopts->as_amcap->amca_local_vars) == NULL)
1941 avpp = *(a_amv_lopts->as_amcap->amca_local_vars =
1942 n_calloc(1, sizeof(*a_amv_lopts->as_amcap->amca_local_vars)));
1943 avpp += avcp->avc_prime;
1944 }else
1945 avpp = &a_amv_vars[avcp->avc_prime];
1947 l = strlen(avcp->avc_name) +1;
1948 avcp->avc_var = avp = n_calloc(1,
1949 n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
1950 avp->av_link = *avpp;
1951 *avpp = avp;
1952 avp->av_flags = (((avscf & a_AMV_VSETCLR_LOCAL)
1953 ? a_AMV_VF_NOLOPTS | a_AMV_VF_EXT_LOCAL
1954 : ((avmp != NULL) ? avmp->avm_flags : 0)) |
1955 (avcp->avc_is_chain_variant ? a_AMV_VF_EXT_CHAIN : a_AMV_VF_NONE));
1956 memcpy(avp->av_name, avcp->avc_name, l);
1957 oval = n_UNCONST(n_empty);
1960 if(avmp == NULL)
1961 avp->av_value = a_amv_var_copy(value);
1962 else{
1963 assert(!(avscf & a_AMV_VSETCLR_LOCAL));
1964 /* Via `set' etc. the user may give even boolean options non-boolean
1965 * values, ignore that and force boolean */
1966 if(!(avp->av_flags & a_AMV_VF_BOOL))
1967 avp->av_value = a_amv_var_copy(value);
1968 else{
1969 if(!(n_pstate & n_PS_ROOT) && (n_poption & n_PO_D_V) &&
1970 *value != '\0')
1971 n_err(_("Ignoring value of boolean variable: %s: %s\n"),
1972 avcp->avc_name, value);
1973 avp->av_value = n_UNCONST(n_1);
1977 /* A `local' setting can skip all the crude special things */
1978 if(!(avscf & a_AMV_VSETCLR_LOCAL)){
1979 ui32_t f;
1981 f = avp->av_flags;
1983 if((avscf & a_AMV_VSETCLR_ENV) && !(f & a_AMV_VF_ENV))
1984 f |= a_AMV_VF_EXT_LINKED;
1985 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
1986 rv = a_amv_var__putenv(avcp, avp);
1987 if(f & a_AMV_VF_VIP)
1988 a_amv_var_check_vips(a_AMV_VIP_SET_POST, avcp->avc_okey, &value);
1990 f &= ~a_AMV_VF_EXT__FROZEN_MASK;
1991 if(!(n_psonce & n_PSO_STARTED_GETOPT) &&
1992 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0)
1993 f |= a_AMV_VF_EXT_FROZEN;
1995 avp->av_flags = f;
1998 a_amv_var_free(oval);
1999 jleave:
2000 NYD2_LEAVE;
2001 return rv;
2004 static bool_t
2005 a_amv_var__putenv(struct a_amv_var_carrier *avcp, struct a_amv_var *avp){
2006 #ifndef HAVE_SETENV
2007 char *cp;
2008 #endif
2009 bool_t rv;
2010 NYD2_ENTER;
2012 #ifdef HAVE_SETENV
2013 rv = (setenv(avcp->avc_name, avp->av_value, 1) == 0);
2014 #else
2015 cp = sstrdup(savecatsep(avcp->avc_name, '=', avp->av_value));
2017 if((rv = (putenv(cp) == 0))){
2018 char *ocp;
2020 ocp = avp->av_env;
2021 avp->av_env = cp;
2022 cp = ocp;
2025 if(cp != NULL)
2026 n_free(cp);
2027 #endif
2028 NYD2_LEAVE;
2029 return rv;
2032 static bool_t
2033 a_amv_var_clear(struct a_amv_var_carrier *avcp,
2034 enum a_amv_var_setclr_flags avscf){
2035 struct a_amv_var **avpp, *avp;
2036 ui32_t f;
2037 struct a_amv_var_map const *avmp;
2038 bool_t rv;
2039 NYD2_ENTER;
2041 rv = FAL0;
2042 f = 0;
2044 if(n_LIKELY((avmp = avcp->avc_map) != NULL)){
2045 if(n_UNLIKELY(((f = avmp->avm_flags) & a_AMV_VF_NODEL) != 0 &&
2046 !(n_pstate & n_PS_ROOT))){
2047 n_err(_("Variable may not be unset: %s\n"), avcp->avc_name);
2048 goto jleave;
2050 if(n_UNLIKELY((f & a_AMV_VF_VIP) != 0 &&
2051 !a_amv_var_check_vips(a_AMV_VIP_CLEAR, avcp->avc_okey, NULL))){
2052 n_err(_("Clearance of variable aborted: %s\n"), avcp->avc_name);
2053 goto jleave;
2057 rv = TRU1;
2059 if(n_UNLIKELY(!a_amv_var_lookup(avcp,
2060 (((avscf & a_AMV_VSETCLR_LOCAL)
2061 ? (a_AMV_VLOOK_LOCAL | a_AMV_VLOOK_LOCAL_ONLY) : 0) |
2062 a_AMV_VLOOK_I3VAL_NONEW | a_AMV_VLOOK_I3VAL_NONEW_REPORT)))){
2063 assert(avcp->avc_var == NULL);
2064 /* This may be a clearance request from the command line, via -S, and we
2065 * need to keep track of that! Unfortunately we are not prepared for
2066 * this, really, so we need to create a fake entry that is known and
2067 * handled correctly by the lowermost variable layer!
2068 * However, all this cannot happen for plain unset of `local' variables */
2069 if(avscf & a_AMV_VSETCLR_LOCAL)
2070 goto jleave;
2071 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
2072 (n_poption & n_PO_S_FLAG_TEMPORARY)) Jfreeze:{
2073 size_t l;
2075 l = strlen(avcp->avc_name) +1;
2076 avp = n_calloc(1, n_VSTRUCT_SIZEOF(struct a_amv_var, av_name) + l);
2077 avp->av_link = *(avpp = &a_amv_vars[avcp->avc_prime]);
2078 *avpp = avp;
2079 avp->av_value = n_UNCONST(n_empty); /* Sth. covered by _var_free()! */
2080 assert(f == (avmp != NULL ? avmp->avm_flags : 0));
2081 avp->av_flags = f | a_AMV_VF_EXT_FROZEN | a_AMV_VF_EXT_FROZEN_UNSET;
2082 memcpy(avp->av_name, avcp->avc_name, l);
2084 if((avscf & a_AMV_VSETCLR_ENV) || (f & a_AMV_VF_ENV))
2085 a_amv_var__clearenv(avcp->avc_name, NULL);
2086 }else if(avscf & a_AMV_VSETCLR_ENV){
2087 jforce_env:
2088 if(!(rv = a_amv_var__clearenv(avcp->avc_name, NULL)))
2089 goto jerr_env_unset;
2090 }else{
2091 /* TODO "cannot unset undefined variable" not echoed in "ROBOT" state,
2092 * TODO should only be like that with "ignerr"! */
2093 jerr_env_unset:
2094 if(!(n_pstate & (n_PS_ROOT | n_PS_ROBOT)) && (n_poption & n_PO_D_V))
2095 n_err(_("Cannot unset undefined variable: %s\n"), avcp->avc_name);
2097 goto jleave;
2098 }else if((avp = avcp->avc_var) == (struct a_amv_var*)-1){
2099 /* Clearance request from command line, via -S? As above.. */
2100 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT) &&
2101 (n_poption & n_PO_S_FLAG_TEMPORARY) != 0))
2102 goto Jfreeze;
2103 avcp->avc_var = NULL;
2104 if(avscf & a_AMV_VSETCLR_ENV)
2105 goto jforce_env;
2106 goto jleave;
2108 assert(avcp->avc_var != NULL);
2110 /* `local' variables bypass "frozen" checks and `localopts' coverage etc. */
2111 if((f = avp->av_flags) & a_AMV_VF_EXT_LOCAL)
2112 goto jdefault_path;
2114 /* If this setting has been established via -S and we still have not reached
2115 * the _STARTED_CONFIG program state, silently ignore request!
2116 * XXX All this is very complicated for the tenth of a second */
2117 /*else*/ if(n_UNLIKELY((f & a_AMV_VF_EXT__FROZEN_MASK) != 0)){
2118 if(!(n_psonce & n_PSO_STARTED_CONFIG)){
2119 if((n_pstate & n_PS_ROOT) ||
2120 (!(n_psonce & n_PSO_STARTED_GETOPT) &&
2121 (n_poption & n_PO_S_FLAG_TEMPORARY))){
2122 /* Be aware this may turn a set into an unset! */
2123 if(!(f & a_AMV_VF_EXT_FROZEN_UNSET)){
2124 if(f & a_AMV_VF_DEFVAL)
2125 goto jdefault_path;
2126 a_amv_var_free(avp->av_value);
2127 f |= a_AMV_VF_EXT_FROZEN_UNSET;
2128 avp->av_flags = f;
2129 avp->av_value = n_UNCONST(n_empty); /* _var_free() covered */
2130 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
2131 goto jforce_env;
2133 goto jleave;
2135 if(n_poption & n_PO_D_VV)
2136 n_err(_("Temporarily frozen by -S, not `unset'ting: %s\n"),
2137 avcp->avc_name);
2138 goto jleave;
2140 f &= ~a_AMV_VF_EXT__FROZEN_MASK;
2141 avp->av_flags = f;
2144 if(n_UNLIKELY(a_amv_lopts != NULL) &&
2145 (avmp == NULL || !(avmp->avm_flags & a_AMV_VF_NOLOPTS)))
2146 a_amv_lopts_add(a_amv_lopts, avcp->avc_name, avcp->avc_var);
2148 jdefault_path:
2149 assert(avp == avcp->avc_var);
2150 avcp->avc_var = NULL;
2151 avpp = &(((f = avp->av_flags) & a_AMV_VF_EXT_LOCAL)
2152 ? *a_amv_lopts->as_amcap->amca_local_vars : a_amv_vars
2153 )[avcp->avc_prime];
2154 assert(*avpp == avp); /* (always listhead after lookup()) */
2155 *avpp = (*avpp)->av_link;
2157 if(f & (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED))
2158 rv = a_amv_var__clearenv(avp->av_name, avp);
2159 a_amv_var_free(avp->av_value);
2160 n_free(avp);
2162 /* XXX Fun part, extremely simple-minded for now: if this variable has
2163 * XXX a default value, immediately reinstantiate it! TODO Heh? */
2164 /* xxx Simply assuming we will never have default values for actual
2165 * xxx -HOST or -USER@HOST chain extensions */
2166 if(n_UNLIKELY(avmp != NULL && (avmp->avm_flags & a_AMV_VF_DEFVAL) != 0)){
2167 a_amv_var_lookup(avcp, a_AMV_VLOOK_I3VAL_NONEW);
2168 if(n_UNLIKELY(!(n_psonce & n_PSO_STARTED_GETOPT)) &&
2169 (n_poption & n_PO_S_FLAG_TEMPORARY))
2170 avcp->avc_var->av_flags |= a_AMV_VF_EXT_FROZEN;
2172 jleave:
2173 NYD2_LEAVE;
2174 return rv;
2177 static bool_t
2178 a_amv_var__clearenv(char const *name, struct a_amv_var *avp){
2179 extern char **environ;
2180 char **ecpp;
2181 bool_t rv;
2182 NYD2_ENTER;
2183 n_UNUSED(avp);
2185 rv = FAL0;
2186 ecpp = environ;
2188 #ifndef HAVE_SETENV
2189 if(avp != NULL && avp->av_env != NULL){
2190 for(; *ecpp != NULL; ++ecpp)
2191 if(*ecpp == avp->av_env){
2193 ecpp[0] = ecpp[1];
2194 while(*ecpp++ != NULL);
2195 n_free(avp->av_env);
2196 avp->av_env = NULL;
2197 rv = TRU1;
2198 break;
2200 }else
2201 #endif
2203 size_t l;
2205 if((l = strlen(name)) > 0){
2206 for(; *ecpp != NULL; ++ecpp)
2207 if(!strncmp(*ecpp, name, l) && (*ecpp)[l] == '='){
2208 #ifdef HAVE_SETENV
2209 unsetenv(name);
2210 #else
2212 ecpp[0] = ecpp[1];
2213 while(*ecpp++ != NULL);
2214 #endif
2215 rv = TRU1;
2216 break;
2220 NYD2_LEAVE;
2221 return rv;
2224 static void
2225 a_amv_var_show_all(void){
2226 struct n_string msg, *msgp;
2227 FILE *fp;
2228 size_t no, i;
2229 struct a_amv_var *avp;
2230 char const **vacp, **cap;
2231 NYD2_ENTER;
2233 if((fp = Ftmp(NULL, "setlist", OF_RDWR | OF_UNLINK | OF_REGISTER)) == NULL){
2234 n_perr(_("`set' list: cannot create temporary file"), 0);
2235 goto jleave;
2238 /* We need to instantiate first-time-inits and default values here, so that
2239 * they will be regular members of our _vars[] table */
2240 for(i = a_AMV_VAR_I3VALS_CNT; i-- > 0;)
2241 n_var_oklook(a_amv_var_i3vals[i].avdv_okey);
2242 for(i = a_AMV_VAR_DEFVALS_CNT; i-- > 0;)
2243 n_var_oklook(a_amv_var_defvals[i].avdv_okey);
2245 for(no = i = 0; i < a_AMV_PRIME; ++i)
2246 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2247 ++no;
2248 no += a_AMV_VAR_VIRTS_CNT;
2250 vacp = n_autorec_alloc(no * sizeof(*vacp));
2252 for(cap = vacp, i = 0; i < a_AMV_PRIME; ++i)
2253 for(avp = a_amv_vars[i]; avp != NULL; avp = avp->av_link)
2254 *cap++ = avp->av_name;
2255 for(i = a_AMV_VAR_VIRTS_CNT; i-- > 0;)
2256 *cap++ = a_amv_var_virts[i].avv_var->av_name;
2258 if(no > 1)
2259 qsort(vacp, no, sizeof *vacp, &a_amv_var__show_cmp);
2261 msgp = &msg;
2262 msgp = n_string_reserve(n_string_creat(msgp), 80);
2263 for(i = 0, cap = vacp; no != 0; ++cap, --no)
2264 i += a_amv_var_show(*cap, fp, msgp);
2265 n_string_gut(&msg);
2267 page_or_print(fp, i);
2268 Fclose(fp);
2269 jleave:
2270 NYD2_LEAVE;
2273 static int
2274 a_amv_var__show_cmp(void const *s1, void const *s2){
2275 int rv;
2276 NYD2_ENTER;
2278 rv = strcmp(*(char**)n_UNCONST(s1), *(char**)n_UNCONST(s2));
2279 NYD2_LEAVE;
2280 return rv;
2283 static size_t
2284 a_amv_var_show(char const *name, FILE *fp, struct n_string *msgp){
2285 /* XXX a_amv_var_show(): if we iterate over all the actually set variables
2286 * XXX via a_amv_var_show_all() there is no need to call
2287 * XXX a_amv_var_revlookup() at all! Revisit this call chain */
2288 struct a_amv_var_carrier avc;
2289 char const *quote;
2290 struct a_amv_var *avp;
2291 bool_t isset;
2292 size_t i;
2293 NYD2_ENTER;
2295 msgp = n_string_trunc(msgp, 0);
2296 i = 0;
2298 a_amv_var_revlookup(&avc, name, TRU1);
2299 isset = a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
2300 avp = avc.avc_var;
2302 if(n_poption & n_PO_D_V){
2303 if(avc.avc_map == NULL){
2304 msgp = n_string_push_cp(msgp, "#assembled variable with value");
2305 i = 1;
2306 }else{
2307 struct{
2308 ui16_t flag;
2309 char msg[22];
2310 } const tbase[] = {
2311 {a_AMV_VF_CHAIN, "variable chain"},
2312 {a_AMV_VF_VIRT, "virtual"},
2313 {a_AMV_VF_RDONLY, "read-only"},
2314 {a_AMV_VF_NODEL, "nodelete"},
2315 {a_AMV_VF_I3VAL, "initial-value"},
2316 {a_AMV_VF_DEFVAL, "default-value"},
2317 {a_AMV_VF_IMPORT, "import-environ-first\0"}, /* assert NUL in max */
2318 {a_AMV_VF_ENV, "sync-environ"},
2319 {a_AMV_VF_NOLOPTS, "no-localopts"},
2320 {a_AMV_VF_NOTEMPTY, "notempty"},
2321 {a_AMV_VF_NUM, "number"},
2322 {a_AMV_VF_POSNUM, "positive-number"},
2323 {a_AMV_VF_OBSOLETE, "obsoleted"},
2324 }, *tp;
2325 assert(!isset || ((avp->av_flags & a_AMV_VF__MASK) ==
2326 (avc.avc_map->avm_flags & a_AMV_VF__MASK)));
2328 for(tp = tbase; PTRCMP(tp, <, &tbase[n_NELEM(tbase)]); ++tp)
2329 if(isset ? (avp->av_flags & tp->flag)
2330 : (avc.avc_map->avm_flags & tp->flag)){
2331 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2332 msgp = n_string_push_cp(msgp, tp->msg);
2333 if(isset){
2334 if((tp->flag == a_AMV_VF_CHAIN) &&
2335 (avp->av_flags & a_AMV_VF_EXT_CHAIN))
2336 msgp = n_string_push_cp(msgp, " (extension)");
2341 if(isset){
2342 if(avp->av_flags & a_AMV_VF_EXT_FROZEN){
2343 msgp = n_string_push_c(msgp, (i++ == 0 ? '#' : ','));
2344 msgp = n_string_push_cp(msgp, "(un)?set via -S");
2348 if(i > 0)
2349 msgp = n_string_push_cp(msgp, "\n ");
2352 /* (Read-only variables are generally shown via comments..) */
2353 if(!isset || (avp->av_flags & a_AMV_VF_RDONLY)){
2354 msgp = n_string_push_c(msgp, n_ns[0]);
2355 if(!isset){
2356 if(avc.avc_map != NULL && (avc.avc_map->avm_flags & a_AMV_VF_BOOL))
2357 msgp = n_string_push_cp(msgp, "boolean; ");
2358 msgp = n_string_push_cp(msgp, "variable not set: ");
2359 msgp = n_string_push_cp(msgp, n_shexp_quote_cp(name, FAL0));
2360 goto jleave;
2364 n_UNINIT(quote, NULL);
2365 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2366 quote = n_shexp_quote_cp(avp->av_value, TRU1);
2367 if(strcmp(quote, avp->av_value))
2368 msgp = n_string_push_cp(msgp, "wysh ");
2369 }else if(n_poption & n_PO_D_V)
2370 msgp = n_string_push_cp(msgp, "wysh "); /* (for shell-style comment) */
2372 if(avp->av_flags & a_AMV_VF_EXT_LINKED)
2373 msgp = n_string_push_cp(msgp, "environ ");
2374 msgp = n_string_push_cp(msgp, "set ");
2375 msgp = n_string_push_cp(msgp, name);
2377 if(!(avp->av_flags & a_AMV_VF_BOOL)){
2378 msgp = n_string_push_c(msgp, '=');
2379 msgp = n_string_push_cp(msgp, quote);
2380 }else if(n_poption & n_PO_D_V)
2381 msgp = n_string_push_cp(msgp, " #boolean");
2383 jleave:
2384 msgp = n_string_push_c(msgp, '\n');
2385 fputs(n_string_cp(msgp), fp);
2386 NYD2_ENTER;
2387 return (i > 0 ? 2 : 1);
2390 static bool_t
2391 a_amv_var_c_set(char **ap, enum a_amv_var_setclr_flags avscf){
2392 char *cp, *cp2, *varbuf, c;
2393 size_t errs;
2394 NYD2_ENTER;
2396 errs = 0;
2397 jouter:
2398 while((cp = *ap++) != NULL){
2399 /* Isolate key */
2400 cp2 = varbuf = n_autorec_alloc(strlen(cp) +1);
2402 for(; (c = *cp) != '=' && c != '\0'; ++cp){
2403 if(cntrlchar(c) || spacechar(c)){
2404 n_err(_("Variable name with control or space character ignored: "
2405 "%s\n"), ap[-1]);
2406 ++errs;
2407 goto jouter;
2409 *cp2++ = c;
2411 *cp2 = '\0';
2412 if(c == '\0')
2413 cp = n_UNCONST(n_empty);
2414 else
2415 ++cp;
2417 if(varbuf == cp2){
2418 n_err(_("Empty variable name ignored\n"));
2419 ++errs;
2420 }else{
2421 struct a_amv_var_carrier avc;
2422 bool_t isunset;
2424 if((isunset = (varbuf[0] == 'n' && varbuf[1] == 'o')))
2425 varbuf = &varbuf[2];
2427 a_amv_var_revlookup(&avc, varbuf, TRU1);
2429 if((avscf & a_AMV_VSETCLR_LOCAL) && avc.avc_map != NULL){
2430 if(n_poption & n_PO_D_V)
2431 n_err(_("Builtin variable not overwritten by `local': %s\n"),
2432 varbuf);
2433 ++errs;
2434 }else if(isunset)
2435 errs += !a_amv_var_clear(&avc, avscf);
2436 else
2437 errs += !a_amv_var_set(&avc, cp, avscf);
2440 NYD2_LEAVE;
2441 return (errs == 0);
2444 FL int
2445 c_define(void *v){
2446 int rv;
2447 char **args;
2448 NYD_ENTER;
2450 rv = 1;
2452 if((args = v)[0] == NULL){
2453 rv = (a_amv_mac_show(a_AMV_MF_NONE) == FAL0);
2454 goto jleave;
2457 if(args[1] == NULL || args[1][0] != '{' || args[1][1] != '\0' ||
2458 args[2] != NULL){
2459 n_err(_("Synopsis: define: <name> {\n"));
2460 goto jleave;
2463 rv = (a_amv_mac_def(args[0], a_AMV_MF_NONE) == FAL0);
2464 jleave:
2465 NYD_LEAVE;
2466 return rv;
2469 FL int
2470 c_undefine(void *v){
2471 int rv;
2472 char **args;
2473 NYD_ENTER;
2475 rv = 0;
2476 args = v;
2478 rv |= !a_amv_mac_undef(*args, a_AMV_MF_NONE);
2479 while(*++args != NULL);
2480 NYD_LEAVE;
2481 return rv;
2484 FL int
2485 c_call(void *vp){
2486 int rv;
2487 NYD_ENTER;
2489 rv = a_amv_mac_call(vp, FAL0);
2490 NYD_LEAVE;
2491 return rv;
2494 FL int
2495 c_call_if(void *vp){
2496 int rv;
2497 NYD_ENTER;
2499 rv = a_amv_mac_call(vp, TRU1);
2500 NYD_LEAVE;
2501 return rv;
2504 FL int
2505 c_account(void *v){
2506 struct a_amv_mac_call_args *amcap;
2507 struct a_amv_mac *amp;
2508 char **args;
2509 int rv, i, oqf, nqf;
2510 NYD_ENTER;
2512 rv = 1;
2514 if((args = v)[0] == NULL){
2515 rv = (a_amv_mac_show(a_AMV_MF_ACCOUNT) == FAL0);
2516 goto jleave;
2519 if(args[1] && args[1][0] == '{' && args[1][1] == '\0'){
2520 if(args[2] != NULL){
2521 n_err(_("Synopsis: account: <name> {\n"));
2522 goto jleave;
2524 if(!asccasecmp(args[0], ACCOUNT_NULL)){
2525 n_err(_("`account': cannot use reserved name: %s\n"),
2526 ACCOUNT_NULL);
2527 goto jleave;
2529 rv = (a_amv_mac_def(args[0], a_AMV_MF_ACCOUNT) == FAL0);
2530 goto jleave;
2533 if(n_pstate & n_PS_HOOK_MASK){
2534 n_err(_("`account': cannot change account from within a hook\n"));
2535 goto jleave;
2538 save_mbox_for_possible_quitstuff();
2540 amp = NULL;
2541 if(asccasecmp(args[0], ACCOUNT_NULL) != 0 &&
2542 (amp = a_amv_mac_lookup(args[0], NULL, a_AMV_MF_ACCOUNT)) == NULL){
2543 n_err(_("`account': account does not exist: %s\n"), args[0]);
2544 goto jleave;
2547 oqf = savequitflags();
2549 /* Shutdown the active account */
2550 if(a_amv_acc_curr != NULL){
2551 char const *cp;
2552 char *var;
2554 /* Is there a cleanup hook? */
2555 var = savecat("on-account-cleanup-", a_amv_acc_curr->am_name);
2556 if((cp = n_var_vlook(var, FAL0)) != NULL ||
2557 (cp = ok_vlook(on_account_cleanup)) != NULL){
2558 struct a_amv_mac *amphook;
2560 if((amphook = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) != NULL){
2561 amcap = n_lofi_alloc(sizeof *amcap);
2562 memset(amcap, 0, sizeof *amcap);
2563 amcap->amca_name = cp;
2564 amcap->amca_amp = amphook;
2565 amcap->amca_unroller = &a_amv_acc_curr->am_lopts;
2566 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2567 amcap->amca_no_xcall = TRU1;
2568 n_pstate |= n_PS_HOOK;
2569 rv = a_amv_mac_exec(amcap);
2570 n_pstate &= ~n_PS_HOOK_MASK;
2571 }else
2572 n_err(_("*on-account-leave* hook %s does not exist\n"),
2573 n_shexp_quote_cp(cp, FAL0));
2576 /* `localopts'? */
2577 if(a_amv_acc_curr->am_lopts != NULL)
2578 a_amv_lopts_unroll(&a_amv_acc_curr->am_lopts);
2580 /* For accounts this lingers */
2581 --a_amv_acc_curr->am_refcnt;
2582 if(a_amv_acc_curr->am_flags & a_AMV_MF_DELETE)
2583 a_amv_mac_free(a_amv_acc_curr);
2586 a_amv_acc_curr = amp;
2588 /* And switch to any non-"null" account */
2589 if(amp != NULL){
2590 assert(amp->am_lopts == NULL);
2591 amcap = n_lofi_alloc(sizeof *amcap);
2592 memset(amcap, 0, sizeof *amcap);
2593 amcap->amca_name = amp->am_name;
2594 amcap->amca_amp = amp;
2595 amcap->amca_unroller = &amp->am_lopts;
2596 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2597 amcap->amca_no_xcall = TRU1;
2598 ++amp->am_refcnt; /* We may not run 0 to avoid being deleted! */
2599 if(!a_amv_mac_exec(amcap)){
2600 /* XXX account switch incomplete, unroll? */
2601 n_err(_("`account': failed to switch to account: %s\n"), amp->am_name);
2602 goto jleave;
2606 n_PS_ROOT_BLOCK((amp != NULL ? ok_vset(account, amp->am_name)
2607 : ok_vclear(account)));
2609 /* Otherwise likely initial setfile() in a_main_rcv_mode() will pick up */
2610 if(n_psonce & n_PSO_STARTED){
2611 assert(!(n_pstate & n_PS_HOOK_MASK));
2612 nqf = savequitflags(); /* TODO obsolete (leave -> void -> new box!) */
2613 restorequitflags(oqf);
2614 i = setfile("%", FEDIT_SYSBOX | FEDIT_ACCOUNT);
2615 restorequitflags(nqf);
2616 if(i < 0)
2617 goto jleave;
2618 temporary_folder_hook_check(FAL0);
2619 if(i != 0 && !ok_blook(emptystart)) /* Avoid annoying "double message" */
2620 goto jleave;
2621 n_folder_announce(n_ANNOUNCE_CHANGE);
2623 rv = 0;
2624 jleave:
2625 NYD_LEAVE;
2626 return rv;
2629 FL int
2630 c_unaccount(void *v){
2631 int rv;
2632 char **args;
2633 NYD_ENTER;
2635 rv = 0;
2636 args = v;
2638 rv |= !a_amv_mac_undef(*args, a_AMV_MF_ACCOUNT);
2639 while(*++args != NULL);
2640 NYD_LEAVE;
2641 return rv;
2644 FL int
2645 c_localopts(void *vp){
2646 enum a_amv_loflags alf, alm;
2647 char const **argv;
2648 int rv;
2649 NYD_ENTER;
2651 rv = 1;
2653 if(a_amv_lopts == NULL){
2654 n_err(_("Cannot use `localopts' in this context\n"));
2655 goto jleave;
2658 if((argv = vp)[1] == NULL || is_asccaseprefix((++argv)[-1], "scope"))
2659 alf = alm = a_AMV_LF_SCOPE;
2660 else if(is_asccaseprefix(argv[-1], "call"))
2661 alf = a_AMV_LF_CALL, alm = a_AMV_LF_CALL_MASK;
2662 else if(is_asccaseprefix(argv[-1], "call-fixate"))
2663 alf = a_AMV_LF_CALL_FIXATE, alm = a_AMV_LF_CALL_MASK;
2664 else{
2665 jesynopsis:
2666 n_err(_("Synopsis: localopts: [<scope|call|call-fixate>] <boolean>\n"));
2667 goto jleave;
2670 if(alf == a_AMV_LF_SCOPE &&
2671 (a_amv_lopts->as_loflags & a_AMV_LF_SCOPE_FIXATE)){
2672 if(n_poption & n_PO_D_V)
2673 n_err(_("Cannot turn off `localopts', setting is fixated\n"));
2674 goto jleave;
2677 if((rv = n_boolify(*argv, UIZ_MAX, FAL0)) < FAL0)
2678 goto jesynopsis;
2679 a_amv_lopts->as_loflags &= ~alm;
2680 if(rv > FAL0)
2681 a_amv_lopts->as_loflags |= alf;
2682 rv = 0;
2683 jleave:
2684 NYD_LEAVE;
2685 return rv;
2688 FL int
2689 c_shift(void *vp){ /* xxx move to bottom, not in macro part! */
2690 struct a_amv_pospar *appp;
2691 ui16_t i;
2692 int rv;
2693 NYD_ENTER;
2695 rv = 1;
2697 if((vp = *(char**)vp) == NULL)
2698 i = 1;
2699 else{
2700 si16_t sib;
2702 if((n_idec_si16_cp(&sib, vp, 10, NULL
2703 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2704 ) != n_IDEC_STATE_CONSUMED || sib < 0){
2705 n_err(_("`shift': invalid argument: %s\n"), vp);
2706 goto jleave;
2708 i = (ui16_t)sib;
2711 /* If in in a macro/xy */
2712 if(a_amv_lopts != NULL){
2713 struct a_amv_mac const *amp;
2714 struct a_amv_mac_call_args *amcap;
2716 /* Explicitly do allow `vpospar' created things! */
2717 amp = (amcap = a_amv_lopts->as_amcap)->amca_amp;
2718 if((amp == NULL || amcap->amca_ps_hook_mask ||
2719 (amp->am_flags & a_AMV_MF_TYPE_MASK) == a_AMV_MF_ACCOUNT) &&
2720 amcap->amca_pospar.app_not_heap){
2721 n_err(_("Cannot use `shift' in `account's or hook macros etc.\n"));
2722 goto jleave;
2724 appp = &amcap->amca_pospar;
2725 }else
2726 appp = &a_amv_pospar;
2728 if(i > appp->app_count){
2729 n_err(_("`shift': cannot shift %hu of %hu parameters\n"),
2730 i, appp->app_count);
2731 goto jleave;
2732 }else{
2733 appp->app_idx += i;
2734 appp->app_count -= i;
2735 rv = 0;
2737 jleave:
2738 NYD_LEAVE;
2739 return rv;
2742 FL int
2743 c_return(void *vp){ /* TODO the exit status should be m_si64! */
2744 int rv;
2745 NYD_ENTER;
2747 if(a_amv_lopts != NULL){
2748 char const **argv;
2750 n_go_input_force_eof();
2751 n_pstate_err_no = n_ERR_NONE;
2752 rv = 0;
2754 if((argv = vp)[0] != NULL){
2755 si32_t i;
2757 if((n_idec_si32_cp(&i, argv[0], 10, NULL
2758 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2759 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2760 rv = (int)i;
2761 else{
2762 n_err(_("`return': return value argument is invalid: %s\n"),
2763 argv[0]);
2764 n_pstate_err_no = n_ERR_INVAL;
2765 rv = 1;
2768 if(argv[1] != NULL){
2769 if((n_idec_si32_cp(&i, argv[1], 10, NULL
2770 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
2771 ) == n_IDEC_STATE_CONSUMED && i >= 0)
2772 n_pstate_err_no = i;
2773 else{
2774 n_err(_("`return': error number argument is invalid: %s\n"),
2775 argv[1]);
2776 n_pstate_err_no = n_ERR_INVAL;
2777 rv = 1;
2781 }else{
2782 n_err(_("Can only use `return' in a macro\n"));
2783 n_pstate_err_no = n_ERR_OPNOTSUPP;
2784 rv = 1;
2786 NYD_LEAVE;
2787 return rv;
2790 FL bool_t
2791 temporary_folder_hook_check(bool_t nmail){ /* TODO temporary, v15: drop */
2792 struct a_amv_mac_call_args *amcap;
2793 struct a_amv_mac *amp;
2794 size_t len;
2795 char const *cp;
2796 char *var;
2797 bool_t rv;
2798 NYD_ENTER;
2800 rv = TRU1;
2801 var = n_autorec_alloc(len = strlen(mailname) + sizeof("folder-hook-") -1 +1);
2803 /* First try the fully resolved path */
2804 snprintf(var, len, "folder-hook-%s", mailname);
2805 if((cp = n_var_vlook(var, FAL0)) != NULL)
2806 goto jmac;
2808 /* If we are under *folder*, try the usual +NAME syntax, too */
2809 if(displayname[0] == '+'){
2810 char *x;
2812 for(x = &mailname[len]; x != mailname; --x)
2813 if(x[-1] == '/'){
2814 snprintf(var, len, "folder-hook-+%s", x);
2815 if((cp = n_var_vlook(var, FAL0)) != NULL)
2816 goto jmac;
2817 break;
2821 /* Plain *folder-hook* is our last try */
2822 if((cp = ok_vlook(folder_hook)) == NULL)
2823 goto jleave;
2825 jmac:
2826 if((amp = a_amv_mac_lookup(cp, NULL, a_AMV_MF_NONE)) == NULL){
2827 n_err(_("Cannot call *folder-hook* for %s: macro does not exist: %s\n"),
2828 n_shexp_quote_cp(displayname, FAL0), cp);
2829 rv = FAL0;
2830 goto jleave;
2833 amcap = n_lofi_alloc(sizeof *amcap);
2834 memset(amcap, 0, sizeof *amcap);
2835 amcap->amca_name = cp;
2836 amcap->amca_amp = amp;
2837 n_pstate &= ~n_PS_HOOK_MASK;
2838 if(nmail){
2839 amcap->amca_unroller = NULL;
2840 n_pstate |= n_PS_HOOK_NEWMAIL;
2841 }else{
2842 amcap->amca_unroller = &a_amv_folder_hook_lopts;
2843 n_pstate |= n_PS_HOOK;
2845 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2846 amcap->amca_ps_hook_mask = TRU1;
2847 amcap->amca_no_xcall = TRU1;
2848 rv = a_amv_mac_exec(amcap);
2849 n_pstate &= ~n_PS_HOOK_MASK;
2851 jleave:
2852 NYD_LEAVE;
2853 return rv;
2856 FL void
2857 temporary_folder_hook_unroll(void){ /* XXX intermediate hack */
2858 NYD_ENTER;
2859 if(a_amv_folder_hook_lopts != NULL){
2860 void *save = a_amv_lopts;
2862 a_amv_lopts = NULL;
2863 a_amv_lopts_unroll(&a_amv_folder_hook_lopts);
2864 assert(a_amv_folder_hook_lopts == NULL);
2865 a_amv_lopts = save;
2867 NYD_LEAVE;
2870 FL void
2871 temporary_compose_mode_hook_call(char const *macname,
2872 void (*hook_pre)(void *), void *hook_arg){
2873 /* TODO compose_mode_hook_call() temporary, v15: generalize; see a_GO_SPLICE
2874 * TODO comment in go.c for the right way of doing things! */
2875 static struct a_amv_lostack *cmh_losp;
2876 struct a_amv_mac_call_args *amcap;
2877 struct a_amv_mac *amp;
2878 NYD_ENTER;
2880 amp = NULL;
2882 if(macname == (char*)-1){
2883 a_amv_mac__finalize(cmh_losp);
2884 cmh_losp = NULL;
2885 }else if(macname != NULL &&
2886 (amp = a_amv_mac_lookup(macname, NULL, a_AMV_MF_NONE)) == NULL)
2887 n_err(_("Cannot call *on-compose-**: macro does not exist: %s\n"),
2888 macname);
2889 else{
2890 amcap = n_lofi_alloc(sizeof *amcap);
2891 memset(amcap, 0, sizeof *amcap);
2892 amcap->amca_name = (macname != NULL) ? macname
2893 : "*on-compose-splice(-shell)?*";
2894 amcap->amca_amp = amp;
2895 amcap->amca_unroller = &a_amv_compose_lopts;
2896 amcap->amca_hook_pre = hook_pre;
2897 amcap->amca_hook_arg = hook_arg;
2898 amcap->amca_loflags = a_AMV_LF_SCOPE_FIXATE;
2899 amcap->amca_ps_hook_mask = TRU1;
2900 amcap->amca_no_xcall = TRU1;
2901 n_pstate &= ~n_PS_HOOK_MASK;
2902 n_pstate |= n_PS_HOOK;
2903 if(macname != NULL)
2904 a_amv_mac_exec(amcap);
2905 else{
2906 cmh_losp = n_lofi_alloc(sizeof *cmh_losp);
2907 memset(cmh_losp, 0, sizeof *cmh_losp);
2908 cmh_losp->as_global_saved = a_amv_lopts;
2909 cmh_losp->as_lopts = *(cmh_losp->as_amcap = amcap)->amca_unroller;
2910 cmh_losp->as_loflags = a_AMV_LF_SCOPE_FIXATE;
2911 a_amv_lopts = cmh_losp;
2914 NYD_LEAVE;
2917 FL void
2918 temporary_compose_mode_hook_unroll(void){ /* XXX intermediate hack */
2919 NYD_ENTER;
2920 if(a_amv_compose_lopts != NULL){
2921 void *save = a_amv_lopts;
2923 a_amv_lopts = NULL;
2924 a_amv_lopts_unroll(&a_amv_compose_lopts);
2925 assert(a_amv_compose_lopts == NULL);
2926 a_amv_lopts = save;
2928 NYD_LEAVE;
2931 FL bool_t
2932 n_var_is_user_writable(char const *name){
2933 struct a_amv_var_carrier avc;
2934 struct a_amv_var_map const *avmp;
2935 bool_t rv;
2936 NYD_ENTER;
2938 a_amv_var_revlookup(&avc, name, TRU1);
2939 if((avmp = avc.avc_map) == NULL)
2940 rv = TRU1;
2941 else
2942 rv = ((avmp->avm_flags & (a_AMV_VF_BOOL | a_AMV_VF_RDONLY)) == 0);
2943 NYD_LEAVE;
2944 return rv;
2947 FL char *
2948 n_var_oklook(enum okeys okey){
2949 struct a_amv_var_carrier avc;
2950 char *rv;
2951 struct a_amv_var_map const *avmp;
2952 NYD_ENTER;
2954 memset(&avc, 0, sizeof avc);
2955 avc.avc_map = avmp = &a_amv_var_map[okey];
2956 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2957 avc.avc_hash = avmp->avm_hash;
2958 avc.avc_okey = okey;
2960 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE))
2961 rv = avc.avc_var->av_value;
2962 else
2963 rv = NULL;
2964 NYD_LEAVE;
2965 return rv;
2968 FL bool_t
2969 n_var_okset(enum okeys okey, uintptr_t val){
2970 struct a_amv_var_carrier avc;
2971 bool_t ok;
2972 struct a_amv_var_map const *avmp;
2973 NYD_ENTER;
2975 memset(&avc, 0, sizeof avc);
2976 avc.avc_map = avmp = &a_amv_var_map[okey];
2977 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2978 avc.avc_hash = avmp->avm_hash;
2979 avc.avc_okey = okey;
2981 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
2982 a_AMV_VSETCLR_NONE);
2983 NYD_LEAVE;
2984 return ok;
2987 FL bool_t
2988 n_var_okclear(enum okeys okey){
2989 struct a_amv_var_carrier avc;
2990 bool_t rv;
2991 struct a_amv_var_map const *avmp;
2992 NYD_ENTER;
2994 memset(&avc, 0, sizeof avc);
2995 avc.avc_map = avmp = &a_amv_var_map[okey];
2996 avc.avc_name = &a_amv_var_names[avmp->avm_keyoff];
2997 avc.avc_hash = avmp->avm_hash;
2998 avc.avc_okey = okey;
3000 rv = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
3001 NYD_LEAVE;
3002 return rv;
3005 FL char const *
3006 n_var_vlook(char const *vokey, bool_t try_getenv){
3007 struct a_amv_var_carrier avc;
3008 char const *rv;
3009 NYD_ENTER;
3011 a_amv_var_revlookup(&avc, vokey, FAL0);
3013 switch((enum a_amv_var_special_category)avc.avc_special_cat){
3014 default: /* silence CC */
3015 case a_AMV_VSC_NONE:
3016 rv = NULL;
3017 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_LOCAL))
3018 rv = avc.avc_var->av_value;
3019 /* Only check the environment for something that is otherwise unknown */
3020 else if(try_getenv && avc.avc_map == NULL &&
3021 !a_amv_var_revlookup_chain(&avc, vokey))
3022 rv = getenv(vokey);
3023 break;
3024 case a_AMV_VSC_GLOBAL:
3025 rv = a_amv_var_vsc_global(&avc);
3026 break;
3027 case a_AMV_VSC_MULTIPLEX:
3028 rv = a_amv_var_vsc_multiplex(&avc);
3029 break;
3030 case a_AMV_VSC_POSPAR:
3031 case a_AMV_VSC_POSPAR_ENV:
3032 rv = a_amv_var_vsc_pospar(&avc);
3033 break;
3035 NYD_LEAVE;
3036 return rv;
3039 FL bool_t
3040 n_var_vexplode(void const **cookie){
3041 struct a_amv_pospar *appp;
3042 NYD_ENTER;
3044 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
3045 : &a_amv_pospar;
3046 *cookie = (appp->app_count > 0) ? &appp->app_dat[appp->app_idx] : NULL;
3047 NYD_LEAVE;
3048 return (*cookie != NULL);
3051 FL bool_t
3052 n_var_vset(char const *vokey, uintptr_t val){
3053 struct a_amv_var_carrier avc;
3054 bool_t ok;
3055 NYD_ENTER;
3057 a_amv_var_revlookup(&avc, vokey, TRU1);
3059 ok = a_amv_var_set(&avc, (val == 0x1 ? n_empty : (char const*)val),
3060 a_AMV_VSETCLR_NONE);
3061 NYD_LEAVE;
3062 return ok;
3065 FL bool_t
3066 n_var_vclear(char const *vokey){
3067 struct a_amv_var_carrier avc;
3068 bool_t ok;
3069 NYD_ENTER;
3071 a_amv_var_revlookup(&avc, vokey, FAL0);
3073 ok = a_amv_var_clear(&avc, a_AMV_VSETCLR_NONE);
3074 NYD_LEAVE;
3075 return ok;
3078 #ifdef HAVE_SOCKETS
3079 FL char *
3080 n_var_xoklook(enum okeys okey, struct url const *urlp,
3081 enum okey_xlook_mode oxm){
3082 struct a_amv_var_carrier avc;
3083 struct str const *us;
3084 size_t nlen;
3085 char *nbuf, *rv;
3086 NYD_ENTER;
3088 assert(oxm & (OXM_PLAIN | OXM_H_P | OXM_U_H_P));
3090 /* For simplicity: allow this case too */
3091 if(!(oxm & (OXM_H_P | OXM_U_H_P))){
3092 nbuf = NULL;
3093 goto jplain;
3096 memset(&avc, 0, sizeof avc);
3097 avc.avc_name = &a_amv_var_names[(avc.avc_map = &a_amv_var_map[okey]
3098 )->avm_keyoff];
3099 avc.avc_okey = okey;
3100 avc.avc_is_chain_variant = TRU1;
3102 us = (oxm & OXM_U_H_P) ? &urlp->url_u_h_p : &urlp->url_h_p;
3103 nlen = strlen(avc.avc_name);
3104 nbuf = n_lofi_alloc(nlen + 1 + us->l +1);
3105 memcpy(nbuf, avc.avc_name, nlen);
3107 /* One of .url_u_h_p and .url_h_p we test in here */
3108 nbuf[nlen++] = '-';
3109 memcpy(&nbuf[nlen], us->s, us->l +1);
3110 avc.avc_name = nbuf;
3111 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
3112 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE))
3113 goto jvar;
3115 /* The second */
3116 if((oxm & (OXM_U_H_P | OXM_H_P)) == (OXM_U_H_P | OXM_H_P)){
3117 us = &urlp->url_h_p;
3118 memcpy(&nbuf[nlen], us->s, us->l +1);
3119 avc.avc_name = nbuf;
3120 avc.avc_hash = a_AMV_NAME2HASH(avc.avc_name);
3121 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE)){
3122 jvar:
3123 rv = avc.avc_var->av_value;
3124 goto jleave;
3128 jplain:
3129 /*avc.avc_is_chain_variant = FAL0;*/
3130 rv = (oxm & OXM_PLAIN) ? n_var_oklook(okey) : NULL;
3131 jleave:
3132 if(nbuf != NULL)
3133 n_lofi_free(nbuf);
3134 NYD_LEAVE;
3135 return rv;
3137 #endif /* HAVE_SOCKETS */
3139 FL int
3140 c_set(void *vp){
3141 int err;
3142 char **ap;
3143 NYD_ENTER;
3145 if(*(ap = vp) == NULL){
3146 a_amv_var_show_all();
3147 err = 0;
3148 }else{
3149 enum a_amv_var_setclr_flags avscf;
3151 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
3152 avscf = a_AMV_VSETCLR_NONE;
3153 else{
3154 if(a_amv_lopts == NULL){
3155 n_err(_("`set': cannot use `local' in this context\n"));
3156 err = 1;
3157 goto jleave;
3159 avscf = a_AMV_VSETCLR_LOCAL;
3161 err = !a_amv_var_c_set(ap, avscf);
3163 jleave:
3164 NYD_LEAVE;
3165 return err;
3168 FL int
3169 c_unset(void *vp){
3170 struct a_amv_var_carrier avc;
3171 char **ap;
3172 int err;
3173 enum a_amv_var_setclr_flags avscf;
3174 NYD_ENTER;
3176 if(!(n_pstate & n_PS_ARGMOD_LOCAL))
3177 avscf = a_AMV_VSETCLR_NONE;
3178 else{
3179 if(a_amv_lopts == NULL){
3180 n_err(_("`unset': cannot use `local' in this context\n"));
3181 err = 1;
3182 goto jleave;
3184 avscf = a_AMV_VSETCLR_LOCAL;
3187 for(err = 0, ap = vp; *ap != NULL; ++ap){
3188 a_amv_var_revlookup(&avc, *ap, FAL0);
3190 err |= !a_amv_var_clear(&avc, avscf);
3192 jleave:
3193 NYD_LEAVE;
3194 return err;
3197 FL int
3198 c_varshow(void *v){
3199 char **ap;
3200 NYD_ENTER;
3202 if(*(ap = v) == NULL)
3203 v = NULL;
3204 else{
3205 struct n_string msg, *msgp = &msg;
3207 msgp = n_string_creat(msgp);
3208 for(; *ap != NULL; ++ap)
3209 a_amv_var_show(*ap, n_stdout, msgp);
3210 n_string_gut(msgp);
3212 NYD_LEAVE;
3213 return (v == NULL ? !STOP : !OKAY); /* xxx 1:bad 0:good -- do some */
3216 FL int
3217 c_varedit(void *v){ /* TODO v15 drop */
3218 struct a_amv_var_carrier avc;
3219 FILE *of, *nf;
3220 char *val, **argv;
3221 int err;
3222 sighandler_type sigint;
3223 NYD_ENTER;
3225 sigint = safe_signal(SIGINT, SIG_IGN);
3227 for(err = 0, argv = v; *argv != NULL; ++argv){
3228 a_amv_var_revlookup(&avc, *argv, TRU1);
3230 if(avc.avc_map != NULL){
3231 if(avc.avc_map->avm_flags & a_AMV_VF_BOOL){
3232 n_err(_("`varedit': cannot edit boolean variable: %s\n"),
3233 avc.avc_name);
3234 continue;
3236 if(avc.avc_map->avm_flags & a_AMV_VF_RDONLY){
3237 n_err(_("`varedit': cannot edit read-only variable: %s\n"),
3238 avc.avc_name);
3239 continue;
3243 a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE);
3245 if((of = Ftmp(NULL, "varedit", OF_RDWR | OF_UNLINK | OF_REGISTER)) ==
3246 NULL){
3247 n_perr(_("`varedit': cannot create temporary file"), 0);
3248 err = 1;
3249 break;
3250 }else if(avc.avc_var != NULL && *(val = avc.avc_var->av_value) != '\0' &&
3251 sizeof *val != fwrite(val, strlen(val), sizeof *val, of)){
3252 n_perr(_("`varedit' failed to write old value to temporary file"), 0);
3253 Fclose(of);
3254 err = 1;
3255 continue;
3258 fflush_rewind(of);
3259 nf = n_run_editor(of, (off_t)-1, 'e', FAL0, NULL,NULL, SEND_MBOX, sigint,
3260 NULL);
3261 Fclose(of);
3263 if(nf != NULL){
3264 int c;
3265 char const *varres;
3266 off_t l;
3268 l = fsize(nf);
3269 if(UICMP(64, l, >=, UIZ_MAX -42)){
3270 n_err(_("`varedit': not enough memory to store variable: %s\n"),
3271 avc.avc_name);
3272 varres = n_empty;
3273 err = 1;
3274 }else{
3275 varres = val = n_autorec_alloc(l +1);
3276 for(; l > 0 && (c = getc(nf)) != EOF; --l)
3277 *val++ = c;
3278 *val++ = '\0';
3279 if(l != 0){
3280 n_err(_("`varedit': I/O while reading new value of: %s\n"),
3281 avc.avc_name);
3282 err = 1;
3286 if(!a_amv_var_set(&avc, varres, a_AMV_VSETCLR_NONE))
3287 err = 1;
3289 Fclose(nf);
3290 }else{
3291 n_err(_("`varedit': cannot start $EDITOR\n"));
3292 err = 1;
3293 break;
3297 safe_signal(SIGINT, sigint);
3298 NYD_LEAVE;
3299 return err;
3302 FL int
3303 c_environ(void *v){
3304 struct a_amv_var_carrier avc;
3305 int err;
3306 char **ap;
3307 bool_t islnk;
3308 NYD_ENTER;
3310 if((islnk = is_asccaseprefix(*(ap = v), "link")) ||
3311 is_asccaseprefix(*ap, "unlink")){
3312 for(err = 0; *++ap != NULL;){
3313 a_amv_var_revlookup(&avc, *ap, TRU1);
3315 if(a_amv_var_lookup(&avc, a_AMV_VLOOK_NONE) && (islnk ||
3316 (avc.avc_var->av_flags & a_AMV_VF_EXT_LINKED))){
3317 if(!islnk){
3318 avc.avc_var->av_flags &= ~a_AMV_VF_EXT_LINKED;
3319 continue;
3320 }else if(avc.avc_var->av_flags &
3321 (a_AMV_VF_ENV | a_AMV_VF_EXT_LINKED)){
3322 if(n_poption & n_PO_D_V)
3323 n_err(_("`environ': link: already established: %s\n"), *ap);
3324 continue;
3326 avc.avc_var->av_flags |= a_AMV_VF_EXT_LINKED;
3327 if(!(avc.avc_var->av_flags & a_AMV_VF_ENV))
3328 a_amv_var__putenv(&avc, avc.avc_var);
3329 }else if(!islnk){
3330 n_err(_("`environ': unlink: no link established: %s\n"), *ap);
3331 err = 1;
3332 }else{
3333 char const *evp = getenv(*ap);
3335 if(evp != NULL)
3336 err |= !a_amv_var_set(&avc, evp, a_AMV_VSETCLR_ENV);
3337 else{
3338 n_err(_("`environ': link: cannot link to non-existent: %s\n"),
3339 *ap);
3340 err = 1;
3344 }else if(is_asccaseprefix(*ap, "set"))
3345 err = !a_amv_var_c_set(++ap, a_AMV_VSETCLR_ENV);
3346 else if(is_asccaseprefix(*ap, "unset")){
3347 for(err = 0; *++ap != NULL;){
3348 a_amv_var_revlookup(&avc, *ap, FAL0);
3350 if(!a_amv_var_clear(&avc, a_AMV_VSETCLR_ENV))
3351 err = 1;
3353 }else{
3354 n_err(_("Synopsis: environ: <link|set|unset> <variable>...\n"));
3355 err = 1;
3357 NYD_LEAVE;
3358 return err;
3361 FL int
3362 c_vexpr(void *v){ /* TODO POSIX expr(1) comp. exit status; overly complicat. */
3363 char pbase, op, iencbuf[2+1/* BASE# prefix*/ + n_IENC_BUFFER_SIZE + 1];
3364 size_t i;
3365 enum n_idec_state ids;
3366 enum n_idec_mode idm;
3367 si64_t lhv, rhv;
3368 char const **argv, *varname, *varres, *cp;
3369 enum{
3370 a_ERR = 1u<<0,
3371 a_SOFTOVERFLOW = 1u<<1,
3372 a_ISNUM = 1u<<2,
3373 a_ISDECIMAL = 1u<<3, /* Print only decimal result */
3374 a_SATURATED = 1u<<4,
3375 a_ICASE = 1u<<5,
3376 a_UNSIGNED_OP = 1u<<6, /* Unsigned right shift (share bit ok) */
3377 a_PBASE = 1u<<7, /* Print additional number base */
3378 a_TMP = 1u<<30
3379 } f;
3380 NYD_ENTER;
3382 f = a_ERR;
3383 argv = v;
3384 varname = (n_pstate & n_PS_ARGMOD_VPUT) ? *argv++ : NULL;
3385 n_UNINIT(varres, n_empty);
3386 n_UNINIT(pbase, '\0');
3388 if((cp = argv[0])[0] == '\0')
3389 goto jesubcmd;
3391 if(cp[1] == '\0'){
3392 op = cp[0];
3393 jnumop:
3394 f |= a_ISNUM;
3395 switch(op){
3396 case '=':
3397 case '~':
3398 if(argv[1] == NULL || argv[2] != NULL)
3399 goto jesynopsis;
3401 if(*(cp = *++argv) == '\0')
3402 lhv = 0;
3403 else{
3404 idm = ((*cp == 'u' || *cp == 'U')
3405 ? (++cp, n_IDEC_MODE_NONE)
3406 : ((*cp == 's' || *cp == 'S')
3407 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3408 : n_IDEC_MODE_SIGNED_TYPE |
3409 n_IDEC_MODE_POW2BASE_UNSIGNED));
3410 if(((ids = n_idec_cp(&lhv, cp, 0, idm, NULL)
3411 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3412 ) != n_IDEC_STATE_CONSUMED){
3413 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3414 goto jenum_range;
3415 f |= a_SOFTOVERFLOW;
3416 break;
3419 if(op == '~')
3420 lhv = ~lhv;
3421 break;
3423 case '+':
3424 case '-':
3425 case '*':
3426 case '/':
3427 case '%':
3428 case '|':
3429 case '&':
3430 case '^':
3431 case '<':
3432 case '>':
3433 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3434 goto jesynopsis;
3435 else{
3436 char xop;
3438 if(*(cp = *++argv) == '\0')
3439 lhv = 0;
3440 else{
3441 idm = ((*cp == 'u' || *cp == 'U')
3442 ? (++cp, n_IDEC_MODE_NONE)
3443 : ((*cp == 's' || *cp == 'S')
3444 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3445 : n_IDEC_MODE_SIGNED_TYPE |
3446 n_IDEC_MODE_POW2BASE_UNSIGNED));
3447 if(((ids = n_idec_cp(&lhv, cp, 0, idm, NULL)
3448 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3449 ) != n_IDEC_STATE_CONSUMED){
3450 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3451 goto jenum_range;
3452 f |= a_SOFTOVERFLOW;
3453 break;
3457 if(*(cp = *++argv) == '\0')
3458 rhv = 0;
3459 else{
3460 idm = ((*cp == 'u' || *cp == 'U')
3461 ? (++cp, n_IDEC_MODE_NONE)
3462 : ((*cp == 's' || *cp == 'S')
3463 ? (++cp, n_IDEC_MODE_SIGNED_TYPE)
3464 : n_IDEC_MODE_SIGNED_TYPE |
3465 n_IDEC_MODE_POW2BASE_UNSIGNED));
3466 if(((ids = n_idec_cp(&rhv, cp, 0, idm, NULL)
3467 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3468 ) != n_IDEC_STATE_CONSUMED){
3469 if(!(ids & n_IDEC_STATE_EOVERFLOW) || !(f & a_SATURATED))
3470 goto jenum_range;
3471 f |= a_SOFTOVERFLOW;
3472 lhv = rhv;
3473 break;
3477 xop = op;
3478 jnumop_again:
3479 switch(xop){
3480 case '+':
3481 if(rhv < 0){
3482 if(rhv != SI64_MIN){
3483 rhv = -rhv;
3484 xop = '-';
3485 goto jnumop_again;
3486 }else if(lhv < 0)
3487 goto jenum_plusminus;
3488 else if(lhv == 0){
3489 lhv = rhv;
3490 break;
3492 }else if(SI64_MAX - rhv < lhv)
3493 goto jenum_plusminus;
3494 lhv += rhv;
3495 break;
3496 case '-':
3497 if(rhv < 0){
3498 if(rhv != SI64_MIN){
3499 rhv = -rhv;
3500 xop = '+';
3501 goto jnumop_again;
3502 }else if(lhv > 0)
3503 goto jenum_plusminus;
3504 else if(lhv == 0){
3505 lhv = rhv;
3506 break;
3508 }else if(SI64_MIN + rhv > lhv){
3509 jenum_plusminus:
3510 if(!(f & a_SATURATED))
3511 goto jenum_overflow;
3512 f |= a_SOFTOVERFLOW;
3513 lhv = (lhv < 0 || xop == '-') ? SI64_MIN : SI64_MAX;
3514 break;
3516 lhv -= rhv;
3517 break;
3518 case '*':
3519 /* Will the result be positive? */
3520 if((lhv < 0) == (rhv < 0)){
3521 if(lhv > 0){
3522 lhv = -lhv;
3523 rhv = -rhv;
3525 if(rhv != 0 && lhv != 0 && SI64_MAX / rhv > lhv){
3526 if(!(f & a_SATURATED))
3527 goto jenum_overflow;
3528 f |= a_SOFTOVERFLOW;
3529 lhv = SI64_MAX;
3530 }else
3531 lhv *= rhv;
3532 }else{
3533 if(rhv > 0){
3534 if(lhv != 0 && SI64_MIN / lhv < rhv){
3535 if(!(f & a_SATURATED))
3536 goto jenum_overflow;
3537 f |= a_SOFTOVERFLOW;
3538 lhv = SI64_MIN;
3539 }else
3540 lhv *= rhv;
3541 }else{
3542 if(rhv != 0 && lhv != 0 && SI64_MIN / rhv < lhv){
3543 if(!(f & a_SATURATED))
3544 goto jenum_overflow;
3545 f |= a_SOFTOVERFLOW;
3546 lhv = SI64_MIN;
3547 }else
3548 lhv *= rhv;
3551 break;
3552 case '/':
3553 if(rhv == 0){
3554 if(!(f & a_SATURATED))
3555 goto jenum_range;
3556 f |= a_SOFTOVERFLOW;
3557 lhv = SI64_MAX;
3558 }else
3559 lhv /= rhv;
3560 break;
3561 case '%':
3562 if(rhv == 0){
3563 if(!(f & a_SATURATED))
3564 goto jenum_range;
3565 f |= a_SOFTOVERFLOW;
3566 lhv = SI64_MAX;
3567 }else
3568 lhv %= rhv;
3569 break;
3570 case '|':
3571 lhv |= rhv;
3572 break;
3573 case '&':
3574 lhv &= rhv;
3575 break;
3576 case '^':
3577 lhv ^= rhv;
3578 break;
3579 case '<':
3580 case '>':
3581 if(!(f & a_TMP)){
3582 argv -= 2;
3583 goto jesubcmd;
3585 if(rhv > 63){ /* xxx 63? */
3586 if(!(f & a_SATURATED))
3587 goto jenum_overflow;
3588 rhv = 63;
3590 if(op == '<')
3591 lhv <<= (ui8_t)rhv;
3592 else if(f & a_UNSIGNED_OP)
3593 lhv = (ui64_t)lhv >> (ui8_t)rhv;
3594 else
3595 lhv >>= (ui8_t)rhv;
3596 break;
3599 break;
3600 default:
3601 goto jesubcmd;
3603 }else if(cp[0] == '<'){
3604 if(*++cp != '<')
3605 goto jesubcmd;
3606 if(*++cp == '@'){
3607 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3608 f |= a_SATURATED;
3609 ++cp;
3611 if(*cp != '\0')
3612 goto jesubcmd;
3613 f |= a_TMP;
3614 op = '<';
3615 goto jnumop;
3616 }else if(cp[0] == '>'){
3617 if(*++cp != '>')
3618 goto jesubcmd;
3619 if(*++cp == '>'){
3620 f |= a_UNSIGNED_OP;
3621 ++cp;
3623 if(*cp == '@'){
3624 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix not suffix"));
3625 f |= a_SATURATED;
3626 ++cp;
3628 if(*cp != '\0')
3629 goto jesubcmd;
3630 f |= a_TMP;
3631 op = '>';
3632 goto jnumop;
3633 }else if(cp[2] == '\0' && cp[1] == '@'){
3634 n_OBSOLETE(_("`vexpr': please use @ modifier as prefix, not suffix"));
3635 f |= a_SATURATED;
3636 op = cp[0];
3637 goto jnumop;
3638 }else if(cp[0] == '@'){
3639 f |= a_SATURATED;
3640 op = *++cp;
3641 if(*++cp != '\0'){
3642 if(op != *cp)
3643 goto jesubcmd;
3644 switch(op){
3645 case '<':
3646 if(*++cp != '\0')
3647 goto jesubcmd;
3648 f |= a_TMP;
3649 break;
3650 case '>':
3651 if(*++cp != '\0'){
3652 if(*cp != '>' || *++cp != '\0')
3653 goto jesubcmd;
3654 f |= a_UNSIGNED_OP;
3656 f |= a_TMP;
3657 break;
3658 default:
3659 goto jesubcmd;
3662 goto jnumop;
3663 }else if(is_asccaseprefix(cp, "pbase")){
3664 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3665 goto jesynopsis;
3667 if(((ids = n_idec_si8_cp(&pbase, argv[1], 0, NULL)
3668 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3669 ) != n_IDEC_STATE_CONSUMED || pbase < 2 || pbase > 36)
3670 goto jenum_range;
3672 f |= a_PBASE;
3673 op = *(argv[0] = cp = "=");
3674 argv[1] = argv[2];
3675 argv[2] = NULL;
3676 goto jnumop;
3677 }else if(is_asccaseprefix(cp, "length")){
3678 f |= a_ISNUM | a_ISDECIMAL;
3679 if(argv[1] == NULL || argv[2] != NULL)
3680 goto jesynopsis;
3682 i = strlen(*++argv);
3683 if(UICMP(64, i, >, SI64_MAX))
3684 goto jestr_overflow;
3685 lhv = (si64_t)i;
3686 }else if(is_asccaseprefix(cp, "hash")){
3687 f |= a_ISNUM | a_ISDECIMAL;
3688 if(argv[1] == NULL || argv[2] != NULL)
3689 goto jesynopsis;
3691 i = n_torek_hash(*++argv);
3692 lhv = (si64_t)i;
3693 }else if(is_asccaseprefix(cp, "find")){
3694 f |= a_ISNUM | a_ISDECIMAL;
3695 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3696 goto jesynopsis;
3698 if((cp = strstr(argv[1], argv[2])) == NULL)
3699 goto jestr_nodata;
3700 i = PTR2SIZE(cp - argv[1]);
3701 if(UICMP(64, i, >, SI64_MAX))
3702 goto jestr_overflow;
3703 lhv = (si64_t)i;
3704 }else if(is_asccaseprefix(cp, "ifind")){
3705 f |= a_ISNUM | a_ISDECIMAL;
3706 if(argv[1] == NULL || argv[2] == NULL || argv[3] != NULL)
3707 goto jesynopsis;
3709 if((cp = asccasestr(argv[1], argv[2])) == NULL)
3710 goto jestr_nodata;
3711 i = PTR2SIZE(cp - argv[1]);
3712 if(UICMP(64, i, >, SI64_MAX))
3713 goto jestr_overflow;
3714 lhv = (si64_t)i;
3715 }else if(is_asccaseprefix(cp, "substring")){
3716 if(argv[1] == NULL || argv[2] == NULL)
3717 goto jesynopsis;
3718 if(argv[3] != NULL && argv[4] != NULL)
3719 goto jesynopsis;
3721 i = strlen(varres = *++argv);
3723 if(*(cp = *++argv) == '\0')
3724 lhv = 0;
3725 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3726 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3727 ) != n_IDEC_STATE_CONSUMED)
3728 goto jestr_numrange;
3729 if(lhv < 0){
3730 if(UICMP(64, i, <, -lhv))
3731 goto jesubstring_off;
3732 lhv = i + lhv;
3734 if(UICMP(64, i, >=, lhv)){
3735 i -= lhv;
3736 varres += lhv;
3737 }else{
3738 jesubstring_off:
3739 if(n_poption & n_PO_D_V)
3740 n_err(_("`vexpr': substring: offset argument too large: %s\n"),
3741 n_shexp_quote_cp(argv[-1], FAL0));
3742 f |= a_SOFTOVERFLOW;
3745 if(argv[1] != NULL){
3746 if(*(cp = *++argv) == '\0')
3747 lhv = 0;
3748 else if((n_idec_si64_cp(&lhv, cp, 0, NULL
3749 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3750 ) != n_IDEC_STATE_CONSUMED)
3751 goto jestr_numrange;
3752 if(lhv < 0){
3753 if(UICMP(64, i, <, -lhv))
3754 goto jesubstring_len;
3755 lhv = i + lhv;
3757 if(UICMP(64, i, >=, lhv)){
3758 if(UICMP(64, i, !=, lhv))
3759 varres = savestrbuf(varres, (size_t)lhv);
3760 }else{
3761 jesubstring_len:
3762 if(n_poption & n_PO_D_V)
3763 n_err(_("`vexpr': substring: length argument too large: %s\n"),
3764 n_shexp_quote_cp(argv[-2], FAL0));
3765 f |= a_SOFTOVERFLOW;
3768 }else if(is_asccaseprefix(cp, "trim")) Jtrim: {
3769 struct str trim;
3770 enum n_str_trim_flags stf;
3772 if(argv[1] == NULL || argv[2] != NULL)
3773 goto jesynopsis;
3775 if(is_asccaseprefix(cp, "trim-front"))
3776 stf = n_STR_TRIM_FRONT;
3777 else if(is_asccaseprefix(cp, "trim-end"))
3778 stf = n_STR_TRIM_END;
3779 else
3780 stf = n_STR_TRIM_BOTH;
3782 trim.l = strlen(trim.s = n_UNCONST(argv[1]));
3783 (void)n_str_trim(&trim, stf);
3784 varres = savestrbuf(trim.s, trim.l);
3785 }else if(is_asccaseprefix(cp, "trim-front"))
3786 goto Jtrim;
3787 else if(is_asccaseprefix(cp, "trim-end"))
3788 goto Jtrim;
3789 else if(is_asccaseprefix(cp, "random")){
3790 if(argv[1] == NULL || argv[2] != NULL)
3791 goto jesynopsis;
3793 if((n_idec_si64_cp(&lhv, argv[1], 0, NULL
3794 ) & (n_IDEC_STATE_EMASK | n_IDEC_STATE_CONSUMED)
3795 ) != n_IDEC_STATE_CONSUMED || lhv < 0 || lhv > PATH_MAX)
3796 goto jestr_numrange;
3797 if(lhv == 0)
3798 lhv = NAME_MAX;
3799 varres = n_random_create_cp((size_t)lhv, NULL);
3800 }else if(is_asccaseprefix(cp, "file-expand")){
3801 if(argv[1] == NULL || argv[2] != NULL)
3802 goto jesynopsis;
3804 if((varres = fexpand(argv[1], FEXP_NVAR | FEXP_NOPROTO)) == NULL)
3805 goto jestr_nodata;
3806 }else if(is_asccaseprefix(cp, "makeprint")){
3807 struct str sin, sout;
3809 if(argv[1] == NULL || argv[2] != NULL)
3810 goto jesynopsis;
3812 /* XXX using strlen for `vexpr makeprint' is wrong for UTF-16 */
3813 sin.l = strlen(sin.s = n_UNCONST(argv[1]));
3814 makeprint(&sin, &sout);
3815 varres = savestrbuf(sout.s, sout.l);
3816 n_free(sout.s);
3817 /* TODO `vexpr': (wide) string length, find, etc!! */
3818 #ifdef HAVE_REGEX
3819 }else if(is_asccaseprefix(cp, "regex")) Jregex:{
3820 regmatch_t rema[1 + n_VEXPR_REGEX_MAX];
3821 regex_t re;
3822 int reflrv;
3824 f |= a_ISNUM | a_ISDECIMAL;
3825 if(argv[1] == NULL || argv[2] == NULL ||
3826 (argv[3] != NULL && argv[4] != NULL))
3827 goto jesynopsis;
3829 reflrv = REG_EXTENDED;
3830 if(f & a_ICASE)
3831 reflrv |= REG_ICASE;
3832 if((reflrv = regcomp(&re, argv[2], reflrv))){
3833 n_err(_("`vexpr': invalid regular expression: %s: %s\n"),
3834 n_shexp_quote_cp(argv[2], FAL0), n_regex_err_to_doc(NULL, reflrv));
3835 assert(f & a_ERR);
3836 n_pstate_err_no = n_ERR_INVAL;
3837 goto jestr;
3839 reflrv = regexec(&re, argv[1], n_NELEM(rema), rema, 0);
3840 regfree(&re);
3841 if(reflrv == REG_NOMATCH)
3842 goto jestr_nodata;
3844 /* Search only? Else replace, which is a bit */
3845 if(argv[3] == NULL){
3846 if(UICMP(64, rema[0].rm_so, >, SI64_MAX))
3847 goto jestr_overflow;
3848 lhv = (si64_t)rema[0].rm_so;
3849 }else{
3850 /* We need to setup some kind of pseudo macro environment for this */
3851 struct a_amv_lostack los;
3852 struct a_amv_mac_call_args amca;
3853 char const **reargv;
3855 memset(&amca, 0, sizeof amca);
3856 amca.amca_name = savestrbuf(&argv[1][rema[0].rm_so],
3857 rema[0].rm_eo - rema[0].rm_so);
3858 amca.amca_amp = a_AMV_MACKY_MACK;
3859 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3861 amca.amca_pospar.app_count = (ui32_t)i - 1;
3862 amca.amca_pospar.app_not_heap = TRU1;
3863 amca.amca_pospar.app_dat =
3864 reargv = n_autorec_alloc(sizeof(char*) * i);
3865 for(i = 1; i < n_NELEM(rema) && rema[i].rm_so != -1; ++i)
3866 *reargv++ = savestrbuf(&argv[1][rema[i].rm_so],
3867 rema[i].rm_eo - rema[i].rm_so);
3868 *reargv = NULL;
3870 memset(&los, 0, sizeof los);
3871 hold_all_sigs(); /* TODO DISLIKE! */
3872 los.as_global_saved = a_amv_lopts;
3873 los.as_amcap = &amca;
3874 los.as_up = los.as_global_saved;
3875 a_amv_lopts = &los;
3877 /* C99 */{
3878 struct str templ;
3879 struct n_string s_b;
3880 enum n_shexp_state shs;
3882 templ.s = n_UNCONST(argv[3]);
3883 templ.l = UIZ_MAX;
3884 shs = n_shexp_parse_token((n_SHEXP_PARSE_LOG |
3885 n_SHEXP_PARSE_IGNORE_EMPTY | n_SHEXP_PARSE_QUOTE_AUTO_FIXED |
3886 n_SHEXP_PARSE_QUOTE_AUTO_DSQ),
3887 n_string_creat_auto(&s_b), &templ, NULL);
3888 if((shs & (n_SHEXP_STATE_ERR_MASK | n_SHEXP_STATE_STOP)
3889 ) == n_SHEXP_STATE_STOP){
3890 varres = n_string_cp(&s_b);
3891 n_string_drop_ownership(&s_b);
3892 }else
3893 varres = NULL;
3896 a_amv_lopts = los.as_global_saved;
3897 rele_all_sigs(); /* TODO DISLIKE! */
3899 if(varres == NULL)
3900 goto jestr_nodata;
3901 f &= ~(a_ISNUM | a_ISDECIMAL);
3903 }else if(is_asccaseprefix(argv[0], "iregex")){
3904 f |= a_ICASE;
3905 goto Jregex;
3906 #endif /* HAVE_REGEX */
3907 }else
3908 goto jesubcmd;
3910 n_pstate_err_no = (f & a_SOFTOVERFLOW) ? n_ERR_OVERFLOW : n_ERR_NONE;
3911 f &= ~a_ERR;
3913 /* Generate the variable value content for numerics.
3914 * Anticipate in our handling below! (Don't do needless work) */
3915 jleave:
3916 if((f & a_ISNUM) && ((f & (a_ISDECIMAL | a_PBASE)) || varname != NULL)){
3917 cp = n_ienc_buf(iencbuf, lhv, (f & a_PBASE ? pbase : 10),
3918 n_IENC_MODE_SIGNED_TYPE);
3919 if(cp != NULL)
3920 varres = cp;
3921 else{
3922 f |= a_ERR;
3923 varres = n_empty;
3927 if(varname == NULL){
3928 /* If there was no error and we are printing a numeric result, print some
3929 * more bases for the fun of it */
3930 if((f & (a_ERR | a_ISNUM | a_ISDECIMAL)) == a_ISNUM){
3931 char binabuf[64 + 64 / 8 +1];
3932 size_t j;
3934 for(j = 1, i = 0; i < 64; ++i){
3935 binabuf[63 + 64 / 8 -j - i] = (lhv & ((ui64_t)1 << i)) ? '1' : '0';
3936 if((i & 7) == 7 && i != 63){
3937 ++j;
3938 binabuf[63 + 64 / 8 -j - i] = ' ';
3941 binabuf[64 + 64 / 8 -1] = '\0';
3943 if(fprintf(n_stdout,
3944 "0b %s\n0%" PRIo64 " | 0x%" PRIX64 " | %" PRId64 "\n",
3945 binabuf, lhv, lhv, lhv) < 0 ||
3946 ((f & a_PBASE) && (assert(varres != NULL),
3947 fprintf(n_stdout, "%s\n", varres) < 0))){
3948 n_pstate_err_no = n_err_no;
3949 f |= a_ERR;
3951 }else if(varres != NULL && fprintf(n_stdout, "%s\n", varres) < 0){
3952 n_pstate_err_no = n_err_no;
3953 f |= a_ERR;
3955 }else if(!n_var_vset(varname, (uintptr_t)varres)){
3956 n_pstate_err_no = n_ERR_NOTSUP;
3957 f |= a_ERR;
3959 NYD_LEAVE;
3960 return (f & a_ERR) ? 1 : 0;
3962 jerr:
3963 f = a_ERR | a_ISNUM;
3964 lhv = -1;
3965 goto jleave;
3966 jesubcmd:
3967 n_err(_("`vexpr': invalid subcommand: %s\n"),
3968 n_shexp_quote_cp(*argv, FAL0));
3969 n_pstate_err_no = n_ERR_INVAL;
3970 goto jerr;
3971 jesynopsis:
3972 n_err(_("Synopsis: vexpr: <operator> <:argument:>\n"));
3973 n_pstate_err_no = n_ERR_INVAL;
3974 goto jerr;
3976 jenum_range:
3977 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3978 n_shexp_quote_cp(*argv, FAL0));
3979 n_pstate_err_no = n_ERR_RANGE;
3980 goto jerr;
3981 jenum_overflow:
3982 n_err(_("`vexpr': expression overflows datatype: %" PRId64 " %c %" PRId64
3983 "\n"), lhv, op, rhv);
3984 n_pstate_err_no = n_ERR_OVERFLOW;
3985 goto jerr;
3987 jestr_numrange:
3988 n_err(_("`vexpr': numeric argument invalid or out of range: %s\n"),
3989 n_shexp_quote_cp(*argv, FAL0));
3990 n_pstate_err_no = n_ERR_RANGE;
3991 goto jestr;
3992 jestr_overflow:
3993 n_err(_("`vexpr': string length or offset overflows datatype\n"));
3994 n_pstate_err_no = n_ERR_OVERFLOW;
3995 goto jestr;
3996 jestr_nodata:
3997 n_pstate_err_no = n_ERR_NODATA;
3998 /* FALLTHRU */
3999 jestr:
4000 varres = n_empty;
4001 f &= ~a_ISNUM;
4002 f |= a_ERR;
4003 goto jleave;
4006 FL int
4007 c_vpospar(void *v){
4008 struct n_cmd_arg *cap;
4009 size_t i;
4010 struct a_amv_pospar *appp;
4011 enum{
4012 a_NONE = 0,
4013 a_ERR = 1u<<0,
4014 a_SET = 1u<<1,
4015 a_CLEAR = 1u<<2,
4016 a_QUOTE = 1u<<3
4017 } f;
4018 char const *varres;
4019 struct n_cmd_arg_ctx *cacp;
4020 NYD_ENTER;
4022 n_pstate_err_no = n_ERR_NONE;
4023 n_UNINIT(varres, n_empty);
4024 cacp = v;
4025 cap = cacp->cac_arg;
4027 if(is_asccaseprefix(cap->ca_arg.ca_str.s, "set"))
4028 f = a_SET;
4029 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "clear"))
4030 f = a_CLEAR;
4031 else if(is_asccaseprefix(cap->ca_arg.ca_str.s, "quote"))
4032 f = a_QUOTE;
4033 else{
4034 n_err(_("`vpospar': invalid subcommand: %s\n"),
4035 n_shexp_quote_cp(cap->ca_arg.ca_str.s, FAL0));
4036 n_pstate_err_no = n_ERR_INVAL;
4037 f = a_ERR;
4038 goto jleave;
4040 --cacp->cac_no;
4042 if((f & (a_CLEAR | a_QUOTE)) && cap->ca_next != NULL){
4043 n_err(_("`vpospar': `%s': takes no argument\n"), cap->ca_arg.ca_str.s);
4044 n_pstate_err_no = n_ERR_INVAL;
4045 f = a_ERR;
4046 goto jleave;
4049 cap = cap->ca_next;
4051 /* If in a macro, we need to overwrite the local instead of global argv */
4052 appp = (a_amv_lopts != NULL) ? &a_amv_lopts->as_amcap->amca_pospar
4053 : &a_amv_pospar;
4055 if(f & (a_SET | a_CLEAR)){
4056 if(cacp->cac_vput != NULL)
4057 n_err(_("`vpospar': `vput' only supported for `quote' subcommand\n"));
4058 if(!appp->app_not_heap && appp->app_maxcount > 0){
4059 for(i = appp->app_maxcount; i-- != 0;)
4060 n_free(n_UNCONST(appp->app_dat[i]));
4061 n_free(appp->app_dat);
4063 memset(appp, 0, sizeof *appp);
4065 if(f & a_SET){
4066 if((i = cacp->cac_no) > a_AMV_POSPAR_MAX){
4067 n_err(_("`vpospar': overflow: %" PRIuZ " arguments!\n"), i);
4068 n_pstate_err_no = n_ERR_OVERFLOW;
4069 f = a_ERR;
4070 goto jleave;
4073 memset(appp, 0, sizeof *appp);
4074 if(i > 0){
4075 appp->app_maxcount = appp->app_count = (ui16_t)i;
4076 /* XXX Optimize: store it all in one chunk! */
4077 ++i;
4078 i *= sizeof *appp->app_dat;
4079 appp->app_dat = n_alloc(i);
4081 for(i = 0; cap != NULL; ++i, cap = cap->ca_next){
4082 appp->app_dat[i] = n_alloc(cap->ca_arg.ca_str.l +1);
4083 memcpy(n_UNCONST(appp->app_dat[i]), cap->ca_arg.ca_str.s,
4084 cap->ca_arg.ca_str.l +1);
4087 appp->app_dat[i] = NULL;
4090 }else{
4091 if(appp->app_count == 0)
4092 varres = n_empty;
4093 else{
4094 struct str in;
4095 struct n_string s, *sp;
4096 char sep1, sep2;
4098 sp = n_string_creat_auto(&s);
4100 sep1 = *ok_vlook(ifs);
4101 sep2 = *ok_vlook(ifs_ws);
4102 if(sep1 == sep2)
4103 sep2 = '\0';
4104 if(sep1 == '\0')
4105 sep1 = ' ';
4107 for(i = 0; i < appp->app_count; ++i){
4108 if(sp->s_len){
4109 if(!n_string_can_book(sp, 2))
4110 goto jeover;
4111 sp = n_string_push_c(sp, sep1);
4112 if(sep2 != '\0')
4113 sp = n_string_push_c(sp, sep2);
4115 in.l = strlen(in.s = n_UNCONST(appp->app_dat[i + appp->app_idx]));
4117 if(!n_string_can_book(sp, in.l)){
4118 jeover:
4119 n_err(_("`vpospar': overflow: string too long!\n"));
4120 n_pstate_err_no = n_ERR_OVERFLOW;
4121 f = a_ERR;
4122 goto jleave;
4124 sp = n_shexp_quote(sp, &in, TRU1);
4127 varres = n_string_cp(sp);
4130 if(cacp->cac_vput == NULL){
4131 if(fprintf(n_stdout, "%s\n", varres) < 0){
4132 n_pstate_err_no = n_err_no;
4133 f |= a_ERR;
4135 }else if(!n_var_vset(cacp->cac_vput, (uintptr_t)varres)){
4136 n_pstate_err_no = n_ERR_NOTSUP;
4137 f |= a_ERR;
4140 jleave:
4141 NYD_LEAVE;
4142 return (f & a_ERR) ? 1 : 0;
4145 /* s-it-mode */