在我的程序中,所需的部分资源是存储数据的目录。按照自定义,我决定制作这个目录~/.program/
。在c ++中,制作此目录的正确方法(在基于UNIX的系统上)是以下代码:
#include <sys/stat.h>
#include <unistd.h>
#include <iostream>
using namespace std;
void mkworkdir()
{
if(stat("~/.program",&st) == 0)
{
cout << "Creating working directory..." << endl;
mkdir("~/.program/", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
mkdir("~/.program/moredata", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
}
else
{
cout << "Working directory found... continuing" << endl;
}
}
int main()
{
mkworkdir();
return 0;
}
现在,在~
中使用mkdir("~/.program/", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH)
的可靠性至少是值得怀疑的,所以我真正想做的是提示用户名,将其存储在string
中(比如string usern; cin >> usern;
),然后执行mkdir("/home/{$USERN}/.program/", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH)
(就像在shell中一样)。但是,我不知道如何将一些等效的$ USERN转换为字符串,因为我不知道如何将可展开的c ++构造转换为字符串。我的意思是我插入任何&#34;形式&#34;变量将被扩展为该变量的内容到字符串中。
如果这个问题令人困惑,我道歉,我似乎无法正确解释我想要的是什么。
或者,更优选的是,是否可以在不提示的情况下获取用户名? (当然,将其存储在一个字符串中)
答案 0 :(得分:4)
您可以使用:
std::string usern;
std::cin >> usern;
std::string directory = "/home/" + usern + "/.program";
mkdir(directory.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
更好的选择,IMO,是使用环境变量HOME的值。
char const* home = std::getenv("HOME");
if ( home == nullptr )
{
// Deal with problem.
}
else
{
std::string directory = home + std::string("/.program");
mkdir(directory.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
}
FWIW,您可以通过在应用程序的命名空间中创建函数make_directory
来简化代码,您可以使用正确的标志等添加检查目录是否存在的详细信息。
namespace MyApp
{
bool directory_exists(std::string const& directory)
{
struct stat st;
// This is simplistic check. It will be a problem
// if the entry exists but is not a directory. Further
// refinement is needed to deal with that case.
return ( stat(directory.c_tr(), &st) == 0 );
}
int make_directory(std::string const& directory)
{
if ( directory_exists(directory) )
{
return 0;
}
return mkdir(directory.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
}
}
然后,您可以在其余代码中MyApp::make_directory
。
char const* home = std::getenv("HOME");
if ( home == nullptr )
{
// Deal with problem.
}
else
{
std::string directory = home + std::string("/.program");
MyApp::make_directory(directory);
MyApp::make_directory(directory + "/moredata");
}
答案 1 :(得分:1)
使用usern
中的用户名,您可以按照您想要的任何方式构建字符串:
std::string dir = "/home/" + usern + "/.program/"
mkdir(dir.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
mkdir(dir.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
要在不向用户投放的情况下获取用户名,您可以使用getlogin_r()