1 files changed, 221 insertions, 0 deletions
diff --git a/meta-security/recipes-test/app-runas/files/app-runas.cpp b/meta-security/recipes-test/app-runas/files/app-runas.cpp
new file mode 100644
index 000000000..58fa15504
--- /dev/null
+++ b/meta-security/recipes-test/app-runas/files/app-runas.cpp
@@ -0,0 +1,221 @@
+// (C) Copyright 2015 Intel Corporation
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+// THE SOFTWARE.
+
+#include <security-manager.h>
+
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <errno.h>
+#include <string.h>
+
+#include <sys/types.h>
+#include <sys/wait.h>
+
+#include <string>
+#include <vector>
+
+#define CHECK(x) { \
+        int _ret = x; \
+        if (_ret != SECURITY_MANAGER_SUCCESS) { \
+            fprintf(stderr, "Failure in %s:%d: %s: %d = %s\n", __FILE__, __LINE__, #x, _ret, security_manager_strerror((lib_retcode)_ret)); \
+            return EXIT_FAILURE; \
+        } \
+   }
+
+static int do_install(app_inst_req *preq)
+{
+    CHECK(security_manager_app_install(preq));
+    return 0;
+}
+
+static int do_uninstall(app_inst_req *preq)
+{
+    CHECK(security_manager_app_uninstall(preq));
+    return 0;
+}
+
+static int do_run(const char *appid, const char *uid, const char *file, char *const argv[])
+{
+    if (!appid || !uid) {
+        fprintf(stderr, "Always need appid, uid for app startup.\n");
+        return EXIT_FAILURE;
+    }
+
+    pid_t child = fork();
+    if (child == -1) {
+        perror("fork");
+        return EXIT_FAILURE;
+    } else if (child) {
+        int status;
+        child = waitpid(child, &status, 0);
+        if (child == -1) {
+            perror("waitpid");
+            return EXIT_FAILURE;
+        }
+    } else {
+        // We cannot change the UID before security_manager_prepare_app()
+        // (because then setup_smack() fails to change Smack labels of
+        // our fds) and we cannot change the UID after it (because then
+        // security_manager_drop_process_privileges() has already dropped
+        // the necessary CAP_SETUID.
+        // Instead, we need to do the steps from security_manager_prepare_app()
+        // ourselves.
+        CHECK(security_manager_set_process_label_from_appid(appid));
+        CHECK(security_manager_set_process_groups_from_appid(appid));
+        if (setuid(atoi(uid))) {
+            fprintf(stderr, "setuid(%s): %s\n", uid, strerror(errno));
+            exit(EXIT_FAILURE);
+        }
+        CHECK(security_manager_drop_process_privileges());
+        // CHECK(security_manager_prepare_app(appid));
+
+        execvp(file, argv);
+        fprintf(stderr, "execvp(%s): %s", argv[optind], strerror(errno));
+        exit(EXIT_FAILURE);
+    }
+    return 0;
+}
+
+int main(int argc, char **argv)
+{
+    int flags, opt;
+    int nsecs, tfnd;
+    const char *appid = NULL;
+    const char *pkgid = NULL;
+    const char *uid = NULL;
+    std::vector<const char *> privileges;
+    std::vector< std::pair<app_install_path_type, std::string> > paths;
+    int install = 0, uninstall = 0, run = 0;
+
+    while ((opt = getopt(argc, argv, "a:p:u:r:t:ide")) != -1) {
+        switch (opt) {
+        case 'a':
+            appid = optarg;
+            break;
+        case 'p':
+            pkgid = optarg;
+            break;
+        case 'u':
+            uid = optarg;
+            break;
+        case 'r':
+            privileges.push_back(optarg);
+            break;
+        case 't': {
+            const char *colon = strchr(optarg, ':');
+            if (!colon) {
+                fprintf(stderr, "-t parameter must be of the format <type>:<path>");
+                return EXIT_FAILURE;
+            }
+            std::string typestr(optarg, colon - optarg);
+            std::string path(colon + 1);
+            app_install_path_type type;
+            if (typestr == "private") {
+                type = SECURITY_MANAGER_PATH_PRIVATE;
+            } else if (typestr == "public") {
+                type = SECURITY_MANAGER_PATH_PUBLIC;
+            } else if (typestr == "public-ro") {
+                type = SECURITY_MANAGER_PATH_PUBLIC_RO;
+            } else if (typestr == "rw") {
+                type = SECURITY_MANAGER_PATH_RW;
+            } else if (typestr == "ro") {
+                type = SECURITY_MANAGER_PATH_PRIVATE;
+            } else {
+                fprintf(stderr, "Invalid -t type: %s", typestr.c_str());
+                return EXIT_FAILURE;
+            }
+            paths.push_back(std::make_pair(type, path));
+            break;
+        }
+        case 'i':
+            install = 1;
+            break;
+        case 'd':
+            uninstall = 1;
+            break;
+        case 'e':
+            run = 1;
+            break;
+        default: /* '?' */
+            fprintf(stderr,
+                    "Usage: %s -i|-e|-d -a appid -u uid -p pkgid -r privilege1 ... -t private|public|public-ro|rw:<path> ... -- command args\n"
+                    "       -i = install, command ignored\n"
+                    "       -e = run command, privileges and pkgid ignored\n"
+                    "       -d = uninstall, command and privileges ignored\n"
+                    "       Install, run, and uninstall can be combined into a single invocation.\n",
+                    argv[0]);
+            exit(EXIT_FAILURE);
+            break;
+        }
+    }
+
+    if ((install || uninstall) &&
+        (!appid || !pkgid || !uid)) {
+         fprintf(stderr, "Always need appid, pkgid, uid for app install or uninstall.\n");
+         return EXIT_FAILURE;
+    }
+    if (run && optind >= argc) {
+        fprintf(stderr, "Expected command after options\n");
+        return EXIT_FAILURE;
+    }
+
+    app_inst_req *preq;
+    CHECK(security_manager_app_inst_req_new(&preq));
+    if (appid) {
+        CHECK(security_manager_app_inst_req_set_app_id(preq, appid));
+    }
+    if (pkgid) {
+        CHECK(security_manager_app_inst_req_set_pkg_id(preq, pkgid));
+    }
+    if (uid) {
+        CHECK(security_manager_app_inst_req_set_uid(preq, atoi(uid)));
+    }
+    for (size_t i = 0; i < paths.size(); i++) {
+        security_manager_app_inst_req_add_path(preq, paths[i].second.c_str(), paths[i].first);
+    }
+    for (size_t i = 0; i < privileges.size(); i++) {
+        CHECK(security_manager_app_inst_req_add_privilege(preq, privileges[i]));
+    }
+
+    int result = 0;
+    bool install_failed = false;
+    if (install) {
+        result = do_install(preq);
+        if (result) {
+            install_failed = true;
+        }
+    }
+    if (run && !install_failed) {
+        int run_result = do_run(appid, uid, argv[optind], argv + optind);
+        if (run_result) {
+            result = run_result;
+        }
+    }
+    if (uninstall && !install_failed) {
+        int uninstall_result = do_uninstall(preq);
+        if (uninstall_result) {
+            result = uninstall_result;
+        }
+    }
+
+    security_manager_app_inst_req_free(preq);
+    return result;
+}