如果没有给出路径,则获取给定路径的目录和控制台或使用当前工作目录

时间:2013-10-18 22:27:24

标签: c

我有以下程序正在运行,但问题是它只在给出路径时才有效。我正在尝试找到一种方法来设置当前工作目录的路径,如果没有给出路径。为此,我使用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);

        }
    }   
}

1 个答案:

答案 0 :(得分: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; 
            }

虽然你会在一本体面的词典中找到'wroth'('adj(archaic):angry'),但你可能意味着'错'。对人们的抨击是不仁慈的。此外,错误消息最好打印到标准错误;这就是它的用途。如果您使用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;
}

尽管有其他人发出警告,但argcargvmain()函数中的局部变量,可以进行修改(仔细)。修改argv[argc]中的数据更加狡猾,但这取决于您的代码是使用空指针Sentinel还是使用计数。如果您使用计数并且永远不会访问(修改的)argv数组末尾,那么您会没事的。如果你进行了超越结束的访问,你就会在大多数Unix变种上踩踏(或读取)你的环境变量。

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

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

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

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

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