GNU/Linux >> Znalost Linux >  >> Linux

Jak mohu vytvořit strom adresářů v C++/Linux?

system("mkdir -p /tmp/a/b/c")

je nejkratší způsob, který mě napadá (z hlediska délky kódu, ne nutně doby provádění).

Není to multiplatformní, ale bude fungovat pod Linuxem.


Snadné s Boost.Filesystem:create_directories

#include <boost/filesystem.hpp>
//...
boost::filesystem::create_directories("/tmp/a/b/c");

Vrací:true pokud byl vytvořen nový adresář, jinak false .


Zde je můj příklad kódu (funguje pro Windows i Linux):

#include <iostream>
#include <string>
#include <sys/stat.h> // stat
#include <errno.h>    // errno, ENOENT, EEXIST
#if defined(_WIN32)
#include <direct.h>   // _mkdir
#endif

bool isDirExist(const std::string& path)
{
#if defined(_WIN32)
    struct _stat info;
    if (_stat(path.c_str(), &info) != 0)
    {
        return false;
    }
    return (info.st_mode & _S_IFDIR) != 0;
#else 
    struct stat info;
    if (stat(path.c_str(), &info) != 0)
    {
        return false;
    }
    return (info.st_mode & S_IFDIR) != 0;
#endif
}

bool makePath(const std::string& path)
{
#if defined(_WIN32)
    int ret = _mkdir(path.c_str());
#else
    mode_t mode = 0755;
    int ret = mkdir(path.c_str(), mode);
#endif
    if (ret == 0)
        return true;

    switch (errno)
    {
    case ENOENT:
        // parent didn't exist, try to create it
        {
            int pos = path.find_last_of('/');
            if (pos == std::string::npos)
#if defined(_WIN32)
                pos = path.find_last_of('\\');
            if (pos == std::string::npos)
#endif
                return false;
            if (!makePath( path.substr(0, pos) ))
                return false;
        }
        // now, try to create again
#if defined(_WIN32)
        return 0 == _mkdir(path.c_str());
#else 
        return 0 == mkdir(path.c_str(), mode);
#endif

    case EEXIST:
        // done!
        return isDirExist(path);

    default:
        return false;
    }
}

int main(int argc, char* ARGV[])
{
    for (int i=1; i<argc; i++)
    {
        std::cout << "creating " << ARGV[i] << " ... " << (makePath(ARGV[i]) ? "OK" : "failed") << std::endl;
    }
    return 0;
}

Použití:

$ makePath 1/2 folderA/folderB/folderC
creating 1/2 ... OK
creating folderA/folderB/folderC ... OK

S C++17 nebo novějším je standardní hlavička <filesystem> s funkcístd::filesystem::create_directories které by se měly používat v moderních programech C++. Standardní funkce C++ však nemají argument explicitních oprávnění (režim) specifický pro POSIX.

Zde je však funkce C, kterou lze zkompilovat pomocí kompilátorů C++.

/*
@(#)File:           mkpath.c
@(#)Purpose:        Create all directories in path
@(#)Author:         J Leffler
@(#)Copyright:      (C) JLSS 1990-2020
@(#)Derivation:     mkpath.c 1.16 2020/06/19 15:08:10
*/

/*TABSTOP=4*/

#include "posixver.h"
#include "mkpath.h"
#include "emalloc.h"

#include <errno.h>
#include <string.h>
/* "sysstat.h" == <sys/stat.h> with fixup for (old) Windows - inc mode_t */
#include "sysstat.h"

typedef struct stat Stat;

static int do_mkdir(const char *path, mode_t mode)
{
    Stat            st;
    int             status = 0;

    if (stat(path, &st) != 0)
    {
        /* Directory does not exist. EEXIST for race condition */
        if (mkdir(path, mode) != 0 && errno != EEXIST)
            status = -1;
    }
    else if (!S_ISDIR(st.st_mode))
    {
        errno = ENOTDIR;
        status = -1;
    }

    return(status);
}

