Apribase

/*
 * File:   commands.hpp
 * Author: kei
 *
 * Created on 2009/08/18
 */
 
#include <string>
 
#ifndef _COMMANDS_HPP
#define _COMMANDS_HPP
 
namespace commands {
    const std::string getoutput(const std::string& command);
}
 
#endif /* _COMMANDS_HPP */

/*
 * File:   commands.cpp
 * Author: kei
 *
 * Created on 2009/08/18
 */
 
#include <cstdio>
#include <sstream>
#include <stdexcept>
#include "commands.hpp"
 
/**
 * 指定したコマンドのプロセスへのファイルディスクリプタを返します。
 * 使用後は close_process(const FILE* process) でクローズしてください。
 * プロセスのオープンに失敗した場合は例外を投げます。
 * @param command
 * @return プロセスへのファイルディスクリプタ
 * @throw std::runtime_error
 */
FILE* open_process(const std::string& command) {
    FILE* process = popen(command.c_str(), "r");
    if (process == NULL) {
        throw std::runtime_error("open " + command + " failed.");
    }
    return process;
}
 
/**
 * 実行プロセスの出力を返します。
 * @param process
 * @return プロセス実行出力
 */
const std::string read_process(FILE* process) {
    std::stringstream output;
    char buffer[128]; // allocate memory
    while (fgets(buffer, sizeof (buffer), process)) {
        output << buffer;
    }
    return output.str();
}
 
/**
 * 指定したプロセスをクローズします。
 * プロセスのクローズに失敗した場合は例外を投げます。
 * @throw std::runtime_error
 */
void close_process(FILE* process) {
    const int result = pclose(process);
    if (result == -1) {
        throw std::runtime_error("close_process faild.");
    }
}
 
/**
 * 指定したコマンドをシェル経由で実行し、出力を返します。
 * プロセスのオープン/実行/クローズのいずれかに失敗した場合は例外を投げます。
 * @param command コマンド
 * @return コマンド出力
 * @throw std::runtime_error
 */
const std::string commands::getoutput(const std::string& command) {
    FILE* process = NULL;
    std::string output;
    try {
        process = open_process(command);
        output = read_process(process);
    } catch (std::runtime_error error) {
        // finally 処理の代わりに、例外時にもクローズ処理を書いてあります。
        // open/read の例外時のため、close の例外確認は行いません。
        if (process != NULL) {
            close_process(process);
        }
        throw error;
    }
    // 正常時の close 処理です。例外確認も行います。
    if (process != NULL) {
        try {
            close_process(process);
        } catch (std::runtime_error error) {
            throw error;
        }
    }
    return output;
}

/*
 * File:   main.cpp
 * Author: kei
 *
 * Created on 2009/08/18
 */
 
#include <cstdlib>
#include <iostream>
#include <stdexcept>
#include "commands.hpp"
 
/**
 * メインエントリです。
 * 以下は date コマンドを実行するサンプルです。
 * @param argc
 * @param argv
 */
int main(int argc, char** argv) {
    try {
        std::cout << commands::getoutput("date") << std::endl;
    } catch(std::runtime_error error) {
        std::cerr << error.what() << std::endl;
    }
    return (EXIT_SUCCESS);
}