428 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			428 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C
		
	
	
	
| /* lib.h - header file for lib directory
 | |
|  *
 | |
|  * Copyright 2006 Rob Landley <rob@landley.net>
 | |
|  */
 | |
| 
 | |
| struct ptr_len {
 | |
|   void *ptr;
 | |
|   long len;
 | |
| };
 | |
| 
 | |
| // llist.c
 | |
| 
 | |
| // All these list types can be handled by the same code because first element
 | |
| // is always next pointer, so next = (mytype *)&struct. (The payloads are
 | |
| // named differently to catch using the wrong type early.)
 | |
| 
 | |
| struct string_list {
 | |
|   struct string_list *next;
 | |
|   char str[];
 | |
| };
 | |
| 
 | |
| struct arg_list {
 | |
|   struct arg_list *next;
 | |
|   char *arg;
 | |
| };
 | |
| 
 | |
| struct double_list {
 | |
|   struct double_list *next, *prev;
 | |
|   char *data;
 | |
| };
 | |
| 
 | |
| struct num_cache {
 | |
|   struct num_cache *next;
 | |
|   long long num;
 | |
|   char data[];
 | |
| };
 | |
| 
 | |
| void llist_free_arg(void *node);
 | |
| void llist_free_double(void *node);
 | |
| void llist_traverse(void *list, void (*using)(void *node));
 | |
| void *llist_pop(void *list);  // actually void **list
 | |
| void *dlist_pop(void *list);  // actually struct double_list **list
 | |
| void *dlist_lpop(void *list); // also struct double_list **list
 | |
| void dlist_add_nomalloc(struct double_list **list, struct double_list *new);
 | |
| struct double_list *dlist_add(struct double_list **list, char *data);
 | |
| void *dlist_terminate(void *list);
 | |
| struct num_cache *get_num_cache(struct num_cache *cache, long long num);
 | |
| struct num_cache *add_num_cache(struct num_cache **cache, long long num,
 | |
|   void *data, int len);
 | |
| 
 | |
| // args.c
 | |
| #define FLAGS_NODASH (1LL<<63)
 | |
| void get_optflags(void);
 | |
| 
 | |
| // dirtree.c
 | |
| 
 | |
| // Values returnable from callback function (bitfield, or them together)
 | |
| // Default with no callback is 0
 | |
| 
 | |
| // Add this node to the tree
 | |
| #define DIRTREE_SAVE         1
 | |
| // Recurse into children
 | |
| #define DIRTREE_RECURSE      2
 | |
| // Call again after handling all children of this directory
 | |
| // (Ignored for non-directories, sets linklen = -1 before second call.)
 | |
| #define DIRTREE_COMEAGAIN    4
 | |
| // Follow symlinks to directories
 | |
| #define DIRTREE_SYMFOLLOW    8
 | |
| // Don't warn about failure to stat
 | |
| #define DIRTREE_SHUTUP      16
 | |
| // Breadth first traversal, conserves filehandles at the expense of memory
 | |
| #define DIRTREE_BREADTH     32 // TODO not implemented yet
 | |
| // skip non-numeric entries
 | |
| #define DIRTREE_PROC        64
 | |
| // Return files we can't stat
 | |
| #define DIRTREE_STATLESS   128
 | |
| // Don't look at any more files in this directory.
 | |
| #define DIRTREE_ABORT      256
 | |
| 
 | |
| #define DIRTREE_ABORTVAL ((struct dirtree *)1)
 | |
| 
 | |
| struct dirtree {
 | |
|   struct dirtree *next, *parent, *child;
 | |
|   long extra; // place for user to store their stuff (can be pointer)
 | |
|   char *symlink;
 | |
|   int dirfd;
 | |
|   struct stat st;
 | |
|   char again, name[];
 | |
| };
 | |
| 
 | |
| int isdotdot(char *name);
 | |
| struct dirtree *dirtree_add_node(struct dirtree *p, char *name, int flags);
 | |
| char *dirtree_path(struct dirtree *node, int *plen);
 | |
| int dirtree_notdotdot(struct dirtree *catch);
 | |
| int dirtree_parentfd(struct dirtree *node);
 | |
| int dirtree_recurse(struct dirtree *node, int (*callback)(struct dirtree *node),
 | |
|   int dirfd, int symfollow);
 | |
| struct dirtree *dirtree_flagread(char *path, int flags,
 | |
|   int (*callback)(struct dirtree *node));
 | |
| struct dirtree *dirtree_read(char *path, int (*callback)(struct dirtree *node));
 | |
| 
 | |