/**
** mkpath - ensure all directories in path exist
** Algorithm takes the pessimistic view and works top-down to ensure
** each directory in path exists, rather than optimistically creating
** the last element and working backwards.
*/
int mkpath(const char *path, mode_t mode)
{
    char           *pp;
    char           *sp;
    int             status;
    char           *copypath = STRDUP(path);

    status = 0;
    pp = copypath;
    while (status == 0 && (sp = strchr(pp, '/')) != 0)
    {
        if (sp != pp)
        {
            /* Neither root nor double slash in path */
            *sp = '\0';
            status = do_mkdir(copypath, mode);
            *sp = '/';
        }
        pp = sp + 1;
    }
    if (status == 0)
        status = do_mkdir(path, mode);
    FREE(copypath);
    return (status);
}

#ifdef TEST

#include <stdio.h>
#include <unistd.h>

/*
** Stress test with parallel running of mkpath() function.
** Before the EEXIST test, code would fail.
** With the EEXIST test, code does not fail.
**
** Test shell script
** PREFIX=mkpath.$$
** NAME=./$PREFIX/sa/32/ad/13/23/13/12/13/sd/ds/ww/qq/ss/dd/zz/xx/dd/rr/ff/ff/ss/ss/ss/ss/ss/ss/ss/ss
** : ${MKPATH:=mkpath}
** ./$MKPATH $NAME &
** [...repeat a dozen times or so...]
** ./$MKPATH $NAME &
** wait
** rm -fr ./$PREFIX/
*/

int main(int argc, char **argv)
{
    int             i;

    for (i = 1; i < argc; i++)
    {
        for (int j = 0; j < 20; j++)
        {
            if (fork() == 0)
            {
                int rc = mkpath(argv[i], 0777);
                if (rc != 0)
                    fprintf(stderr, "%d: failed to create (%d: %s): %s\n",
                            (int)getpid(), errno, strerror(errno), argv[i]);
                exit(rc == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
            }
        }
        int status;
        int fail = 0;
        while (wait(&status) != -1)
        {
            if (WEXITSTATUS(status) != 0)
                fail = 1;
        }
        if (fail == 0)
            printf("created: %s\n", argv[i]);
    }
    return(0);
}

#endif /* TEST */

Makra STRDUP() a FREE() jsou verze pro kontrolu chyb strdup() a free() , deklarovaný v emalloc.h (a implementováno vemalloc.c a estrdup.c )."sysstat.h" hlavička se zabývá nefunkčními verzemi <sys/stat.h> a lze jej nahradit <sys/stat.h> na moderních unixových systémech (ale v roce 1990 bylo mnoho problémů). A "mkpath.h" deklaruje mkpath() .

Změna mezi v1.12 (původní verze odpovědi) a v1.13 (pozměněná verze odpovědi) byla testem pro EEXIST vdo_mkdir() .Switch na to upozornil, že je to nutné – děkuji, Switch. Testovací kód byl aktualizován a reprodukoval problém na MacBookPro (2,3 GHz Intel Core i7, se systémem Mac OS X 10.7.4) a naznačuje, že problém je vyřešen v revize (ale testování může ukázat pouze přítomnost chyb, nikdy jejich nepřítomnost). Zobrazený kód je nyní v1.16; od verze 1.13 byly provedeny kosmetické nebo administrativní změny (např. použití mkpath.h místo jlss.h a zahrnout <unistd.h> bezpodmínečně pouze v testovacím kódu). Je rozumné tvrdit, že "sysstat.h" by měla být nahrazena <sys/stat.h> pokud nemáte neobvykle nepoddajný systém.

(Tímto dostáváte povolení používat tento kód pro jakýkoli účel s uvedením zdroje.)

Tento kód je k dispozici v mém úložišti SOQ (Stack Overflow Questions) na GitHubu jako soubory mkpath.c amkpath.h (atd.) v podadresáři thesrc/so-0067-5039.


Linux
  1. Jak přidat adresář do PATH v Linuxu [s příklady]

  2. Jak vytvořit dočasný adresář v C++?

  3. Jak můžeme v Linuxu vytvořit více fiktivních rozhraní?

  1. Jak vytvořit sdílený adresář pro všechny uživatele v Linuxu

  2. Jak najít nejstarší soubor ve stromu adresářů v Linuxu

  3. Jak vytvořit nový adresář v Linuxu

  1. Jak používat příkaz mkdir k vytvoření nebo vytvoření adresáře Linux

  2. Mohu přidat zkratku pro nahrazení cesty v Linuxu?

  3. Jak vytvořit vlastní příkazy v Unixu/Linuxu?