1 /* lib.h - header file for lib directory 2 * 3 * Copyright 2006 Rob Landley <rob@landley.net> 4 */ 5 6 struct ptr_len { 7 void *ptr; 8 long len; 9 }; 10 11 struct str_len { 12 char *str; 13 long len; 14 }; 15 16 // llist.c 17 18 // All these list types can be handled by the same code because first element 19 // is always next pointer, so next = (mytype *)&struct. (The payloads are 20 // named differently to catch using the wrong type early.) 21 22 struct string_list { 23 struct string_list *next; 24 char str[0]; 25 }; 26 27 struct arg_list { 28 struct arg_list *next; 29 char *arg; 30 }; 31 32 struct double_list { 33 struct double_list *next, *prev; 34 char *data; 35 }; 36 37 struct num_cache { 38 struct num_cache *next; 39 long long num; 40 char data[]; 41 }; 42 43 void llist_free_arg(void *node); 44 void llist_free_double(void *node); 45 void llist_traverse(void *list, void (*using)(void *node)); 46 void *llist_pop(void *list); // actually void **list 47 void *dlist_pop(void *list); // actually struct double_list **list 48 void dlist_add_nomalloc(struct double_list **list, struct double_list *new); 49 struct double_list *dlist_add(struct double_list **list, char *data); 50 void *dlist_terminate(void *list); 51 struct num_cache *get_num_cache(struct num_cache *cache, long long num); 52 struct num_cache *add_num_cache(struct num_cache **cache, long long num, 53 void *data, int len); 54 55 // args.c 56 void get_optflags(void); 57 58 // dirtree.c 59 60 // Values returnable from callback function (bitfield, or them together) 61 // Default with no callback is 0 62 63 // Add this node to the tree 64 #define DIRTREE_SAVE 1 65 // Recurse into children 66 #define DIRTREE_RECURSE 2 67 // Call again after handling all children of this directory 68 // (Ignored for non-directories, sets linklen = -1 before second call.) 69 #define DIRTREE_COMEAGAIN 4 70 // Follow symlinks to directories 71 #define DIRTREE_SYMFOLLOW 8 72 // Don't warn about failure to stat 73 #define DIRTREE_SHUTUP 16 74 // Breadth first traversal, conserves filehandles at the expense of memory 75 #define DIRTREE_BREADTH 32 76 // skip non-numeric entries 77 #define DIRTREE_PROC 64 78 // Don't look at any more files in this directory. 79 #define DIRTREE_ABORT 256 80 81 #define DIRTREE_ABORTVAL ((struct dirtree *)1) 82 83 struct dirtree { 84 struct dirtree *next, *parent, *child; 85 long extra; // place for user to store their stuff (can be pointer) 86 struct stat st; 87 char *symlink; 88 int dirfd; 89 char again; 90 char name[]; 91 }; 92 93 int isdotdot(char *name); 94 struct dirtree *dirtree_add_node(struct dirtree *p, char *name, int flags); 95 char *dirtree_path(struct dirtree *node, int *plen); 96 int dirtree_notdotdot(struct dirtree *catch); 97 int dirtree_parentfd(struct dirtree *node); 98 int dirtree_recurse(struct dirtree *node, int (*callback)(struct dirtree *node), 99 int dirfd, int symfollow); 100 struct dirtree *dirtree_flagread(char *path, int flags, 101 int (*callback)(struct dirtree *node)); 102 struct dirtree *dirtree_read(char *path, int (*callback)(struct dirtree *node)); 103 104 // help.c 105 106 void show_help(FILE *out); 107 108 // Tell xopen and friends to print warnings but return -1 as necessary 109 // The largest O_BLAH flag so far is arch/alpha's O_PATH at 0x800000 so 110 // plenty of headroom. 111 #define WARN_ONLY (1<<31) 112 113 // xwrap.c 114 void xstrncpy(char *dest, char *src, size_t size); 115 void xstrncat(char *dest, char *src, size_t size); 116 void _xexit(void) noreturn; 117 void xexit(void) noreturn; 118 void *xmmap(void *addr, size_t length, int prot, int flags, int fd, off_t off); 119 void *xmalloc(size_t size); 120 void *xzalloc(size_t size); 121 void *xrealloc(void *ptr, size_t size); 122 char *xstrndup(char *s, size_t n); 123 char *xstrdup(char *s); 124 void *xmemdup(void *s, long len); 125 char *xmprintf(char *format, ...) printf_format; 126 void xprintf(char *format, ...) printf_format; 127 void xputs(char *s); 128 void xputc(char c); 129 void xflush(void); 130 void xexec(char **argv); 131 pid_t xpopen_both(char **argv, int *pipes); 132 int xwaitpid(pid_t pid); 133 int xpclose_both(pid_t pid, int *pipes); 134 pid_t xpopen(char **argv, int *pipe, int isstdout); 135 pid_t xpclose(pid_t pid, int pipe); 136 int xrun(char **argv); 137 int xpspawn(char **argv, int*pipes); 138 void xaccess(char *path, int flags); 139 void xunlink(char *path); 140 int xcreate(char *path, int flags, int mode); 141 int xopen(char *path, int flags); 142 int xcreate_stdio(char *path, int flags, int mode); 143 int xopen_stdio(char *path, int flags); 144 int openro(char *path, int flags); 145 int xopenro(char *path); 146 void xpipe(int *pp); 147 void xclose(int fd); 148 int xdup(int fd); 149 int notstdio(int fd); 150 FILE *xfdopen(int fd, char *mode); 151 FILE *xfopen(char *path, char *mode); 152 size_t xread(int fd, void *buf, size_t len); 153 void xreadall(int fd, void *buf, size_t len); 154 void xwrite(int fd, void *buf, size_t len); 155 off_t xlseek(int fd, off_t offset, int whence); 156 char *xreadfile(char *name, char *buf, off_t len); 157 int xioctl(int fd, int request, void *data); 158 char *xgetcwd(void); 159 void xstat(char *path, struct stat *st); 160 char *xabspath(char *path, int exact); 161 void xchdir(char *path); 162 void xchroot(char *path); 163 struct passwd *xgetpwuid(uid_t uid); 164 struct group *xgetgrgid(gid_t gid); 165 struct passwd *xgetpwnam(char *name); 166 struct group *xgetgrnam(char *name); 167 unsigned xgetuid(char *name); 168 unsigned xgetgid(char *name); 169 void xsetuser(struct passwd *pwd); 170 char *xreadlink(char *name); 171 double xstrtod(char *s); 172 long xparsetime(char *arg, long units, long *fraction); 173 void xpidfile(char *name); 174 void xregcomp(regex_t *preg, char *rexec, int cflags); 175 char *xtzset(char *new); 176 void xsignal(int signal, void *handler); 177 178 // lib.c 179 void verror_msg(char *msg, int err, va_list va); 180 void error_msg(char *msg, ...) printf_format; 181 void perror_msg(char *msg, ...) printf_format; 182 void error_exit(char *msg, ...) printf_format noreturn; 183 void perror_exit(char *msg, ...) printf_format noreturn; 184 void help_exit(char *msg, ...) printf_format noreturn; 185 void error_msg_raw(char *msg); 186 void perror_msg_raw(char *msg); 187 void error_exit_raw(char *msg); 188 void perror_exit_raw(char *msg); 189 ssize_t readall(int fd, void *buf, size_t len); 190 ssize_t writeall(int fd, void *buf, size_t len); 191 off_t lskip(int fd, off_t offset); 192 int mkpathat(int atfd, char *dir, mode_t lastmode, int flags); 193 struct string_list **splitpath(char *path, struct string_list **list); 194 char *readfileat(int dirfd, char *name, char *buf, off_t *len); 195 char *readfile(char *name, char *buf, off_t len); 196 void msleep(long miliseconds); 197 int highest_bit(unsigned long l); 198 int64_t peek_le(void *ptr, unsigned size); 199 int64_t peek_be(void *ptr, unsigned size); 200 int64_t peek(void *ptr, unsigned size); 201 void poke(void *ptr, uint64_t val, int size); 202 struct string_list *find_in_path(char *path, char *filename); 203 long long estrtol(char *str, char **end, int base); 204 long long xstrtol(char *str, char **end, int base); 205 long long atolx(char *c); 206 long long atolx_range(char *numstr, long long low, long long high); 207 int stridx(char *haystack, char needle); 208 int utf8towc(wchar_t *wc, char *str, unsigned len); 209 char *strlower(char *s); 210 char *strafter(char *haystack, char *needle); 211 char *chomp(char *s); 212 int unescape(char c); 213 char *strend(char *str, char *suffix); 214 int strstart(char **a, char *b); 215 off_t fdlength(int fd); 216 void loopfiles_rw(char **argv, int flags, int permissions, 217 void (*function)(int fd, char *name)); 218 void loopfiles(char **argv, void (*function)(int fd, char *name)); 219 void loopfiles_lines(char **argv, void (*function)(char **pline, long len)); 220 long long xsendfile(int in, int out); 221 int wfchmodat(int rc, char *name, mode_t mode); 222 int copy_tempfile(int fdin, char *name, char **tempname); 223 void delete_tempfile(int fdin, int fdout, char **tempname); 224 void replace_tempfile(int fdin, int fdout, char **tempname); 225 void crc_init(unsigned int *crc_table, int little_endian); 226 void base64_init(char *p); 227 int yesno(int def); 228 int qstrcmp(const void *a, const void *b); 229 void create_uuid(char *uuid); 230 char *show_uuid(char *uuid); 231 char *next_printf(char *s, char **start); 232 char *strnstr(char *line, char *str); 233 int dev_minor(int dev); 234 int dev_major(int dev); 235 int dev_makedev(int major, int minor); 236 struct passwd *bufgetpwuid(uid_t uid); 237 struct group *bufgetgrgid(gid_t gid); 238 int readlinkat0(int dirfd, char *path, char *buf, int len); 239 int readlink0(char *path, char *buf, int len); 240 int regexec0(regex_t *preg, char *string, long len, int nmatch, 241 regmatch_t pmatch[], int eflags); 242 char *getusername(uid_t uid); 243 char *getgroupname(gid_t gid); 244 void do_lines(int fd, void (*call)(char **pline, long len)); 245 long environ_bytes(); 246 long long millitime(void); 247 248 #define HR_SPACE 1 // Space between number and units 249 #define HR_B 2 // Use "B" for single byte units 250 #define HR_1000 4 // Use decimal instead of binary units 251 int human_readable(char *buf, unsigned long long num, int style); 252 253 // linestack.c 254 255 struct linestack { 256 long len, max; 257 struct ptr_len idx[]; 258 }; 259 260 void linestack_addstack(struct linestack **lls, struct linestack *throw, 261 long pos); 262 void linestack_insert(struct linestack **lls, long pos, char *line, long len); 263 void linestack_append(struct linestack **lls, char *line); 264 struct linestack *linestack_load(char *name); 265 int crunch_escape(FILE *out, int cols, int wc); 266 int crunch_rev_escape(FILE *out, int cols, int wc); 267 int crunch_str(char **str, int width, FILE *out, char *escmore, 268 int (*escout)(FILE *out, int cols, int wc)); 269 int draw_str(char *start, int width); 270 int utf8len(char *str); 271 int utf8skip(char *str, int width); 272 int draw_trim_esc(char *str, int padto, int width, char *escmore, 273 int (*escout)(FILE *out, int cols,int wc)); 274 int draw_trim(char *str, int padto, int width); 275 276 // interestingtimes.c 277 int tty_fd(void); 278 int terminal_size(unsigned *xx, unsigned *yy); 279 int terminal_probesize(unsigned *xx, unsigned *yy); 280 int scan_key_getsize(char *scratch, int miliwait, unsigned *xx, unsigned *yy); 281 int set_terminal(int fd, int raw, struct termios *old); 282 void xset_terminal(int fd, int raw, struct termios *old); 283 int scan_key(char *scratch, int miliwait); 284 void tty_esc(char *s); 285 void tty_jump(int x, int y); 286 void tty_reset(void); 287 void tty_sigreset(int i); 288 289 // net.c 290 int xsocket(int domain, int type, int protocol); 291 void xsetsockopt(int fd, int level, int opt, void *val, socklen_t len); 292 struct addrinfo *xgetaddrinfo(char *host, char *port, int family, int socktype, 293 int protocol, int flags); 294 int xconnect(struct addrinfo *ai_arg); 295 int xpoll(struct pollfd *fds, int nfds, int timeout); 296 int pollinate(int in1, int in2, int out1, int out2, int timeout, int shutdown_timeout); 297 char *ntop(struct sockaddr *sa); 298 299 // password.c 300 int get_salt(char *salt, char * algo); 301 302 // getmountlist.c 303 struct mtab_list { 304 struct mtab_list *next, *prev; 305 struct stat stat; 306 struct statvfs statvfs; 307 char *dir; 308 char *device; 309 char *opts; 310 char type[0]; 311 }; 312 313 void comma_args(struct arg_list *al, void *data, char *err, 314 char *(*callback)(void *data, char *str, int len)); 315 void comma_collate(char **old, char *new); 316 char *comma_iterate(char **list, int *len); 317 int comma_scan(char *optlist, char *opt, int clean); 318 int comma_scanall(char *optlist, char *scanlist); 319 int mountlist_istype(struct mtab_list *ml, char *typelist); 320 struct mtab_list *xgetmountlist(char *path); 321 322 // signal 323 324 void generic_signal(int signal); 325 void exit_signal(int signal); 326 void sigatexit(void *handler); 327 int sig_to_num(char *pidstr); 328 char *num_to_sig(int sig); 329 330 mode_t string_to_mode(char *mode_str, mode_t base); 331 void mode_to_string(mode_t mode, char *buf); 332 char *getbasename(char *name); 333 void names_to_pid(char **names, int (*callback)(pid_t pid, char *name)); 334 335 pid_t __attribute__((returns_twice)) xvforkwrap(pid_t pid); 336 #define XVFORK() xvforkwrap(vfork()) 337 338 // Wrapper to make xfuncs() return (via longjmp) instead of exiting. 339 // Assigns true/false "did it exit" value to first argument. 340 #define WOULD_EXIT(y, x) do { jmp_buf _noexit; \ 341 int _noexit_res; \ 342 toys.rebound = &_noexit; \ 343 _noexit_res = setjmp(_noexit); \ 344 if (!_noexit_res) do {x;} while(0); \ 345 toys.rebound = 0; \ 346 y = _noexit_res; \ 347 } while(0); 348 349 // Wrapper that discards true/false "did it exit" value. 350 #define NOEXIT(x) WOULD_EXIT(_noexit_res, x) 351 352 #define minof(a, b) ({typeof(a) aa = (a); typeof(b) bb = (b); aa<bb ? aa : bb;}) 353 #define maxof(a, b) ({typeof(a) aa = (a); typeof(b) bb = (b); aa>bb ? aa : bb;}) 354 355 // Functions in need of further review/cleanup 356 #include "lib/pending.h" 357