apue.h
/* * Our own header, to be included before all standard system headers. */ #ifndef _APUE_H #define _APUE_H #define _POSIX_C_SOURCE 200809L #if defined(SOLARIS) /* Solaris 10 */ #define _XOPEN_SOURCE 600 #else #define _XOPEN_SOURCE 700 #endif #include <sys/types.h> /* some systems still require this */ #include <sys/stat.h> #include <sys/termios.h> /* for winsize */ #if defined(MACOS) || !defined(TIOCGWINSZ) #include <sys/ioctl.h> #endif #include <stdio.h> /* for convenience */ #include <stdlib.h> /* for convenience */ #include <stddef.h> /* for offsetof */ #include <string.h> /* for convenience */ #include <unistd.h> /* for convenience */ #include <signal.h> /* for SIG_ERR */ #define MAXLINE 4096 /* max line length */ /* * Default file access permissions for new files. */ #define FILE_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH) /* * Default permissions for new directories. */ #define DIR_MODE (FILE_MODE | S_IXUSR | S_IXGRP | S_IXOTH) typedef void Sigfunc(int); /* for signal handlers */ #define min(a,b) ((a) < (b) ? (a) : (b)) #define max(a,b) ((a) > (b) ? (a) : (b)) /* * Prototypes for our own functions. */ char *path_alloc(size_t *); /* {Prog pathalloc} */ long open_max(void); /* {Prog openmax} */ int set_cloexec(int); /* {Prog setfd} */ void clr_fl(int, int); void set_fl(int, int); /* {Prog setfl} */ void pr_exit(int); /* {Prog prexit} */ void pr_mask(const char *); /* {Prog prmask} */ Sigfunc *signal_intr(int, Sigfunc *); /* {Prog signal_intr_function} */ void daemonize(const char *); /* {Prog daemoninit} */ void sleep_us(unsigned int); /* {Ex sleepus} */ ssize_t readn(int, void *, size_t); /* {Prog readn_writen} */ ssize_t writen(int, const void *, size_t); /* {Prog readn_writen} */ int fd_pipe(int *); /* {Prog sock_fdpipe} */ int recv_fd(int, ssize_t (*func)(int, const void *, size_t)); /* {Prog recvfd_sockets} */ int send_fd(int, int); /* {Prog sendfd_sockets} */ int send_err(int, int, const char *); /* {Prog senderr} */ int serv_listen(const char *); /* {Prog servlisten_sockets} */ int serv_accept(int, uid_t *); /* {Prog servaccept_sockets} */ int cli_conn(const char *); /* {Prog cliconn_sockets} */ int buf_args(char *, int (*func)(int, char **)); /* {Prog bufargs} */ int tty_cbreak(int); /* {Prog raw} */ int tty_raw(int); /* {Prog raw} */ int tty_reset(int); /* {Prog raw} */ void tty_atexit(void); /* {Prog raw} */ struct termios *tty_termios(void); /* {Prog raw} */ int ptym_open(char *, int); /* {Prog ptyopen} */ int ptys_open(char *); /* {Prog ptyopen} */ #ifdef TIOCGWINSZ pid_t pty_fork(int *, char *, int, const struct termios *, const struct winsize *); /* {Prog ptyfork} */ #endif int lock_reg(int, int, int, off_t, int, off_t); /* {Prog lockreg} */ #define read_lock(fd, offset, whence, len) \ lock_reg((fd), F_SETLK, F_RDLCK, (offset), (whence), (len)) #define readw_lock(fd, offset, whence, len) \ lock_reg((fd), F_SETLKW, F_RDLCK, (offset), (whence), (len)) #define write_lock(fd, offset, whence, len) \ lock_reg((fd), F_SETLK, F_WRLCK, (offset), (whence), (len)) #define writew_lock(fd, offset, whence, len) \ lock_reg((fd), F_SETLKW, F_WRLCK, (offset), (whence), (len)) #define un_lock(fd, offset, whence, len) \ lock_reg((fd), F_SETLK, F_UNLCK, (offset), (whence), (len)) pid_t lock_test(int, int, off_t, int, off_t); /* {Prog locktest} */ #define is_read_lockable(fd, offset, whence, len) \ (lock_test((fd), F_RDLCK, (offset), (whence), (len)) == 0) #define is_write_lockable(fd, offset, whence, len) \ (lock_test((fd), F_WRLCK, (offset), (whence), (len)) == 0) void err_msg(const char *, ...); /* {App misc_source} */ void err_dump(const char *, ...) __attribute__((noreturn)); void err_quit(const char *, ...) __attribute__((noreturn)); void err_cont(int, const char *, ...); void err_exit(int, const char *, ...) __attribute__((noreturn)); void err_ret(const char *, ...); void err_sys(const char *, ...) __attribute__((noreturn)); void log_msg(const char *, ...); /* {App misc_source} */ void log_open(const char *, int, int); void log_quit(const char *, ...) __attribute__((noreturn)); void log_ret(const char *, ...); void log_sys(const char *, ...) __attribute__((noreturn)); void log_exit(int, const char *, ...) __attribute__((noreturn)); void TELL_WAIT(void); /* parent/child from {Sec race_conditions} */ void TELL_PARENT(pid_t); void TELL_CHILD(pid_t); void WAIT_PARENT(void); void WAIT_CHILD(void); #endif /* _APUE_H */
#include <errno.h> /* for definition of errno */ #include <stdarg.h> /* ISO C variable aruments */ static void err_doit(int, int, const char *, va_list); /* * Nonfatal error related to a system call. * Print a message and return. */ void err_ret(const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(1, errno, fmt, ap); va_end(ap); } /* * Fatal error related to a system call. * Print a message and terminate. */ void err_sys(const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(1, errno, fmt, ap); va_end(ap); exit(1); } /* * Fatal error unrelated to a system call. * Error code passed as explict parameter. * Print a message and terminate. */ void err_exit(int error, const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(1, error, fmt, ap); va_end(ap); exit(1); } /* * Fatal error related to a system call. * Print a message, dump core, and terminate. */ void err_dump(const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(1, errno, fmt, ap); va_end(ap); abort(); /* dump core and terminate */ exit(1); /* shouldn't get here */ } /* * Nonfatal error unrelated to a system call. * Print a message and return. */ void err_msg(const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(0, 0, fmt, ap); va_end(ap); } /* * Fatal error unrelated to a system call. * Print a message and terminate. */ void err_quit(const char *fmt, ...) { va_list ap; va_start(ap, fmt); err_doit(0, 0, fmt, ap); va_end(ap); exit(1); } /* * Print a message and return to caller. * Caller specifies "errnoflag". */ static void err_doit(int errnoflag, int error, const char *fmt, va_list ap) { char buf[MAXLINE]; vsnprintf(buf, MAXLINE, fmt, ap); if (errnoflag) snprintf(buf+strlen(buf), MAXLINE-strlen(buf), ": %s", strerror(error)); strcat(buf, "\n"); fflush(stdout); /* in case stdout and stderr are the same */ fputs(buf, stderr); fflush(NULL); /* flushes all stdio output streams */ }
#include "apue.h" #include "my_err.h" #include<dirent.h> int main(int argc, char *argv[]) { DIR *dp; struct dirent *dirp; if (argc != 2) err_quit("usage: ls directory_name"); if ((dp = opendir(argv[1])) == NULL) err_sys("cant't open %s", argv[1]); while ((dirp = readdir(dp)) != NULL) printf("%s\n", dirp->d_name); closedir(dp); exit(0); }
1-3.c程序,必须指出,apue.h 中包含的出错例程(函数)仅用于将信息输出到标准错误;
当程序作为守护进程运行时,需要有另外的出错例程(函数)来处理守护进程可能输出的信息,例如与 syslog 进程通信并记录到日志等,
限于篇幅,这里没有整合相应的源码,有兴趣的童鞋,请参考
《UNIX 环境高级编程(第三版)》一书中,第727页开始的内容
my_LS_command_implement.c 程序的功能很简单,它通过用户在 shell 命令行中指定的绝对路径,来列出该目录下的内容
该程序第7,8行分别定义了 DIR 结构型指针与 dirent 结构型指针,用于指向后面
opendir() 与 readdir() 函数返回的内容;
1
2
|
DIR *dp;
struct
dirent *dirp;
|
第10~11行首先判断,用户输入的 shell 命令行参数个数(命令行参数个数通过 main 函数的第一个参数 argc 传递;命令行参数内容通过 main 函数的第二个参数 argv[] 传递)是否有2个,如果没有2个,则调用作者编写的标准出错例程提醒用户:必须指定要打开的目录;
1
2
|
if
(argc != 2)
err_quit(
"usage: ls directory_name"
);
|
第13~14行调用库函数 opendir 打开用户指定的目录,该目录通过 argv[1] 传递给 opendir(),通过将后者返回的值(DIR结构型变量)赋给 dp 并判断:如果 dp 为空指针则说明打开目录失败,此时调用作者编写的标准出错例程提醒用户:无法打开指定的目录;
1
2
|
if
((dp = opendir(argv[1])) == NULL)
err_sys(
"can't open %s"
, argv[1]);
|
第15~16行将指向 opendir() 成功打开的目录( DIR 结构型变量)的指针 dp 作为参数传递并在一个 while 循环中反复调用库函数 readdir ,将后者返回的值(dirent结构型变量)赋给 dirp 指针并判断:如果 readdir() 读取完目录中的所有文件和子目录,此时 dirp 为空指针,退出循环;反之,每当 readdir() 成功读取目录中的一个文件或子目录,就打印该文件或目录的名称(通过访问 dirp 指向的 dirent结构型的 d_name 成员,该成员存储名称)
1
2
|
while
((dirp = readdir(dp)) != NULL)
printf
(
"%s\n"
, dirp->d_name);
|
第18行将 dp 指针作为参数传递并调用库函数 closedir,用来关闭 opendir()打开的目录
1
|
closedir(dp);
|
第19行使用标准的退出程序方式:以参数0调用 exit 函数,表示没有错误,正常退出
1
|
exit
(0);
|