FreeBSD源码学习(2):echo命令源码分析

 

   今天给大家看一个Linux/Unix下面最最简单的echo命令的源码。

   echo命令的功能是在显示器上显示一段文字，一般起到一个提示的作用。

   该命令的一般格式为： echo [ -n ] 字符串

   其中选项n表示输出文字后不换行；字符串能加引号，也能不加引号。用echo命令输出加引号的字符串时，将字符串原样输出；用echo命令输出不加引号的字符串时，将字符串中的各个单词作为字符串输出，各字符串之间用一个空格分割。

   另外，在FreeBSD下echo命令的最后一个参数以/c结尾的效果以-n选项相同，都是输出后不换行。

 

 

/*输入参数存在动态生成的iovec结构体数组中*/ struct iovec { void *iov_base; /* Base address. */ size_t iov_len; /* Length. */ }; int main(int argc, char *argv[]) { int nflag; /* if not set, output a trailing newline. */ int veclen; /* number of writev arguments. */ struct iovec *iov, *vp; /* Elements to write, current element. */ char space[] = " "; char newline[] = "/n"; char *progname = argv[0]; /* 查找是否有-n选项*/ if (*++argv && !strcmp(*argv, "-n")) { ++argv; --argc; nflag = 1; } else nflag = 0; /*空格也算做字符串*/ veclen = (argc >= 2) ? (argc - 2) * 2 + 1 : 0; if ((vp = iov = malloc((veclen + 1) * sizeof(struct iovec))) == NULL) errexit(progname, "malloc"); while (argv[0] != NULL) { size_t len; len = strlen(argv[0]); /* * 如果下个参数为空，则当前参数是最后一个参数 * 检查这个参数是否以/c结束 */ if (argv[1] == NULL) { /* is there room for a '/c' and is there one? */ if (len >= 2 && argv[0][len - 2] == '//' && argv[0][len - 1] == 'c') { /* chop it and set the no-newline flag. */ len -= 2; nflag = 1; } } vp->iov_base = *argv; vp++->iov_len = len; if (*++argv) { //如果还有字符串就中间添加一个空格 vp->iov_base = space; vp++->iov_len = 1; } } /*有换行标志，输出换行符*/ if (!nflag) { veclen++; vp->iov_base = newline; vp++->iov_len = 1; } /* assert(veclen == (vp - iov)); */ while (veclen) { int nwrite; //每次最大输出1024个字符串，大于的分次输出 nwrite = (veclen > IOV_MAX) ? IOV_MAX : veclen; if (writev(STDOUT_FILENO, iov, nwrite) == -1) errexit(progname, "write"); iov += nwrite; veclen -= nwrite; } return 0; }