add quiz/set_env_from_file

Author: lasyard

quiz/CMakeLists.txt

@@ -9,6 +9,7 @@ add_test(longest_symmetrical_length longest_symmetrical_length.cpp)
 add_test(magic_square magic_square.cpp)
 add_test(null_obj null_obj.cpp)
 add_test(reverse_words_in_sentence reverse_words_in_sentence.cpp)
+add_test(set_env_from_file set_env_from_file_test.cpp set_env_from_file.c)
 add_test(shared_ptr shared_ptr.cpp)
 add_test(sort_list_and_remove_dup sort_list_and_remove_dup.cpp)
 

quiz/set_env_from_file.c

@@ -0,0 +1,26 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "set_env_from_file.h"
+
+#define MAX_LINE_LENGTH 256
+
+bool set_env_from_file(const char *filename)
+{
+    FILE *file = fopen(filename, "r");
+    if (!file) {
+        return false;
+    }
+    char line[MAX_LINE_LENGTH];
+    char *saveptr;
+    while (fgets(line, sizeof(line), file)) {
+        char *key = strtok_r(line, "=", &saveptr);
+        char *value = strtok_r(NULL, "\n", &saveptr);
+        if (key && value) {
+            setenv(key, value, 1);
+        }
+    }
+    fclose(file);
+    return true;
+}

quiz/set_env_from_file.h

@@ -0,0 +1,16 @@
+#ifndef _QUIZ_SET_ENV_FROM_FILE_H_
+#define _QUIZ_SET_ENV_FROM_FILE_H_
+
+#include <stdbool.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+bool set_env_from_file(const char *filename);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* _QUIZ_SET_ENV_FROM_FILE_H_ */

quiz/set_env_from_file_test.cpp

@@ -0,0 +1,22 @@
+#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
+#include "doctest/doctest.h"
+
+#include <cstdlib>
+#include <fstream>
+
+#include "set_env_from_file.h"
+
+#define FILE_NAME "set_env_from_file_test.dat"
+
+TEST_CASE("Test set_env_from_file()")
+{
+    std::ofstream file(FILE_NAME);
+    CHECK(file);
+    file << "TEST_ENV=test_string" << std::endl;
+    file.close();
+    CHECK(set_env_from_file(FILE_NAME));
+    const char *v = getenv("TEST_ENV");
+    CHECK(v != NULL);
+    CHECK(strcmp(v, "test_string") == 0);
+    std::remove(FILE_NAME);
+}