debugd/src/lp_tools.h

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef DEBUGD_SRC_LP_TOOLS_H_
#define DEBUGD_SRC_LP_TOOLS_H_

#include <string>
#include <vector>

#include <base/environment.h>
#include <base/files/file_path.h>

#include "debugd/src/process_with_output.h"

namespace debugd {

class LpTools {
 public:
  virtual ~LpTools() = default;

  // Runs lpadmin with the provided |arg_list| and |std_input|.
  virtual int Lpadmin(const ProcessWithOutput::ArgList& arg_list,
                      bool inherit_usergroups = false,
                      const base::EnvironmentMap& env = {},
                      const std::vector<uint8_t>* std_input = nullptr) = 0;

  // Runs lpstat with the provided |arg_list| and |std_input|.
  virtual int Lpstat(const ProcessWithOutput::ArgList& arg_list,
                     std::string* output) = 0;

  // Runs cupstestppd with |ppd_content| and returns the exit code.
  virtual int CupsTestPpd(const std::vector<uint8_t>& ppd_content) const = 0;

  // Runs the cups_uri_helper on |uri| and returns the exit code.
  virtual int CupsUriHelper(const std::string& uri) const = 0;

  virtual const base::FilePath& GetCupsPpdDir() const = 0;

  // Returns the exit code for the executed process.
  virtual int RunAsUser(const std::string& user,
                        const std::string& group,
                        const std::string& command,
                        const std::string& seccomp_policy,
                        const ProcessWithOutput::ArgList& arg_list,
                        const std::vector<uint8_t>* std_input = nullptr,
                        bool inherit_usergroups = false,
                        const base::EnvironmentMap& env = {},
                        std::string* out = nullptr) const = 0;

  // Change ownership of a file.  Return 0 on success, -1 on error.
  virtual int Chown(const std::string& path,
                    uid_t owner,
                    gid_t group) const = 0;
};

class LpToolsImpl : public LpTools {
 public:
  ~LpToolsImpl() override = default;

  int Lpadmin(const ProcessWithOutput::ArgList& arg_list,
              bool inherit_usergroups = false,
              const base::EnvironmentMap& env = {},
              const std::vector<uint8_t>* std_input = nullptr) override;

  int Lpstat(const ProcessWithOutput::ArgList& arg_list,
             std::string* output) override;

  int CupsTestPpd(const std::vector<uint8_t>& ppd_content) const override;

  int CupsUriHelper(const std::string& uri) const override;

  const base::FilePath& GetCupsPpdDir() const override;

  int RunAsUser(const std::string& user,
                const std::string& group,
                const std::string& command,
                const std::string& seccomp_policy,
                const ProcessWithOutput::ArgList& arg_list,
                const std::vector<uint8_t>* std_input = nullptr,
                bool inherit_usergroups = false,
                const base::EnvironmentMap& env = {},
                std::string* out = nullptr) const override;

  int Chown(const std::string& path, uid_t owner, gid_t group) const override;
};

}  // namespace debugd

#endif  // DEBUGD_SRC_LP_TOOLS_H_