C言語とCGIで大量のFAX送信作業を自動化した話

// ロックファイルを開く
HANDLE hFile = CreateFile(lockFileName.c_str(), GENERIC_READ | GENERIC_WRITE, FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE) {
	std::cerr << "Unable to open or create lock file." << std::endl;
	return 1;
}

// LockFileExを使用してファイルを排他ロック
OVERLAPPED overlapped = { 0 };

// ロックが取得できるまでループ
while (true) {
	if (LockFileEx(hFile, LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY, 0, MAXDWORD, MAXDWORD, &overlapped)) {
		std::cerr << "Lock file is locked." << std::endl;
		break; // ロック成功
	}

	// ロック取得に失敗した場合、少し待機してから再試行
	if (GetLastError() == ERROR_LOCK_VIOLATION) {
		std::cerr << "Failed to lock the file. Retrying..." << std::endl;
		Sleep(1000); // 1秒待機
	}
	else {
		std::cerr << "Failed to lock the file. Error: " << GetLastError() << std::endl;
		CloseHandle(hFile);
		return 1;
	}
}

// ロック成功、Fax処理を実行
exec_proc();

	// REQUEST_METHODの取得
	std::string request_method = std::getenv("REQUEST_METHOD");
	
	if (request_method != "POST"){
		// 省略
	]
	
	// CONTENT_TYPEの取得
	char* contentType = getenv("CONTENT_TYPE");
	
	// 省略
	
	// CONTENT_LENGTHの取得
	char* contentLengthStr = getenv("CONTENT_LENGTH");
	
	// CONTENT_LENGTHの値を数値に変換
	int contentLength = 0;
	if (contentLengthStr != nullptr) {
		contentLength = std::stoi(contentLengthStr);
	}

int parseMultipartFormData(const std::string& boundary, std::istream& stream, FaxRequest& jsonData, std::string fileName, int contentLength) {
		// バイナリモードを設定
		_setmode(_fileno(stdin), _O_BINARY);

		const int readSize = contentLength;

		// バイナリモードで全データを読み込む
		char* buffer = new char[readSize];
		std::cin.read(buffer, sizeof(char) * readSize);

		// バッファに取り込んだデータをstd::vector<char>に変換する
		std::vector<char> data(buffer, buffer + readSize);
		// マルチパートの境界を指定
		std::string endBoundary = boundary + "--";
		auto endBoundaryPos = findPattern(data.begin(), data.end(), endBoundary);

		auto currentPos = data.begin();
		
		while (currentPos != data.end()) {

			auto boundaryPos = findPattern(currentPos, data.end(), boundary);
			if (boundaryPos == data.end()) break; // 境界が見つからない場合は終了

			// 次の境界を探す
			auto nextBoundaryPos = findPattern(boundaryPos + boundary.length(), data.end(), boundary);
			
			// boundaryが終端だったらパースを終了する
			if (nextBoundaryPos == data.end()) break;
			
			// ヘッダとペイロードを分離する
			auto headerEndPos = findPattern(boundaryPos, nextBoundaryPos, "\\r\\n\\r\\n");
			if (headerEndPos == nextBoundaryPos) break;

			std::string header(boundaryPos, headerEndPos);
			if (header.find("name=\\"body\\"") != std::string::npos) {
				// JSONパートの処理
				
				// ヘッダの終わりからデータ開始
				auto jsonStart = headerEndPos + 1;
				// 次の境界の手前まで
				std::string jsonStr(jsonStart, nextBoundaryPos - 1);
				// JSON文字列をパースして構造体に変換
				jsonData = nlohmann::json::parse(jsonStr).get<FaxRequest>();
			}
			else if (header.find("name=\\"file\\"") != std::string::npos) {
				// ファイルパートの処理
				
				// ファイルの読み取り開始位置の指定
				auto fileStart = headerEndPos + 4;
				// ファイルに書き出す
				std::ofstream outFile(fileName, std::ios::out | std::ios::binary);
				if (!outFile) {
					std::cerr << "failed to open file: " << fileName << std::endl;
					return 1;
				}
				outFile.write(&(*fileStart), std::distance(fileStart, nextBoundaryPos - 2));
			}
			currentPos = nextBoundaryPos;
		}
		_setmode(_fileno(stdin), _O_TEXT);

		delete[] buffer;
		return 0;
	}