| // Tell xopen and friends to print warnings but return -1 as necessary
 | |
| // The largest O_BLAH flag so far is arch/alpha's O_PATH at 0x800000 so
 | |
| // plenty of headroom.
 | |
| #define WARN_ONLY        (1<<31) // don't exit, just warn
 | |
| #define LOOPFILES_ANYWAY (1<<30) // call function with fd -1
 | |
| 
 | |
| // xabspath flags
 | |
| #define ABS_PATH 1 // all but last path component must exist
 | |
| #define ABS_FILE 2 // last path component must exist
 | |
| #define ABS_KEEP 4 // don't resolve symlinks in path to last component
 | |
| #define ABS_LAST 8 // don't resolve symlink in last path component
 | |
| 
 | |
| // xwrap.c
 | |
| void xstrncpy(char *dest, char *src, size_t size);
 | |
| void xstrncat(char *dest, char *src, size_t size);
 | |
| _Noreturn void _xexit(void);
 | |
| _Noreturn void xexit(void);
 | |
| void *xmmap(void *addr, size_t length, int prot, int flags, int fd, off_t off);
 | |
| void *xmalloc(size_t size);
 | |
| void *xzalloc(size_t size);
 | |
| void *xrealloc(void *ptr, size_t size);
 | |
| char *xstrndup(char *s, size_t n);
 | |
| char *xstrdup(char *s);
 | |
| void *xmemdup(void *s, long len);
 | |
| char *xmprintf(char *format, ...) printf_format;
 | |
| void xflush(int flush);
 | |
| void xprintf(char *format, ...) printf_format;
 | |
| void xputsl(char *s, int len);
 | |
| void xputsn(char *s);
 | |
| void xputs(char *s);
 | |
| void xputc(char c);
 | |
| void xvdaemon(void);
 | |
| void xexec(char **argv);
 | |
| pid_t xpopen_setup(char **argv, int *pipes, void (*callback)(char **argv));
 | |
| pid_t xpopen_both(char **argv, int *pipes);
 | |
| int xwaitpid(pid_t pid);
 | |
| int xpclose_both(pid_t pid, int *pipes);
 | |
| pid_t xpopen(char **argv, int *pipe, int isstdout);
 | |
| pid_t xpclose(pid_t pid, int pipe);
 | |
| int xrun(char **argv);
 | |
| int xpspawn(char **argv, int*pipes);
 | |
| void xaccess(char *path, int flags);
 | |
| void xunlink(char *path);
 | |
| void xrename(char *from, char *to);
 | |
| int xtempfile(char *name, char **tempname);
 | |
| int xcreate(char *path, int flags, int mode);
 | |
| int xopen(char *path, int flags);
 | |
| int xcreate_stdio(char *path, int flags, int mode);
 | |
| int xopen_stdio(char *path, int flags);
 | |
| int openro(char *path, int flags);
 | |
| int xopenro(char *path);
 | |
| void xpipe(int *pp);
 | |
| void xclose(int fd);
 | |
| int xdup(int fd);
 | |
| int notstdio(int fd);
 | |
| FILE *xfdopen(int fd, char *mode);
 | |
| FILE *xfopen(char *path, char *mode);
 | |
| size_t xread(int fd, void *buf, size_t len);
 | |
| void xreadall(int fd, void *buf, size_t len);
 | |
| void xwrite(int fd, void *buf, size_t len);
 | |
| off_t xlseek(int fd, off_t offset, int whence);
 | |
| char *xreadfile(char *name, char *buf, off_t len);
 | |
| int xioctl(int fd, int request, void *data);
 | |
| char *xgetcwd(void);
 | |
| void xstat(char *path, struct stat *st);
 | |
| char *xabspath(char *path, int exact);
 | |
| void xchdir(char *path);
 | |
| void xchroot(char *path);
 | |
| struct passwd *xgetpwuid(uid_t uid);
 | |
| struct group *xgetgrgid(gid_t gid);
 | |
| struct passwd *xgetpwnam(char *name);
 | |
| struct group *xgetgrnam(char *name);
 | |
| unsigned xgetuid(char *name);
 | |
| unsigned xgetgid(char *name);
 | |
| void xsetuser(struct passwd *pwd);
 | |
| char *xreadlinkat(int dir, char *name);
 | |
| char *xreadlink(char *name);
 | |
| double xstrtod(char *s);
 | |
| long xparsetime(char *arg, long units, long *fraction);
 | |
| void xparsetimespec(char *arg, struct timespec *ts);
 | |
| long long xparsemillitime(char *arg);
 | |
| void xpidfile(char *name);
 | |
