執行系統命令,並返回輸出的結果
首先需要了解mkstemp():
mkstemp()函數在系統中以唯一的文件名創建一個文件並打開,而且只有當前用戶才能訪問這個
臨時文件,並進行讀、寫操作。
mkstemp函數在系統中以唯一的文件名創建一個文件並打開,而且只有當前用戶才能訪問這個
臨時文件,並進行讀、寫操作。 mkstemp函數只有一個參數,這個參數是個以“XXXXXX”結尾的非
空字符串。mkstemp函數會用隨機產生的字符串替換“XXXXXX”,保證 了文件名的唯一性。 函數返回一個
文件描述符,如果執行失敗返回-1。
在glibc 2.0.6 以及更早的
glibc庫中這個文件的訪問權限是0666,glibc 2.0.7以后的庫這個文件的訪問權限是0600。
臨時文件使用完成后應及時刪除,否則臨時文件目錄會塞滿垃圾。由於mkstemp函數創建的
臨時文件不能自動刪除,所以執行完 mkstemp函數后要調用
unlink函數,unlink函數刪除文件的目錄入口,但臨時文件還可以通過文件描述符進行訪問,直到最后一個打開的進程關 閉文件
操作符,或者程序退出后臨時文件被自動徹底地刪除。
函數功能表示:讀取命令執行結果,放到臨時文件,並將結果讀到buf中,刪除臨時文件
/**
*@brief 執行系統命令,並返回輸出的結果
*@param[in] cmdstring,命令串
*@param[in] buf,存放命令結果的緩沖區
*@param[in] size,緩沖區的大小
*@param[out]
*@return 返回寫入到buf中的字符串長度,不含\0 ; -1: 失敗;
*@remark
*@version V1.0.0
*@note buf中最多返回size-1個字符,字符串始終以\0結尾。
*/
int get_cmd_results(const char *cmdstring, char *buff, int size)
{
char cmd_string[200] = {0};
char tmpfile[100] = {0};
char tmp_buf[100] = {0};
int fd;
int tmp_fd;
int nbytes;
memset(buff, 0, size);
if((cmdstring == NULL) ||
(strlen(cmdstring) > (sizeof(tmpfile) + 8)) ||
((strlen(cmdstring) + strlen(tmpfile) + 5) > sizeof(cmd_string)))
{
printf("cmd is too long or NULL!\n");
return -1;
}
sscanf(cmdstring, "%[a-Z]", tmp_buf);/*%[a-z] 表示匹配a到z中任意字符,貪婪性(盡可能多的匹配) */
sprintf(tmpfile, "/tmp/%s-XXXXXX", tmp_buf);
tmp_fd = mkstemp(tmpfile);
if(tmp_fd < 0)
{
printf("mkstemp failed\n");
return -1;
}
close(tmp_fd);
sprintf(cmd_string, "%s > %s 2>&1", cmdstring, tmpfile);/*標准輸出(1),標准錯誤(2)都輸出到臨時文件*/
if(system_ex(cmd_string, 20) < 0)
{
printf("run \"%s\" ret < 0!\n", cmd_string);
}
fd = open(tmpfile, O_RDONLY);
if(fd < 0)
{
printf("open %s failed!\n", tmpfile);
nbytes = -1;
}
else
{
nbytes = read(fd, buff, size - 1);
close(fd);
}
memset(cmd_string, 0, sizeof(cmd_string));
sprintf(cmd_string, "rm -rf /tmp/%s-*", tmp_buf);
system_ex(cmd_string, 20);
return nbytes;
}
system_ex:
首先請了解信號:signal和sigaction
https://blog.csdn.net/weibo1230123/article/details/81411827
https://www.cnblogs.com/lidabo/p/4581065.html
創建子進程執行命令,父進程忽略ignore SIGINT and SIGQUIT ,來避免影響子進程執行即需要子進程執行完畢,子進程不忽略信號,當超時或者子進程退出
則恢復信號狀態,超時的話,還需要kill掉子進程等
//system函數擴展,加入超時值(0表示永久等待)
//超時時返回-2,其他情況返回不變。
int system_ex(const char *cmdstring, unsigned int timeout) /* with appropriate signal handling */
{
pid_t pid;
int status;
struct sigaction ignore, saveintr, savequit;
sigset_t chldmask, savemask;
//精度換成十分之一秒
timeout *= 10;
if (timeout == 0)
timeout = 0xFFFFFFFF;
if (cmdstring == NULL)
return(1); /* always a command processor with UNIX */
ignore.sa_handler = SIG_IGN; /* ignore SIGINT and SIGQUIT */
sigemptyset(&ignore.sa_mask);
ignore.sa_flags = 0;
if (sigaction(SIGINT, &ignore, &saveintr) < 0)
return(-1);
if (sigaction(SIGQUIT, &ignore, &savequit) < 0)
return(-1);
sigemptyset(&chldmask); /* now block SIGCHLD */
sigaddset(&chldmask, SIGCHLD);
if (sigprocmask(SIG_BLOCK, &chldmask, &savemask) < 0)
return(-1);
if ((pid = fork()) < 0) {
return -1; /* probably out of processes */
} else if (pid == 0) { /* child */
/* restore previous signal actions & reset signal mask */
sigaction(SIGINT, &saveintr, NULL);
sigaction(SIGQUIT, &savequit, NULL);
sigprocmask(SIG_SETMASK, &savemask, NULL);
/*通常exec會放在fork() 函數的子進程部分, 來替代子進程執行啦, 執行成功后子程序就會消失, 但是執行失敗的話, 必須用exit()函數來讓子進程退出!*/
execl("/bin/sh", "sh", "-c", cmdstring, (char *)0);/*exec函數會取代執行它的進程, 也就是說, 一旦exec函數執行成功, 它就不會返回了, 進程結束. 但是如果exec函數執行失敗, 它會返回失敗的信息, 而且進程繼續執行后面的代碼!*/
_exit(127); /* exec error */
}
/* parent */
int ret = 0;
while (timeout-- > 0 &&
(ret = waitpid(pid, &status, WNOHANG)) == 0)
usleep(100*1000);
/* restore previous signal actions & reset signal mask */
if (sigaction(SIGINT, &saveintr, NULL) < 0)
return(-1);
if (sigaction(SIGQUIT, &savequit, NULL) < 0)
return(-1);
if (sigprocmask(SIG_SETMASK, &savemask, NULL) < 0)
return(-1);
if (ret < 0)
return -1;
if (ret > 0)
return status;
kill(pid, SIGKILL);
waitpid(pid, &status, 0);
return -2;
}
