Interceptor: add helper functionality for env and args

Add some convenience argument and environment parsing into standard
containers to make them easier accessible. This might be bad for
performance and needs to be replaced later, but helps during initial
development.

Bug: 205577427
Signed-off-by: Matthias Maennich <maennich@google.com>
Change-Id: I6db40054c58a117dd5b3434bced85001ac037f81

1 files changed, 44 insertions, 0 deletions

diff --git a/interceptor/interceptor.cc b/interceptor/interceptor.cc
index 64cdb76..f078e42 100644
--- a/interceptor/interceptor.cc
+++ b/interceptor/interceptor.cc
@@ -19,6 +19,13 @@
 #include <dlfcn.h>
 #include <unistd.h>
 
+#include <filesystem>
+#include <string>
+#include <string_view>
+#include <utility>
+
+namespace fs = std::filesystem;
+
 // OVERLOADS for LD_PRELOAD USE
 
 // Intercept execve calls, for that capture the original execve call
@@ -29,3 +36,40 @@ int execve(const char* filename, char* const argv[], char* const envp[]) {
   return old_execve(filename, argv, envp);
 }
 }  // extern "C"
+
+// LIBRARY IMPLEMENTATION
+
+namespace interceptor {
+
+Command::Command(const char* program, char* const argv[], char* const envp[])
+    : program_(program), cwd_(fs::current_path()), argv_(argv), envp_(envp) {}
+
+const ArgVec& Command::args() const {
+  if (!args_.has_value()) {
+    args_ = ArgVec();
+    for (auto current_arg = argv_; *current_arg; ++current_arg) {
+      args_->emplace_back(*current_arg);
+    }
+  }
+  return *args_;
+}
+
+const EnvMap& Command::env() const {
+  if (!env_.has_value()) {
+    env_ = EnvMap();
+    for (auto current_env = envp_; *current_env; ++current_env) {
+      const std::string_view s(*current_env);
+      const auto pos = s.find('=');
+      if (pos == EnvMap::key_type::npos) {
+        continue;
+      }
+      env_->emplace(s.substr(0, pos), s.substr(pos + 1));
+    }
+  }
+  return *env_;
+}
+
+const std::string& Command::program() const {
+  return program_;
+}
+}  // namespace interceptor