| void xregcomp(regex_t *preg, char *rexec, int cflags);
 | |
| char *xtzset(char *new);
 | |
| void xsignal_flags(int signal, void *handler, int flags);
 | |
| void xsignal(int signal, void *handler);
 | |
| time_t xvali_date(struct tm *tm, char *str);
 | |
| void xparsedate(char *str, time_t *t, unsigned *nano, int endian);
 | |
| char *xgetline(FILE *fp);
 | |
| time_t xmktime(struct tm *tm, int utc);
 | |
| 
 | |
| // lib.c
 | |
| void verror_msg(char *msg, int err, va_list va);
 | |
| void error_msg(char *msg, ...) printf_format;
 | |
| void perror_msg(char *msg, ...) printf_format;
 | |
| _Noreturn void error_exit(char *msg, ...) printf_format;
 | |
| _Noreturn void perror_exit(char *msg, ...) printf_format;
 | |
| _Noreturn void help_exit(char *msg, ...) printf_format;
 | |
| void error_msg_raw(char *msg);
 | |
| void perror_msg_raw(char *msg);
 | |
| _Noreturn void error_exit_raw(char *msg);
 | |
| _Noreturn void perror_exit_raw(char *msg);
 | |
| ssize_t readall(int fd, void *buf, size_t len);
 | |
| ssize_t writeall(int fd, void *buf, size_t len);
 | |
| off_t lskip(int fd, off_t offset);
 | |
| #define MKPATHAT_MKLAST  1
 | |
| #define MKPATHAT_MAKE    2
 | |
| #define MKPATHAT_VERBOSE 4
 | |
| int mkpathat(int atfd, char *dir, mode_t lastmode, int flags);
 | |
| int mkpath(char *dir);
 | |
| struct string_list **splitpath(char *path, struct string_list **list);
 | |
| char *readfd(int fd, char *ibuf, off_t *plen);
 | |
| char *readfileat(int dirfd, char *name, char *buf, off_t *len);
 | |
| char *readfile(char *name, char *buf, off_t len);
 | |
| void msleep(long milliseconds);
 | |
| void nanomove(struct timespec *ts, long long offset);
 | |
| long long nanodiff(struct timespec *old, struct timespec *new);
 | |
| int highest_bit(unsigned long l);
 | |
| int64_t peek_le(void *ptr, unsigned size);
 | |
| int64_t peek_be(void *ptr, unsigned size);
 | |
| int64_t peek(void *ptr, unsigned size);
 | |
| void poke_le(void *ptr, long long val, unsigned size);
 | |
| void poke_be(void *ptr, long long val, unsigned size);
 | |
| void poke(void *ptr, long long val, unsigned size);
 | |
| struct string_list *find_in_path(char *path, char *filename);
 | |
| long long estrtol(char *str, char **end, int base);
 | |
| long long xstrtol(char *str, char **end, int base);
 | |
| long long atolx(char *c);
 | |
| long long atolx_range(char *numstr, long long low, long long high);
 | |
| int stridx(char *haystack, char needle);
 | |
| int wctoutf8(char *s, unsigned wc);
 | |
| int utf8towc(unsigned *wc, char *str, unsigned len);
 | |
| char *strlower(char *s);
 | |
| char *strafter(char *haystack, char *needle);
 | |
| char *chomp(char *s);
 | |
| int unescape(char c);
 | |
| int unescape2(char **c, int echo);
 | |
| char *strend(char *str, char *suffix);
 | |
| int strstart(char **a, char *b);
 | |
| int strcasestart(char **a, char *b);
 | |
| off_t fdlength(int fd);
 | |
| void loopfiles_rw(char **argv, int flags, int permissions,
 | |
|   void (*function)(int fd, char *name));
 | |
| void loopfiles(char **argv, void (*function)(int fd, char *name));
 | |
| void loopfiles_lines(char **argv, void (*function)(char **pline, long len));
 | |
| long long sendfile_len(int in, int out, long long len, long long *consumed);
 | |
| long long xsendfile_len(int in, int out, long long len);
 | |
| void xsendfile_pad(int in, int out, long long len);
 | |
| long long xsendfile(int in, int out);
 | |
| int wfchmodat(int rc, char *name, mode_t mode);
 | |
| int copy_tempfile(int fdin, char *name, char **tempname);
 | |
| void delete_tempfile(int fdin, int fdout, char **tempname);
 | |
| void replace_tempfile(int fdin, int fdout, char **tempname);
 | |
| void crc_init(unsigned *crc_table, int little_endian);
 | |
| void base64_init(char *p);
 | |