// FAX送信のメインの処理を行う関数
int exec_proc() {
	// 省略
	

	// 標準入力に入ったCGIに対するPOSTリクエストのボディを解析して構造体と一時ファイルを作成
	int code = prtimes::parseMultipartFormData(boundary, std::cin, req, fileName, contentLength);
	
	if (code == 1) {
		std::cout << prtimes::responseFactory("Failed to save temporary file", prtimes::STATUS_INTERNAL) << std::endl;
		return 0;
	}
	else if (code > 0) {
		std::cout << prtimes::responseFactory("Failed to save temporary file", prtimes::STATUS_INTERNAL) << std::endl;
		return 0;
	}
	
	// ここからFAXの送信処理
	// SEHはstd::runtime_errorに変換されるため、このtry句で全てキャッチできる
	try {
		// FAXの送信処理
		code = prtimes::SendMultiFax(req, fileName);
	}
	catch (const std::runtime_error& e) {
		code = prtimes::FAX_UNEXPECTED_ERROR;
		std::cerr << e.what() << std::endl;
	}
	catch (const std::exception& e) {
		code = prtimes::FAX_UNEXPECTED_ERROR;
		std::cerr << "std::exception: " << e.what() << std::endl;
	}
	catch (...) {
		code = prtimes::FAX_UNEXPECTED_ERROR;
	}

	if (code == prtimes::FAX_INIT_ERROR) {
		std::cout << prtimes::responseFactory("Failed to init MpfApi", prtimes::STATUS_INTERNAL) << std::endl;
		return 0;
	}
	else if (code == prtimes::FAX_CONNECT_ERROR) {
		std::cout << prtimes::responseFactory("Failed to establish a fax server connection", prtimes::STATUS_INTERNAL) << std::endl;
		return 0;
	}
	else if (code == prtimes::FAX_PROCESS_ERROR) {
		std::cout << prtimes::responseFactory("Failed to send fax data", prtimes::STATUS_INTERNAL) << std::endl;
		return 0;
	}
	else if (code == prtimes::FAX_FAILED_EXIT_PROCESS) {
		std::cout << prtimes::responseFactory("Unexpected error has occured. But, fax process is success.", prtimes::STATUS_OK) << std::endl;
		return 0;
	}
	else if (code == prtimes::FAX_UNEXPECTED_ERROR) {
		std::cout << prtimes::responseFactory("some error has occured.", prtimes::STATUS_INTERNAL) << std::endl;
		return 1;
	}

	std::cout << prtimes::responseFactory("success", prtimes::STATUS_OK) << std::endl;
	return 0;
}

struct FaxAnalytics {
	std::string fax_number; // FAX送信番号
	std::string company_name; // 宛先会社名
	std::string company_depart; // 宛先会社の部署情報
	std::string name; // FAX担当者名
	std::string title; // FAX送信タイトル
	int company_id; // 会社ID
	int release_id; // リリースID
	unsigned long log_id; // FAXの送信ジョブに割り当てられたユニークなID
	char send_status; // FAXサーバーのエラーコード
	std::string message; // エラーコードのメッセージ{送信正常終了, 送信異常終了, 送信処理中}
	std::string ip_address; // FAX送信サーバーのIPアドレス
	time_t received_at; // FAX処理受付日時
	time_t started_at; // FAX送信処理開始日時
	time_t ended_at; // FAX送信処理完了日時
};

int main(int argc, char* argv[]) {
	// stderrをファイルにリダイレクト
	auto result = freopen(prtimes::getLogfileName().c_str(), "a+", stderr);
	std::cerr << "start process" << std::endl;

	// ロケールを日本語に合わせる。マルチバイト文字の変換に影響するので必ず入れる。
	// 基本的にはUTF-8での処理を行い、必要な個所で変換を行う。
	std::wcout.imbue(std::locale("ja"));

	// Windows独自の例外であるSEHのハンドラを定義。これがないとセグフォなどのシステムエラーがキャッチできない。
	_set_se_translator(TranslateSEHtoCplusplusException);

	if (argc != 2) {
		std::cerr << "Usage: " << argv[0] << " <date in YYYY-MM-DD format>" << std::endl;
		return 1;
	}

	std::string input_date(argv[1]);
	std::tm tm = {};
	std::istringstream ss(input_date);
	ss >> std::get_time(&tm, "%Y-%m-%d");

	if (ss.fail()) {
		std::cerr << "Failed to parse date." << std::endl;
		return 1;
	}

	// 指定された日付の00:00:00にする
	tm.tm_hour = 0;
	tm.tm_min = 0;
	tm.tm_sec = 0;

	std::time_t start_time = std::mktime(&tm);
	if (start_time == -1) {
		std::cerr << "Failed to convert time." << std::endl;
		return 1;
	}

	// 23:59:59を求める
	tm.tm_hour = 23;
	tm.tm_min = 59;
	tm.tm_sec = 59;
	std::time_t end_time = std::mktime(&tm);
	if (end_time == -1) {
		std::cerr << "Failed to convert time." << std::endl;
		return 1;
	}
	// FAXの送信ログにアクセスしてログを収集する
	exec_log_proc(start_time, end_time);

	return 0;
}

auto result = freopen(prtimes::getLogfileName().c_str(), "a+", stderr);

// SEH例外をC++例外に変換する
// C++ではtryのネストができないため、SEH例外をC++例外に変換してcatchすることで一元的に扱えるようにします
void TranslateSEHtoCplusplusException(unsigned int u, EXCEPTION_POINTERS* pExp) {
	std::stringstream ss;
	ss << "SEH exception has occurred, error code: ";

	// 例外コードを取得
	auto code = pExp->ExceptionRecord->ExceptionCode;
	ss << std::hex << code;

	// std::runtime_errorにエラーメッセージを渡して投げる
	throw std::runtime_error(ss.str());
}

// 省略

// Windows独自の例外であるSEHのハンドラを定義。これがないとセグフォなどのシステムエラーがキャッチできない。
_set_se_translator(TranslateSEHtoCplusplusException);