1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
2 *@ Auxiliary functions.
4 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
5 * Copyright (c) 2012 - 2014 Steffen "Daode" Nurpmeso <sdaoden@users.sf.net>.
8 * Copyright (c) 1980, 1993
9 * The Regents of the University of California. All rights reserved.
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
14 * 1. Redistributions of source code must retain the above copyright
15 * notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 * notice, this list of conditions and the following disclaimer in the
18 * documentation and/or other materials provided with the distribution.
19 * 3. All advertising materials mentioning features or use of this software
20 * must display the following acknowledgement:
21 * This product includes software developed by the University of
22 * California, Berkeley and its contributors.
23 * 4. Neither the name of the University nor the names of its contributors
24 * may be used to endorse or promote products derived from this software
25 * without specific prior written permission.
27 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
29 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
30 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
31 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
32 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
33 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
34 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
35 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
36 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
40 #ifndef HAVE_AMALGAMATION
44 #include <sys/utsname.h>
52 # include <sys/socket.h>
62 /* Create an ISO 6429 (ECMA-48/ANSI) terminal control escape sequence */
64 static char * _colour_iso6429(char const *wish
);
67 /* {hold,rele}_all_sigs() */
68 static size_t _alls_depth
;
69 static sigset_t _alls_nset
, _alls_oset
;
71 /* {hold,rele}_sigs() */
72 static size_t _hold_sigdepth
;
73 static sigset_t _hold_nset
, _hold_oset
;
77 _colour_iso6429(char const *wish
)
79 char const * const wish_orig
= wish
;
80 char *xwish
, *cp
, cfg
[3] = {0, 0, 0};
82 /* Since we use salloc(), reuse the strcomma() buffer also for the return
83 * value, ensure we have enough room for that */
85 size_t i
= strlen(wish
) + 1;
86 xwish
= salloc(MAX(i
, sizeof("\033[1;30;40m")));
87 memcpy(xwish
, wish
, i
);
91 /* Iterate over the colour spec */
92 while ((cp
= strcomma(&xwish
, TRU1
)) != NULL
) {
93 char *y
, *x
= strchr(cp
, '=');
96 fprintf(stderr
, tr(527,
97 "Invalid colour specification \"%s\": >>> %s <<<\n"),
103 /* TODO convert the ft/fg/bg parser into a table-based one! */
104 if (!asccasecmp(cp
, "ft")) {
105 if (!asccasecmp(x
, "bold"))
107 else if (!asccasecmp(x
, "inverse"))
109 else if (!asccasecmp(x
, "underline"))
113 } else if (!asccasecmp(cp
, "fg")) {
116 } else if (!asccasecmp(cp
, "bg")) {
119 if (!asccasecmp(x
, "black"))
121 else if (!asccasecmp(x
, "blue"))
123 else if (!asccasecmp(x
, "green"))
125 else if (!asccasecmp(x
, "red"))
127 else if (!asccasecmp(x
, "brown"))
129 else if (!asccasecmp(x
, "magenta"))
131 else if (!asccasecmp(x
, "cyan"))
133 else if (!asccasecmp(x
, "white"))
141 /* Restore our salloc() buffer, create return value */
142 xwish
= UNCONST(wish
);
143 if (cfg
[0] || cfg
[1] || cfg
[2]) {
157 if (cfg
[0] || cfg
[1])
166 return UNCONST(wish
);
168 #endif /* HAVE_COLOUR */
171 panic(char const *format
, ...)
175 fprintf(stderr
, tr(1, "Panic: "));
177 va_start(ap
, format
);
178 vfprintf(stderr
, format
, ap
);
188 warn(char const *format
, ...)
192 fprintf(stderr
, tr(1, "Panic: "));
194 va_start(ap
, format
);
195 vfprintf(stderr
, format
, ap
);
204 safe_signal(int signum
, sighandler_type handler
)
206 struct sigaction nact
, oact
;
208 nact
.sa_handler
= handler
;
209 sigemptyset(&nact
.sa_mask
);
212 nact
.sa_flags
|= SA_RESTART
;
214 return ((sigaction(signum
, &nact
, &oact
) != 0) ? SIG_ERR
: oact
.sa_handler
);
220 if (_alls_depth
++ == 0) {
221 sigfillset(&_alls_nset
);
222 sigdelset(&_alls_nset
, SIGABRT
);
224 sigdelset(&_alls_nset
, SIGBUS
);
226 sigdelset(&_alls_nset
, SIGCHLD
);
227 sigdelset(&_alls_nset
, SIGFPE
);
228 sigdelset(&_alls_nset
, SIGILL
);
229 sigdelset(&_alls_nset
, SIGKILL
);
230 sigdelset(&_alls_nset
, SIGSEGV
);
231 sigdelset(&_alls_nset
, SIGSTOP
);
232 sigprocmask(SIG_BLOCK
, &_alls_nset
, &_alls_oset
);
239 if (--_alls_depth
== 0)
240 sigprocmask(SIG_SETMASK
, &_alls_oset
, (sigset_t
*)NULL
);
246 if (_hold_sigdepth
++ == 0) {
247 sigemptyset(&_hold_nset
);
248 sigaddset(&_hold_nset
, SIGHUP
);
249 sigaddset(&_hold_nset
, SIGINT
);
250 sigaddset(&_hold_nset
, SIGQUIT
);
251 sigprocmask(SIG_BLOCK
, &_hold_nset
, &_hold_oset
);
258 if (--_hold_sigdepth
== 0)
259 sigprocmask(SIG_SETMASK
, &_hold_oset
, NULL
);
263 * Touch the named message by setting its MTOUCH flag.
264 * Touched messages have the effect of not being sent
265 * back to the system mailbox on exit.
268 touch(struct message
*mp
)
271 mp
->m_flag
|= MTOUCH
;
272 if ((mp
->m_flag
& MREAD
) == 0)
273 mp
->m_flag
|= MREAD
|MSTATUS
;
277 * Test to see if the passed file name is a directory.
278 * Return true if it is.
281 is_dir(char const *name
)
285 if (stat(name
, &sbuf
) < 0)
287 return(S_ISDIR(sbuf
.st_mode
));
291 * Count the number of arguments in the given string raw list.
294 argcount(char **argv
)
298 for (ap
= argv
; *ap
++ != NULL
;)
300 return ap
- argv
- 1;
304 colalign(const char *cp
, int col
, int fill
, int *cols_decr_used_or_null
)
306 int col_orig
= col
, n
, sz
;
309 np
= nb
= salloc(mb_cur_max
* strlen(cp
) + col
+ 1);
312 if (mb_cur_max
> 1) {
315 if ((sz
= mbtowc(&wc
, cp
, mb_cur_max
)) < 0) {
318 if ((n
= wcwidth(wc
)) < 0)
329 if (sz
== 1 && spacechar(*cp
)) {
337 if (fill
&& col
!= 0) {
339 memmove(nb
+ col
, nb
, (size_t)(np
- nb
));
340 memset(nb
, ' ', col
);
342 memset(np
, ' ', col
);
348 if (cols_decr_used_or_null
!= NULL
)
349 *cols_decr_used_or_null
-= col_orig
- col
;
358 cp
= ok_vlook(PAGER
);
359 if (cp
== NULL
|| *cp
== '\0')
365 paging_seems_sensible(void)
370 if (IS_TTY_SESSION() && (cp
= ok_vlook(crt
)) != NULL
)
371 ret
= (*cp
!= '\0') ? (size_t)atol(cp
) : (size_t)scrnheight
;
376 page_or_print(FILE *fp
, size_t lines
)
383 if ((rows
= paging_seems_sensible()) != 0 && lines
== 0) {
384 while ((c
= getc(fp
)) != EOF
)
385 if (c
== '\n' && ++lines
> rows
)
390 if (rows
!= 0 && lines
>= rows
)
391 run_command(get_pager(), 0, fileno(fp
), -1, NULL
, NULL
, NULL
);
393 while ((c
= getc(fp
)) != EOF
)
398 which_protocol(const char *name
)
400 register const char *cp
;
406 if (name
[0] == '%' && name
[1] == ':')
408 for (cp
= name
; *cp
&& *cp
!= ':'; cp
++)
409 if (!alnumchar(*cp
&0377))
411 if (cp
[0] == ':' && cp
[1] == '/' && cp
[2] == '/') {
412 if (strncmp(name
, "pop3://", 7) == 0)
417 tr(216, "No POP3 support compiled in.\n"));
419 if (strncmp(name
, "pop3s://", 8) == 0)
424 tr(225, "No SSL support compiled in.\n"));
426 if (strncmp(name
, "imap://", 7) == 0)
431 tr(269, "No IMAP support compiled in.\n"));
433 if (strncmp(name
, "imaps://", 8) == 0)
438 tr(225, "No SSL support compiled in.\n"));
440 return PROTO_UNKNOWN
;
442 /* TODO This is the de facto maildir code and thus belongs
443 * TODO into maildir! */
444 file
: p
= PROTO_FILE
;
445 np
= ac_alloc((sz
= strlen(name
)) + 5);
446 memcpy(np
, name
, sz
+ 1);
447 if (stat(name
, &st
) == 0) {
448 if (S_ISDIR(st
.st_mode
)) {
449 strcpy(&np
[sz
], "/tmp");
450 if (stat(np
, &st
) == 0 && S_ISDIR(st
.st_mode
)) {
451 strcpy(&np
[sz
], "/new");
452 if (stat(np
, &st
) == 0 &&
453 S_ISDIR(st
.st_mode
)) {
454 strcpy(&np
[sz
], "/cur");
455 if (stat(np
, &st
) == 0 &&
462 strcpy(&np
[sz
], ".gz");
463 if (stat(np
, &st
) < 0) {
464 strcpy(&np
[sz
], ".bz2");
465 if (stat(np
, &st
) < 0) {
466 if ((cp
= ok_vlook(newfolders
)) != NULL
&&
467 strcmp(cp
, "maildir") == 0)
478 torek_hash(char const *name
)
480 /* Chris Torek's hash.
481 * NOTE: need to change *at least* create-okey-map.pl when changing the
485 while (*name
!= '\0') {
499 h
= (h
<< 4 & 0xffffffff) + (*cp
&0377);
500 if ((g
= h
& 0xf0000000) != 0) {
511 const long primes
[] = {
512 509, 1021, 2039, 4093, 8191, 16381, 32749, 65521,
513 131071, 262139, 524287, 1048573, 2097143, 4194301,
514 8388593, 16777213, 33554393, 67108859, 134217689,
515 268435399, 536870909, 1073741789, 2147483647
520 for (i
= 0; i
< sizeof primes
/ sizeof *primes
; i
++)
521 if ((mprime
= primes
[i
]) >= (n
< 65536 ? n
*4 :
522 n
< 262144 ? n
*2 : n
))
524 if (i
== sizeof primes
/ sizeof *primes
)
525 mprime
= n
; /* not so prime, but better than failure */
530 expand_shell_escape(char const **s
, bool_t use_nail_extensions
)
535 if ((c
= *xs
& 0xFF) == '\0')
541 switch ((c
= *xs
& 0xFF)) {
543 case 'a': c
= '\a'; break;
544 case 'b': c
= '\b'; break;
545 case 'c': c
= PROMPT_STOP
; break;
546 case 'f': c
= '\f'; break;
547 case 'n': c
= '\n'; break;
548 case 'r': c
= '\r'; break;
549 case 't': c
= '\t'; break;
550 case 'v': c
= '\v'; break;
552 for (++xs
, c
= 0, n
= 4; --n
> 0 && octalchar(*xs
); ++xs
) {
557 /* S-nail extension for nice (get)prompt(()) support */
562 if (use_nail_extensions
) {
564 case '&': c
= ok_blook(bsdcompat
) ? '&' : '?'; break;
565 case '?': c
= exec_last_comm_error
? '1' : '0'; break;
566 case '$': c
= PROMPT_DOLLAR
; break;
567 case '@': c
= PROMPT_AT
; break;
573 /* A sole <backslash> at EOS is treated as-is! */
588 static char buf
[PROMPT_BUFFER_SIZE
];
593 if ((ccp
= ok_vlook(prompt
)) == NULL
|| *ccp
== '\0')
596 for (; PTRCMP(cp
, <, buf
+ sizeof(buf
) - 1); ++cp
) {
599 int c
= expand_shell_escape(&ccp
, TRU1
);
605 if (c
== 0 || c
== PROMPT_STOP
)
608 a
= (c
== PROMPT_DOLLAR
) ? account_name
: displayname
;
612 if (PTRCMP(cp
+ l
, >=, buf
+ sizeof(buf
) - 1))
625 nodename(int mayoverride
)
627 static char *hostname
;
632 struct addrinfo hints
, *res
;
634 struct hostent
*hent
;
638 if (mayoverride
&& (hn
= ok_vlook(hostname
)) != NULL
&& *hn
!= '\0') {
639 if (hostname
!= NULL
)
641 hostname
= sstrdup(hn
);
642 } else if (hostname
== NULL
) {
647 memset(&hints
, 0, sizeof hints
);
648 hints
.ai_family
= AF_UNSPEC
;
649 hints
.ai_socktype
= SOCK_DGRAM
; /* dummy */
650 hints
.ai_flags
= AI_CANONNAME
;
651 if (getaddrinfo(hn
, "0", &hints
, &res
) == 0) {
652 if (res
->ai_canonname
!= NULL
) {
653 size_t l
= strlen(res
->ai_canonname
);
654 hn
= ac_alloc(l
+ 1);
655 memcpy(hn
, res
->ai_canonname
, l
+ 1);
660 hent
= gethostbyname(hn
);
666 hostname
= sstrdup(hn
);
667 #if defined HAVE_SOCKETS && defined HAVE_IPV6
668 if (hn
!= ut
.nodename
)
676 lookup_password_for_token(char const *token
)
682 var
= ac_alloc(tl
+ 10);
684 memcpy(var
, "password-", 9);
685 memcpy(var
+ 9, token
, tl
);
688 if ((cp
= vok_vlook(var
)) != NULL
)
695 getrandstring(size_t length
)
697 static unsigned char nodedigest
[16];
709 data
= ac_alloc(length
);
710 if ((fd
= open("/dev/urandom", O_RDONLY
)) < 0 ||
711 length
!= (size_t)read(fd
, data
, length
)) {
718 md5_update(&ctx
, (unsigned char*)cp
, strlen(cp
));
719 md5_final(nodedigest
, &ctx
);
721 /* In that case it's only used for boundaries and
722 * Message-Id:s so that srand(3) should suffice */
724 for (i
= 0; i
< sizeof(nodedigest
); ++i
)
725 nodedigest
[i
] = (unsigned char)(
729 for (i
= 0; i
< length
; i
++)
731 (int)(255 * (rand() / (RAND_MAX
+ 1.0))) ^
732 nodedigest
[i
% sizeof nodedigest
]);
737 (void)b64_encode_buf(&b64
, data
, length
, B64_SALLOC
);
739 assert(length
< b64
.l
);
740 b64
.s
[length
] = '\0';
746 md5tohex(char hex
[MD5TOHEX_SIZE
], void const *vp
)
751 for (i
= 0; i
< MD5TOHEX_SIZE
/ 2; i
++) {
753 hex
[j
] = hexchar((cp
[i
] & 0xf0) >> 4);
754 hex
[++j
] = hexchar(cp
[i
] & 0x0f);
760 cram_md5_string(char const *user
, char const *pass
, char const *b64
)
763 char digest
[16], *cp
;
769 (void)b64_decode(&out
, &in
, NULL
);
770 assert(out
.s
!= NULL
);
772 hmac_md5((unsigned char*)out
.s
, out
.l
, UNCONST(pass
), strlen(pass
),
775 cp
= md5tohex(salloc(MD5TOHEX_SIZE
+ 1), digest
);
778 in
.l
= lu
+ MD5TOHEX_SIZE
+1;
779 in
.s
= ac_alloc(lu
+ 1 + MD5TOHEX_SIZE
+1);
780 memcpy(in
.s
, user
, lu
);
782 memcpy(in
.s
+ lu
+ 1, cp
, MD5TOHEX_SIZE
);
783 (void)b64_encode(&out
, &in
, B64_SALLOC
|B64_CRLF
);
787 #endif /* HAVE_MD5 */
790 makedir(const char *name
)
795 if (mkdir(name
, 0700) < 0) {
797 if ((e
== EEXIST
|| e
== ENOSYS
) &&
798 stat(name
, &st
) == 0 &&
799 (st
.st_mode
&S_IFMT
) == S_IFDIR
)
810 if ((cw
->cw_fd
= open(".", O_RDONLY
)) < 0)
812 if (fchdir(cw
->cw_fd
) < 0) {
822 if (fchdir(cw
->cw_fd
) < 0)
828 cwrelse(struct cw
*cw
)
832 #else /* !HAVE_FCHDIR */
836 if (getcwd(cw
->cw_wd
, sizeof cw
->cw_wd
) == NULL
|| chdir(cw
->cw_wd
) < 0)
844 if (chdir(cw
->cw_wd
) < 0)
851 cwrelse(struct cw
*cw
)
855 #endif /* !HAVE_FCHDIR */
858 makeprint(struct str
const *in
, struct str
*out
)
860 static int print_all_chars
= -1;
861 char const *inp
, *maxp
;
865 if (print_all_chars
== -1)
866 print_all_chars
= ok_blook(print_all_chars
);
869 out
->s
= outp
= smalloc(msz
);
873 if (print_all_chars
) {
875 memcpy(outp
, inp
, out
->l
);
879 #ifdef HAVE_C90AMEND1
880 if (mb_cur_max
> 1) {
881 char mbb
[MB_LEN_MAX
+ 1];
889 n
= mbtowc(&wc
, inp
, maxp
- inp
);
895 /* FIXME Why mbtowc() resetting here?
896 * FIXME what about ISO 2022-JP plus -- those
897 * FIXME will loose shifts, then!
898 * FIXME THUS - we'd need special "known points"
899 * FIXME to do so - say, after a newline!!
900 * FIXME WE NEED TO CHANGE ALL USES +MBLEN! */
901 (void)mbtowc(&wc
, NULL
, mb_cur_max
);
902 wc
= utf8
? 0xFFFD : '?';
907 if (!iswprint(wc
) && wc
!= '\n' && wc
!= '\r' &&
908 wc
!= '\b' && wc
!= '\t') {
909 if ((wc
& ~(wchar_t)037) == 0)
910 wc
= utf8
? 0x2400 | wc
: '?';
912 wc
= utf8
? 0x2421 : '?';
914 wc
= utf8
? 0x2426 : '?';
916 if ((n
= wctomb(mbb
, wc
)) <= 0)
919 if (out
->l
>= msz
- 1) {
920 dist
= outp
- out
->s
;
921 out
->s
= srealloc(out
->s
, msz
+= 32);
922 outp
= &out
->s
[dist
];
924 for (i
= 0; i
< n
; i
++)
928 #endif /* C90AMEND1 */
933 if (!isprint(c
) && c
!= '\n' && c
!= '\r' &&
934 c
!= '\b' && c
!= '\t')
941 out
->s
[out
->l
] = '\0';
952 makeprint(&in
, &out
);
953 rp
= salloc(out
.l
+ 1);
954 memcpy(rp
, out
.s
, out
.l
);
961 prout(const char *s
, size_t sz
, FILE *fp
)
968 makeprint(&in
, &out
);
969 n
= fwrite(out
.s
, 1, out
.l
, fp
);
975 putuc(int u
, int c
, FILE *fp
)
980 #ifdef HAVE_C90AMEND1
981 if (utf8
&& (u
& ~(wchar_t)0177)) {
982 char mbb
[MB_LEN_MAX
];
984 if ((n
= wctomb(mbb
, u
)) > 0) {
986 for (i
= 0; i
< n
; ++i
)
987 if (putc(mbb
[i
] & 0377, fp
) == EOF
) {
992 rv
= (putc('\0', fp
) != EOF
);
997 rv
= (putc(c
, fp
) != EOF
);
1003 colour_table_create(char const *pager_used
)
1005 union {char *cp
; char const *ccp
; void *vp
; struct colour_table
*ctp
;} u
;
1007 struct colour_table
*ct
;
1009 if (ok_blook(colour_disable
))
1012 /* If pager, check wether it is allowed to use colour */
1013 if (pager_used
!= NULL
) {
1016 if ((u
.cp
= ok_vlook(colour_pagers
)) == NULL
)
1017 u
.ccp
= COLOUR_PAGERS
;
1018 pager
= savestr(u
.cp
);
1020 while ((u
.cp
= strcomma(&pager
, TRU1
)) != NULL
)
1021 if (strstr(pager_used
, u
.cp
) != NULL
)
1026 /* $TERM is different in that we default to false unless whitelisted */
1028 char *term
, *okterms
;
1030 /* Don't use getenv(), but force copy-in into our own tables.. */
1031 if ((term
= _var_voklook("TERM")) == NULL
)
1033 if ((okterms
= ok_vlook(colour_terms
)) == NULL
)
1034 okterms
= UNCONST(COLOUR_TERMS
);
1035 okterms
= savestr(okterms
);
1038 while ((u
.cp
= strcomma(&okterms
, TRU1
)) != NULL
)
1039 if (!strncmp(u
.cp
, term
, i
))
1045 colour_table
= ct
= salloc(sizeof *ct
); /* XXX lex.c yet resets (FILTER!) */
1048 enum colourspec cspec
;
1051 {ok_v_colour_msginfo
, COLOURSPEC_MSGINFO
, COLOUR_MSGINFO
},
1052 {ok_v_colour_partinfo
, COLOURSPEC_PARTINFO
, COLOUR_PARTINFO
},
1053 {ok_v_colour_from_
, COLOURSPEC_FROM_
, COLOUR_FROM_
},
1054 {ok_v_colour_header
, COLOURSPEC_HEADER
, COLOUR_HEADER
},
1055 {ok_v_colour_uheader
, COLOURSPEC_UHEADER
, COLOUR_UHEADER
}
1058 for (i
= 0; i
< NELEM(map
); ++i
) {
1059 if ((u
.cp
= _var_oklook(map
[i
].okey
)) == NULL
)
1060 u
.ccp
= map
[i
].defval
;
1061 u
.cp
= _colour_iso6429(u
.ccp
);
1062 ct
->ct_csinfo
[map
[i
].cspec
].l
= strlen(u
.cp
);
1063 ct
->ct_csinfo
[map
[i
].cspec
].s
= u
.cp
;
1066 ct
->ct_csinfo
[COLOURSPEC_RESET
].l
= sizeof("\033[0m") - 1;
1067 ct
->ct_csinfo
[COLOURSPEC_RESET
].s
= UNCONST("\033[0m");
1069 if ((u
.cp
= ok_vlook(colour_user_headers
)) == NULL
)
1070 u
.ccp
= COLOUR_USER_HEADERS
;
1071 ct
->ct_csinfo
[COLOURSPEC_RESET
+ 1].l
= i
= strlen(u
.ccp
);
1072 ct
->ct_csinfo
[COLOURSPEC_RESET
+ 1].s
= (i
== 0) ? NULL
: savestr(u
.ccp
);
1078 colour_put(FILE *fp
, enum colourspec cs
)
1080 if (colour_table
!= NULL
) {
1081 struct str
const *cp
= colour_get(cs
);
1083 fwrite(cp
->s
, cp
->l
, 1, fp
);
1088 colour_put_header(FILE *fp
, char const *name
)
1090 enum colourspec cs
= COLOURSPEC_HEADER
;
1091 struct str
const *uheads
;
1092 char *cp
, *cp_base
, *x
;
1095 if (colour_table
== NULL
)
1097 /* Normal header colours if there are no user headers */
1098 uheads
= colour_table
->ct_csinfo
+ COLOURSPEC_RESET
+ 1;
1099 if (uheads
->s
== NULL
)
1102 /* Iterate over all entries in the *colour-user-headers* list */
1103 cp
= ac_alloc(uheads
->l
+ 1);
1104 memcpy(cp
, uheads
->s
, uheads
->l
+ 1);
1106 namelen
= strlen(name
);
1107 while ((x
= strcomma(&cp
, TRU1
)) != NULL
) {
1108 size_t l
= (cp
!= NULL
) ? PTR2SIZE(cp
- x
) - 1 : strlen(x
);
1109 if (l
== namelen
&& !ascncasecmp(x
, name
, namelen
)) {
1110 cs
= COLOURSPEC_UHEADER
;
1122 colour_reset(FILE *fp
)
1124 if (colour_table
!= NULL
)
1125 fwrite("\033[0m", 4, 1, fp
);
1128 FL
struct str
const *
1129 colour_get(enum colourspec cs
)
1131 struct str
const *rv
= NULL
;
1133 if (colour_table
!= NULL
)
1134 if ((rv
= colour_table
->ct_csinfo
+ cs
)->s
== NULL
)
1138 #endif /* HAVE_COLOUR */
1141 time_current_update(struct time_current
*tc
, bool_t full_update
)
1143 tc
->tc_time
= time(NULL
);
1145 memcpy(&tc
->tc_gm
, gmtime(&tc
->tc_time
), sizeof tc
->tc_gm
);
1146 memcpy(&tc
->tc_local
, localtime(&tc
->tc_time
),
1147 sizeof tc
->tc_local
);
1148 sstpcpy(tc
->tc_ctime
, ctime(&tc
->tc_time
));
1153 _out_of_memory(void)
1160 smalloc(size_t s SMALLOC_DEBUG_ARGS
)
1166 if ((rv
= malloc(s
)) == NULL
)
1172 srealloc(void *v
, size_t s SMALLOC_DEBUG_ARGS
)
1180 else if ((rv
= realloc(v
, s
)) == NULL
)
1186 scalloc(size_t nmemb
, size_t size SMALLOC_DEBUG_ARGS
)
1192 if ((rv
= calloc(nmemb
, size
)) == NULL
)
1197 #else /* !HAVE_DEBUG */
1198 CTA(sizeof(char) == sizeof(ui8_t
));
1200 # define _HOPE_SIZE (2 * 8 * sizeof(char))
1201 # define _HOPE_SET(C) \
1203 union ptr __xl, __xu;\
1204 struct chunk *__xc;\
1209 __xl.ui8p[0]=0xDE; __xl.ui8p[1]=0xAA; __xl.ui8p[2]=0x55; __xl.ui8p[3]=0xAD;\
1210 __xl.ui8p[4]=0xBE; __xl.ui8p[5]=0x55; __xl.ui8p[6]=0xAA; __xl.ui8p[7]=0xEF;\
1211 __xu.ui8p += __xc->size - 8;\
1212 __xu.ui8p[0]=0xDE; __xu.ui8p[1]=0xAA; __xu.ui8p[2]=0x55; __xu.ui8p[3]=0xAD;\
1213 __xu.ui8p[4]=0xBE; __xu.ui8p[5]=0x55; __xu.ui8p[6]=0xAA; __xu.ui8p[7]=0xEF;\
1215 # define _HOPE_GET_TRACE(C,BAD) do {(C).cp += 8; _HOPE_GET(C, BAD);} while(0)
1216 # define _HOPE_GET(C,BAD) \
1218 union ptr __xl, __xu;\
1219 struct chunk *__xc;\
1227 if (__xl.ui8p[0] != 0xDE) __i |= 1<<0;\
1228 if (__xl.ui8p[1] != 0xAA) __i |= 1<<1;\
1229 if (__xl.ui8p[2] != 0x55) __i |= 1<<2;\
1230 if (__xl.ui8p[3] != 0xAD) __i |= 1<<3;\
1231 if (__xl.ui8p[4] != 0xBE) __i |= 1<<4;\
1232 if (__xl.ui8p[5] != 0x55) __i |= 1<<5;\
1233 if (__xl.ui8p[6] != 0xAA) __i |= 1<<6;\
1234 if (__xl.ui8p[7] != 0xEF) __i |= 1<<7;\
1237 warn("%p: corrupted lower canary: 0x%02X: %s, line %u",\
1238 __xl.p, __i, mdbg_file, mdbg_line);\
1241 __xu.ui8p += __xc->size - 8;\
1243 if (__xu.ui8p[0] != 0xDE) __i |= 1<<0;\
1244 if (__xu.ui8p[1] != 0xAA) __i |= 1<<1;\
1245 if (__xu.ui8p[2] != 0x55) __i |= 1<<2;\
1246 if (__xu.ui8p[3] != 0xAD) __i |= 1<<3;\
1247 if (__xu.ui8p[4] != 0xBE) __i |= 1<<4;\
1248 if (__xu.ui8p[5] != 0x55) __i |= 1<<5;\
1249 if (__xu.ui8p[6] != 0xAA) __i |= 1<<6;\
1250 if (__xu.ui8p[7] != 0xEF) __i |= 1<<7;\
1253 warn("%p: corrupted upper canary: 0x%02X: %s, line %u",\
1254 __xl.p, __i, mdbg_file, mdbg_line);\
1257 warn(" ..canary last seen: %s, line %u", __xc->file, __xc->line);\
1277 struct chunk
*_mlist
, *_mfree
;
1280 (smalloc
)(size_t s SMALLOC_DEBUG_ARGS
)
1286 s
+= sizeof(struct chunk
) + _HOPE_SIZE
;
1288 if ((p
.p
= (malloc
)(s
)) == NULL
)
1291 if ((p
.c
->next
= _mlist
) != NULL
)
1293 p
.c
->file
= mdbg_file
;
1294 p
.c
->line
= (ui16_t
)mdbg_line
;
1296 p
.c
->size
= (ui32_t
)s
;
1303 (srealloc
)(void *v
, size_t s SMALLOC_DEBUG_ARGS
)
1308 if ((p
.p
= v
) == NULL
) {
1309 p
.p
= (smalloc
)(s
, mdbg_file
, mdbg_line
);
1313 _HOPE_GET(p
, isbad
);
1316 fprintf(stderr
, "srealloc(): region freed! At %s, line %d\n"
1317 "\tLast seen: %s, line %d\n",
1318 mdbg_file
, mdbg_line
, p
.c
->file
, p
.c
->line
);
1325 p
.c
->prev
->next
= p
.c
->next
;
1326 if (p
.c
->next
!= NULL
)
1327 p
.c
->next
->prev
= p
.c
->prev
;
1332 s
+= sizeof(struct chunk
) + _HOPE_SIZE
;
1334 if ((p
.p
= (realloc
)(p
.c
, s
)) == NULL
)
1337 if ((p
.c
->next
= _mlist
) != NULL
)
1339 p
.c
->file
= mdbg_file
;
1340 p
.c
->line
= (ui16_t
)mdbg_line
;
1342 p
.c
->size
= (ui32_t
)s
;
1350 (scalloc
)(size_t nmemb
, size_t size SMALLOC_DEBUG_ARGS
)
1359 size
+= sizeof(struct chunk
) + _HOPE_SIZE
;
1361 if ((p
.p
= (malloc
)(size
)) == NULL
)
1363 memset(p
.p
, 0, size
);
1365 if ((p
.c
->next
= _mlist
) != NULL
)
1367 p
.c
->file
= mdbg_file
;
1368 p
.c
->line
= (ui16_t
)mdbg_line
;
1370 p
.c
->size
= (ui32_t
)size
;
1377 (sfree
)(void *v SMALLOC_DEBUG_ARGS
)
1382 if ((p
.p
= v
) == NULL
) {
1383 fprintf(stderr
, "sfree(NULL) from %s, line %d\n", mdbg_file
, mdbg_line
);
1387 _HOPE_GET(p
, isbad
);
1390 fprintf(stderr
, "sfree(): double-free avoided at %s, line %d\n"
1391 "\tLast seen: %s, line %d\n",
1392 mdbg_file
, mdbg_line
, p
.c
->file
, p
.c
->line
);
1399 p
.c
->prev
->next
= p
.c
->next
;
1400 if (p
.c
->next
!= NULL
)
1401 p
.c
->next
->prev
= p
.c
->prev
;
1404 if (options
& OPT_DEBUG
) {
1417 size_t c
= 0, s
= 0;
1419 for (p
.c
= _mfree
; p
.c
!= NULL
;) {
1428 if (options
& OPT_DEBUG
)
1429 fprintf(stderr
, "smemreset(): freed %" ZFMT
" chunks/%" ZFMT
" bytes\n",
1436 /* For _HOPE_GET() */
1437 char const * const mdbg_file
= "smemtrace()";
1438 int const mdbg_line
= -1;
1447 if ((fp
= Ftemp(&cp
, "Ra", "w+", 0600, 1)) == NULL
) {
1454 fprintf(fp
, "Currently allocated memory chunks:\n");
1455 for (lines
= 0, p
.c
= _mlist
; p
.c
!= NULL
; ++lines
, p
.c
= p
.c
->next
) {
1458 _HOPE_GET_TRACE(xp
, isbad
);
1459 fprintf(fp
, "%s%p (%5" ZFMT
" bytes): %s, line %u\n",
1460 (isbad
? "! CANARY ERROR: " : ""), xp
.p
,
1461 (size_t)(p
.c
->size
- sizeof(struct chunk
)), p
.c
->file
, p
.c
->line
);
1464 if (options
& OPT_DEBUG
) {
1465 fprintf(fp
, "sfree()d memory chunks awaiting free():\n");
1466 for (p
.c
= _mfree
; p
.c
!= NULL
; ++lines
, p
.c
= p
.c
->next
) {
1469 _HOPE_GET_TRACE(xp
, isbad
);
1470 fprintf(fp
, "%s%p (%5" ZFMT
" bytes): %s, line %u\n",
1471 (isbad
? "! CANARY ERROR: " : ""), xp
.p
,
1472 (size_t)(p
.c
->size
- sizeof(struct chunk
)), p
.c
->file
, p
.c
->line
);
1476 page_or_print(fp
, lines
);
1485 _smemcheck(char const *mdbg_file
, int mdbg_line
)
1488 bool_t anybad
= FAL0
, isbad
;
1491 for (lines
= 0, p
.c
= _mlist
; p
.c
!= NULL
; ++lines
, p
.c
= p
.c
->next
) {
1494 _HOPE_GET_TRACE(xp
, isbad
);
1498 "! CANARY ERROR: %p (%5" ZFMT
" bytes): %s, line %u\n",
1499 xp
.p
, (size_t)(p
.c
->size
- sizeof(struct chunk
)),
1500 p
.c
->file
, p
.c
->line
);
1504 if (options
& OPT_DEBUG
) {
1505 for (p
.c
= _mfree
; p
.c
!= NULL
; ++lines
, p
.c
= p
.c
->next
) {
1508 _HOPE_GET_TRACE(xp
, isbad
);
1512 "! CANARY ERROR: %p (%5" ZFMT
" bytes): %s, line %u\n",
1513 xp
.p
, (size_t)(p
.c
->size
- sizeof(struct chunk
)),
1514 p
.c
->file
, p
.c
->line
);
1520 # endif /* MEMCHECK */
1521 #endif /* HAVE_DEBUG */
1523 /* vim:set fenc=utf-8:s-it-mode (TODO only partial true) */