| int yesno(int def);
 | |
| int fyesno(FILE *fp, int def);
 | |
| int qstrcmp(const void *a, const void *b);
 | |
| void create_uuid(char *uuid);
 | |
| char *show_uuid(char *uuid);
 | |
| char *next_printf(char *s, char **start);
 | |
| struct passwd *bufgetpwnamuid(char *name, uid_t uid);
 | |
| struct passwd *bufgetpwuid(uid_t uid);
 | |
| struct group *bufgetgrnamgid(char *name, gid_t gid);
 | |
| struct group *bufgetgrgid(gid_t gid);
 | |
| int readlinkat0(int dirfd, char *path, char *buf, int len);
 | |
| int readlink0(char *path, char *buf, int len);
 | |
| int regexec0(regex_t *preg, char *string, long len, int nmatch,
 | |
|   regmatch_t pmatch[], int eflags);
 | |
| char *getusername(uid_t uid);
 | |
| char *getgroupname(gid_t gid);
 | |
| void do_lines(int fd, char delim, void (*call)(char **pline, long len));
 | |
| long long millitime(void);
 | |
| char *format_iso_time(char *buf, size_t len, struct timespec *ts);
 | |
| void loggit(int priority, char *format, ...);
 | |
| unsigned tar_cksum(void *data);
 | |
| int is_tar_header(void *pkt);
 | |
| char *elf_arch_name(int type);
 | |
| 
 | |
| #define HR_SPACE  1 // Space between number and units
 | |
| #define HR_B      2 // Use "B" for single byte units
 | |
| #define HR_1000   4 // Use decimal instead of binary units
 | |
| #define HR_NODOT  8 // No tenths for single digit units
 | |
| int human_readable_long(char *buf, unsigned long long num, int dgt, int unit,
 | |
|   int style);
 | |
| int human_readable(char *buf, unsigned long long num, int style);
 | |
| 
 | |
| // env.c
 | |
| 
 | |
| long environ_bytes(void);
 | |
| char *xsetenv(char *name, char *val);
 | |
| void xunsetenv(char *name);
 | |
| char *xpop_env(char *name); // because xpopenv() looks like xpopen_v()
 | |
| void xclearenv(void);
 | |
| void reset_env(struct passwd *p, int clear);
 | |
| 
 | |
| // utf8.c
 | |
| 
 | |
| int crunch_escape(FILE *out, int cols, int wc);
 | |
| int crunch_rev_escape(FILE *out, int cols, int wc);
 | |
| int crunch_str(char **str, int width, FILE *out, char *escmore,
 | |
|   int (*escout)(FILE *out, int cols, int wc));
 | |
| int draw_str(char *start, int width);
 | |
| int utf8len(char *str);
 | |
| int utf8skip(char *str, int width);
 | |
| int draw_trim_esc(char *str, int padto, int width, char *escmore,
 | |
|   int (*escout)(FILE *out, int cols,int wc));
 | |
| int draw_trim(char *str, int padto, int width);
 | |
| 
 | |
| // tty.c
 | |
| int tty_fd(void);
 | |
| int terminal_size(unsigned *xx, unsigned *yy);
 | |
| int terminal_probesize(unsigned *xx, unsigned *yy);
 | |
| #define KEY_UP 0
 | |
| #define KEY_DOWN 1
 | |
| #define KEY_RIGHT 2
 | |
| #define KEY_LEFT 3
 | |
| #define KEY_PGUP 4
 | |
| #define KEY_PGDN 5
 | |
| #define KEY_HOME 6
 | |
| #define KEY_END 7
 | |
| #define KEY_INSERT 8
 | |
| #define KEY_DELETE 9
 | |
| #define KEY_FN 10 // F1 = KEY_FN+1, F2 = KEY_FN+2, ...
 | |
| #define KEY_SHIFT (1<<16)
 | |
| #define KEY_CTRL (1<<17)
 | |
| #define KEY_ALT (1<<18)
 | |
| int scan_key(char *scratch, int timeout_ms);
 | |
| int scan_key_getsize(char *scratch, int timeout_ms, unsigned *xx, unsigned *yy);
 | |
| void xsetspeed(struct termios *tio, int speed);
 | |
| int set_terminal(int fd, int raw, int speed, struct termios *old);
 | |
| void xset_terminal(int fd, int raw, int speed, struct termios *old);
 | |
| void tty_reset(void);
 | |
| void tty_sigreset(int i);
 | |
| void start_redraw(unsigned *width, unsigned *height);
 | |
| 
 | |
| // net.c
 | |
| 
 | |
