0

我有以下程序正在运行,但问题是它仅在给出路径时才有效。如果没有给出路径,我正在尝试找到一种方法来设置当前工作目录的路径。为此,我正在使用 char *cdir = getcwd(0,0); 我需要找到一种方法将其设置为 argv,以便它指向该路径而不是 null。谁能检查我的代码并告诉我我做错了什么。我正在使用 unix 系统来编译它。

#include <dirent.h>
#include <sys/stat.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <dirent.h>
#include <unistd.h>


typedef struct stat Sub;
typedef struct dirent Dir;


void skimPath(const char *);

main(int argc, char *argv[])
{
    int i;
    Sub path;
    char *cdir = getcwd(0,0);

    if (argc <= 1)
    {
       /*
       this is the part i'm having trouble with, everything else works. I need a way to set  
       the path that is in cdir, to argv, so that it would work just like the case below      
       where argc is more than 2
       */

        argv = &cdir;
        printf("%s",argv);
        for (i = 1; i < argc; i++)
        {
            if (stat(*(argv + i), &path) == -1)
            {
                printf("WROTH PATH, DIRECTORY NOT FOUND \n%s\n", *(argv + i) ); 
                continue; 
            }

            if (S_ISDIR(path.st_mode)) 
                skimPath(*(argv + i));
        }
    }

    if (argc >= 2)
    {

        for (i = 1; i < argc; i++)
        {
            if (stat(*(argv + i), &path) == -1)
            {
                printf("WROTH PATH, DIRECTORY NOT FOUND \n%s\n", *(argv + i) ); 
                continue;
            }

            if (S_ISDIR(path.st_mode)) 
                skimPath(*(argv + i));

        }
    }
}


void skimPath(const char *dirName)
{
    char str[100];
    DIR *dir;
    Sub path;
    Dir *d;
    if ((dir = opendir(dirName)) == NULL)
    {
        printf(str, "File or Directory Could Not Open");
    }

    while ((d = readdir(dir)) != NULL)
    {
        // check if directory is d or d's paren   
        if (strcmp(d->d_name, ".") == 0 || strcmp(d->d_name, "..") == 0) 
            continue; // if ture rest will be ignored from while loop

        // saves in a buffer pointed by str
        sprintf(str, "%s/%s", dirName, d->d_name);


        if (stat(str, &path) == -1)
        {
            continue;
        }

        //checks to see if its a d
        if (S_ISDIR(path.st_mode))
        {
            printf("%s \n",d->d_name);

            // directory goes in str
            skimPath(str);

        }
    }   
}
4

1 回答 1

1

我们可以分析您的代码,但您尝试做的事情相当奇怪(详细地说——我认为还有其他方法可以做您想做的事情)。

main(int argc, char *argv[])
{
    int i;
    Sub path;
    char *cdir = getcwd(0,0);

由于您并不总是使用cdir,因此您可以 - 应该 - 在使用它的块内声明它。 getcwd()是一个昂贵的功能,特别是如果您有多个挂载的文件系统要处理,尤其是 NFS 挂载的文件系统。

    if (argc <= 1)
    {
       /*
       this is the part i'm having trouble with, everything else works. I need a way to set  
       the path that is in cdir, to argv, so that it would work just like the case below      
       where argc is more than 2
       */

        argv = &cdir;

这个声明是“合法的”,但你没有考虑过后果。您现在argv正好指向一个字符串(指针列表上没有空终止)并且argc现在无关紧要。

        printf("%s",argv);

这是错误的;它应该是以下几行之一:

printf("%s\n", argv[0]);
printf("%s\n", *argv);
printf("%s\n", cdir);

您已经删除了原始参数列表,剩下的唯一参数是当前目录。

        for (i = 1; i < argc; i++)
        {

由于argv现在指向cdir,因此您无法迭代参数。双重的,你不能从索引 1 开始这样做。

            if (stat(*(argv + i), &path) == -1)

是的,你可以argv[i]那样写,但你为什么要这样做?

            {
                printf("WROTH PATH, DIRECTORY NOT FOUND \n%s\n", *(argv + i) ); 
                continue; 
            }

虽然您会在一本像样的字典中找到“愤怒”(“adj (archaic):anger”),但您的意思可能是“错误的”。对人大喊大叫是不友善的。此外,错误消息最好打印为标准错误;这就是它的用途。如果你使用else(或else if)你可以避免continue.

            if (S_ISDIR(path.st_mode)) 
                skimPath(*(argv + i));
        }
    }

这么多还可以。

    if (argc >= 2)
    {
        ...
    }

我会把它写成else当前代码结构的一个子句。

由于如果用户明确地将名称作为参数传递,您将执行相同的操作.,因此如果用户没有提供当前目录作为第一个参数,则很容易通过提供当前目录作为第一个参数来伪造事物。因为当你输入 时main(),条件argv[argc] == NULL为真,你实际上可以这样写:

int main(int argc, char **argv)
{
    if (argc == 1)
        argv[argc++] = ".";

    assert(argc > 1);
    for (int i = 1; i < argc; i++)
    {
        ...code from the if (argc >= 2) part of your code...
    }
    return 0;
}

如果您需要插入多个参数,则必须经历更多的麻烦,更像是:

if (argc < XXX)
{
    static char *alt_argv[] = { 0, "xyz", "pqr", "abc", 0 };
    alt_argv[0] = argv[0];
    argv = alt_argv;
    argc = (sizeof(alt_argv) / sizeof(alt_argv[0])) - 1;
}

尽管来自其他人的警告,argc并且argvmain()函数中的局部变量,并且可以修改(小心)。修改 中的数据更接近狡猾argv[argc],但这取决于您的代码是使用空指针哨兵还是使用计数。如果您使用计数并且从不访问超出(修改的)argv数组端,那么您会没事的。如果您确实访问超出了结尾,那么您将在大多数 Unix 变体上践踏(或读取)您的环境变量。

如果您确实决定要使用当前目录的绝对路径名,那么您仍然可以调整我概述的方案来使用它。假设您在 Linux 或 BSD 派生平台上工作,您的版本getcwd()将在给定空指针时分配内存,因此您可以编写:

if (argc == 1)
    argv[argc++] = getcwd(NULL, 0);

唯一需要注意的是空指针:

for (i = 1; i < argc && argv[i] != NULL; i++)
    ...

当您需要在现实生活中完成这项工作而不是练习基本的系统调用时,请考虑使用nftw() 为您遍历目录层次结构。

于 2013-10-19T01:27:39.060 回答