load_environment_file support

Author: macskas

.github/workflows/cmake.yml

@@ -6,7 +6,7 @@ env:
   # Customize the CMake build type here (Release, Debug, RelWithDebInfo, etc.)
   BUILD_TYPE: Release
   PACKAGE_NAME: docker-runner-php
-  PACKAGE_VERSION: 1.0.1-1
+  PACKAGE_VERSION: 1.0.2-1
   ARCH: amd64
 
 jobs:
@@ -31,7 +31,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [ubuntu-latest, ubuntu-20.04]
+        os: [ubuntu-22.04, ubuntu-20.04, ubuntu-18.04]
 
     steps:
     - uses: actions/checkout@v2

CMakeLists.txt

@@ -4,4 +4,4 @@ project(docker-runner)
 set(CMAKE_CXX_STANDARD 14)
 set(CMAKE_CXX_FLAGS "-mtune=generic -march=x86-64 -std=c++0x")
 LINK_LIBRARIES(rt dl)
-add_executable(docker-runner main.cpp globals.h cdocker.cpp cdocker.h cdockercontainer.cpp cdockercontainer.h cphprunner.cpp cphprunner.h)
+add_executable(docker-runner main.cpp globals.h cdocker.cpp cdocker.h cdockercontainer.cpp cdockercontainer.h cphprunner.cpp cphprunner.h utils.cpp utils.h)

globals.h

@@ -7,5 +7,8 @@
 #define DBUF_SIZE_DEFAULT 4096
 #define DBUF_SIZE_SMALL   1024
 
+#define ENV_FILE_PRELOAD_DEFAULT "/etc/docker-images/.env"
+#define ENV_FILE_PRELOAD_ENV_KEY "ENV_FILE_PRELOAD"
+
 #endif // GLOBALS
 

main.cpp

@@ -7,6 +7,7 @@
 #include "cdocker.h"
 #include "cphprunner.h"
 #include "globals.h"
+#include "utils.h"
 
 extern "C" {
 #include <libgen.h>
@@ -82,6 +83,7 @@ int main(int argc, char **argv)
     PHPRunner.getSelectedVersion(&dockerHostname);
     selected_version_int = PHPRunner.getSelectedVersion();
 
+    load_environment_file();
     return Docker.run(dockerHostname, selected_version_int, newargs);
 }
 

utils.cpp

@@ -0,0 +1,128 @@
+#include "globals.h"
+#include <string>
+#include <unistd.h>
+#include <cstring>
+#include <cstdio>
+
+static
+void rtrim(char *line) {
+    if (line == nullptr)
+        return;
+    size_t line_length = strlen(line);
+    char *lhelper = line+line_length-1;
+    if (line_length == 0)
+        return;
+
+    while (1) {
+        if (lhelper < line)
+            break;
+
+        // space allowed
+        if (isspace(*lhelper)) {
+            *lhelper = 0;
+            lhelper--;
+            continue;
+        } else {
+            break;
+        }
+    }
+}
+
+static
+char* ltrim(char *line) {
+    if (line == nullptr)
+        return line;
+    size_t line_length = strlen(line);
+    char *line_new = line;
+    for (size_t i=0, e=line_length; i<e; i++) {
+        if (!isspace(line[i])) {
+            line_new = line+i;
+            break;
+        }
+    }
+    return line_new;
+}
+
+static
+bool is_valid_key(char *key) {
+    if (!key)
+        return false;
+
+    size_t key_len = strlen(key);
+    if (key_len == 0)
+        return false;
+
+
+    for (size_t i=0; i<key_len; i++) {
+        if (isalnum(key[i]) || key[i] == '_') {
+            continue;
+        } else {
+            return false;
+        }
+    }
+    return true;
+}
+
+static
+void load_environment_line(char *line, size_t length) {
+    rtrim(line);
+    char *line_start = ltrim(line);
+    size_t line_length = strlen(line_start);
+    if (line_length < 3)
+        return;
+
+    if (line_start[0] == '#') {
+        return;
+    }
+    char *eq_pos = strchr(line_start, '=');
+    if (eq_pos == nullptr)
+        return;
+    if (line_length-(eq_pos-line_start) <= 1)
+        return;
+
+    *eq_pos = 0;
+    char *key = line_start;
+    if (!is_valid_key(key)) {
+        return;
+    }
+
+    char *val = eq_pos+1;
+    rtrim(key);
+    size_t val_len = strlen(val);
+    if (val_len >= 2) {
+        if ((val[0] == '"' || val[0] == '\'') && val[0] == val[val_len-1]) {
+            val[val_len-1] = 0;
+            val++;
+            val_len -= 2;
+        }
+    }
+    if (val_len == 0) {
+        unsetenv(key);
+        return;
+    }
+
+    setenv(key, val, 1);
+}
+
+void load_environment_file()
+{
+    std::string env_file = ENV_FILE_PRELOAD_DEFAULT;
+    char *env_file_from_env = secure_getenv(ENV_FILE_PRELOAD_ENV_KEY);
+    FILE *f = nullptr;
+    char linebuf[4096]{};
+
+    if (env_file_from_env != nullptr) {
+        env_file.assign(env_file_from_env);
+    }
+    if (access(env_file.c_str(), R_OK) != 0)
+        return;
+    f = fopen(env_file.c_str(), "r");
+    if (f == nullptr) {
+        return;
+    }
+
+    while (fgets(linebuf, sizeof(linebuf), f)) {
+        load_environment_line(linebuf, sizeof(linebuf));
+    }
+    fclose(f);
+}

utils.h

@@ -0,0 +1,10 @@
+//
+// Created by macskas on 10/13/23.
+//
+
+#ifndef DOCKER_RUNNER_UTILS_H
+#define DOCKER_RUNNER_UTILS_H
+
+void load_environment_file();
+
+#endif //DOCKER_RUNNER_UTILS_H