| union socksaddr {
 | |
|   struct sockaddr s;
 | |
|   struct sockaddr_in in;
 | |
|   struct sockaddr_in6 in6;
 | |
| };
 | |
| 
 | |
| int xsocket(int domain, int type, int protocol);
 | |
| void xsetsockopt(int fd, int level, int opt, void *val, socklen_t len);
 | |
| struct addrinfo *xgetaddrinfo(char *host, char *port, int family, int socktype,
 | |
|   int protocol, int flags);
 | |
| void xbind(int fd, const struct sockaddr *sa, socklen_t len);
 | |
| void xconnect(int fd, const struct sockaddr *sa, socklen_t len);
 | |
| int xconnectany(struct addrinfo *ai);
 | |
| int xbindany(struct addrinfo *ai);
 | |
| int xpoll(struct pollfd *fds, int nfds, int timeout);
 | |
| int pollinate(int in1, int in2, int out1, int out2, int timeout, int shutdown_timeout);
 | |
| char *ntop(struct sockaddr *sa);
 | |
| void xsendto(int sockfd, void *buf, size_t len, struct sockaddr *dest);
 | |
| int xrecvwait(int fd, char *buf, int len, union socksaddr *sa, int timeout);
 | |
| 
 | |
| // password.c
 | |
| int get_salt(char *salt, char * algo);
 | |
| 
 | |
| // commas.c
 | |
| void comma_args(struct arg_list *al, void *data, char *err,
 | |
|   char *(*callback)(void *data, char *str, int len));
 | |
| void comma_collate(char **old, char *new);
 | |
| char *comma_iterate(char **list, int *len);
 | |
| int comma_scan(char *optlist, char *opt, int clean);
 | |
| int comma_scanall(char *optlist, char *scanlist);
 | |
| int comma_remove(char *optlist, char *opt);
 | |
| 
 | |
| // deflate.c
 | |
| 
 | |
| long long gzip_fd(int infd, int outfd);
 | |
| long long gunzip_fd(int infd, int outfd);
 | |
| 
 | |
| // getmountlist.c
 | |
| struct mtab_list {
 | |
|   struct mtab_list *next, *prev;
 | |
|   struct stat stat;
 | |
|   struct statvfs statvfs;
 | |
|   char *dir;
 | |
|   char *device;
 | |
|   char *opts;
 | |
|   char type[0];
 | |
| };
 | |
| 
 | |
| int mountlist_istype(struct mtab_list  *ml, char *typelist);
 | |
| struct mtab_list *xgetmountlist(char *path);
 | |
| 
 | |
| // signal
 | |
| 
 | |
| void generic_signal(int signal);
 | |
| void exit_signal(int signal);
 | |
| void sigatexit(void *handler);
 | |
| void list_signals(void);
 | |
| 
 | |
| mode_t string_to_mode(char *mode_str, mode_t base);
 | |
| void mode_to_string(mode_t mode, char *buf);
 | |
| char *getbasename(char *name);
 | |
| char *fileunderdir(char *file, char *dir);
 | |
| char *relative_path(char *from, char *to);
 | |
| void names_to_pid(char **names, int (*callback)(pid_t pid, char *name),
 | |
|     int scripts);
 | |
| 
 | |
| pid_t __attribute__((returns_twice)) xvforkwrap(pid_t pid);
 | |
| #define XVFORK() xvforkwrap(vfork())
 | |
| 
 | |
| // Wrapper to make xfuncs() return (via siglongjmp) instead of exiting.
 | |
| // Assigns true/false "did it exit" value to first argument.
 | |
| #define WOULD_EXIT(y, x) do { sigjmp_buf _noexit; \
 | |
|   int _noexit_res; \
 | |
|   toys.rebound = &_noexit; \
 | |
|   _noexit_res = sigsetjmp(_noexit, 1); \
 | |
|   if (!_noexit_res) do {x;} while(0); \
 | |
|   toys.rebound = 0; \
 | |
|   y = _noexit_res; \
 | |
| } while(0)
 | |
| 
 | |
| // Wrapper that discards true/false "did it exit" value.
 | |
| #define NOEXIT(x) WOULD_EXIT(_noexit_res, x)
 | |
| 
 | |
| #define minof(a, b) ({typeof(a) aa = (a); typeof(b) bb = (b); aa<bb ? aa : bb;})
 | |
| #define maxof(a, b) ({typeof(a) aa = (a); typeof(b) bb = (b); aa>bb ? aa : bb;})
 | |
| 
 | |
| // Functions in need of further review/cleanup
 | |
| #include "lib/pending.h"
 |