process.h

Go to the documentation of this file.

 
#pragma once
 
#ifndef __kernel_entry
  #define __kernel_entry
#endif
 
// standard includes
#include <optional>
#include <unordered_map>
 
// lib includes
#include <boost/process/v1.hpp>
 
// local includes
#include "config.h"
#include "platform/common.h"
#include "rtsp.h"
#include "utility.h"
 
#define DEFAULT_APP_IMAGE_PATH SUNSHINE_ASSETS_DIR "/box.png"
 
namespace proc {
  using file_t = util::safe_ptr_v2<FILE, int, fclose>;
 
  typedef config::prep_cmd_t cmd_t;
 
  struct ctx_t {
    std::vector<cmd_t> prep_cmds;  
 
    std::vector<std::string> detached;
 
    std::string name;  
    std::string cmd;  
    std::string working_dir;  
    std::string output;  
    std::string image_path;  
    std::string id;  
    bool elevated;  
    bool auto_detach;  
    bool wait_all;  
    std::chrono::seconds exit_timeout;  
  };
 
  class proc_t {
  public:
    KITTY_DEFAULT_CONSTR_MOVE_THROW(proc_t)
 
    
    proc_t(
      boost::process::v1::environment &&env,
      std::vector<ctx_t> &&apps
    ):
        _app_id(0),
        _env(std::move(env)),
        _apps(std::move(apps)) {
    }
 
    int execute(int app_id, std::shared_ptr<rtsp_stream::launch_session_t> launch_session);
 
    int running();
 
    ~proc_t();
 
    const std::vector<ctx_t> &get_apps() const;
    std::vector<ctx_t> &get_apps();
    std::string get_app_image(int app_id);
    std::string get_last_run_app_name();
    void terminate();
 
  private:
    int _app_id;
 
    boost::process::v1::environment _env;
    std::vector<ctx_t> _apps;
    ctx_t _app;
    std::chrono::steady_clock::time_point _app_launch_time;
 
    // If no command associated with _app_id, yet it's still running
    bool placebo {};
 
    boost::process::v1::child _process;
    boost::process::v1::group _process_group;
 
    file_t _pipe;
    std::vector<cmd_t>::const_iterator _app_prep_it;
    std::vector<cmd_t>::const_iterator _app_prep_begin;
  };
 
  std::tuple<std::string, std::string> calculate_app_id(const std::string &app_name, std::string app_image_path, int index);
 
  bool check_valid_png(const std::filesystem::path &path);
  std::string validate_app_image_path(std::string app_image_path);
  void refresh(const std::string &file_name);
  std::optional<proc::proc_t> parse(const std::string &file_name);
 
  std::unique_ptr<platf::deinit_t> init();
 
  void terminate_process_group(boost::process::v1::child &proc, boost::process::v1::group &group, std::chrono::seconds exit_timeout);
 
  extern proc_t proc;
}  // namespace proc