C++实现SCP命令

scp命令的实现：

注意：仅适用于linux系统下，在window下会存在权限不足

实现代码

#include <iostream>
#include <cstdlib> 
#include <sys/stat.h> 
#include <sys/types.h> 

// 函数原型声明
bool checkDirectoryPermissions(const std::string& directory);
bool setDirectoryPermissions(const std::string& directory);

int main(int argc, char *argv[]) {
    // 检查命令行参数数量是否正确
    if (argc != 5) {
        std::cout << "用法：" << argv[0] << " <用户名> <IP地址> <远程文件路径> <本地目标目录>" << std::endl;
        return 1;
    }

    // 从命令行参数中获取远程主机的用户名、IP地址、远程文件路径和本地目标目录
    std::string username = argv[1];
    std::string ip_address = argv[2];
    std::string remote_file_path = argv[3];
    std::string local_directory = argv[4];

    // 检查目标目录的读写权限
    if (!checkDirectoryPermissions(local_directory)) {
        if (!setDirectoryPermissions(local_directory)) {
            std::cout << "无法设置目标目录权限！" << std::endl;
            return 1;
        }
    }
    // 构建 scp 命令
    std::string pass_command = "scp -r " + username + "@" + ip_address + ":" + remote_file_path + " " + local_directory;
    std::cout << pass_command << std::endl;

    // 执行 pass 命令
    int result = system(pass_command.c_str());

    // 检查命令执行结果
    if (result == 0) {
       std::cout << "文件传输成功！目标目录：" << local_directory << std::endl;
    } else {
        std::cout << "文件传输失败！" << std::endl;
    }

    return 0;
}
// 检查目录的读写权限
bool checkDirectoryPermissions(const std::string& directory) {
    struct stat info;
    if (stat(directory.c_str(), &info) != 0) {
        // 获取目录状态信息失败
        return false;
    }
    // 检查目录的读写权限
    if ((info.st_mode & S_IWUSR) == 0) {
        // 用户没有写权限
        return false;
    }
    // 默认情况下返回 false
    return false;
}

// 设置目录的读写权限
bool setDirectoryPermissions(const std::string& directory) {
    if (chmod(directory.c_str(), S_IRWXU) != 0) {
        
        return false;
    }
    return true;
}