transport/utils/bash_command.h

/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef UTILS_BASH_COMMAND_H_
#define UTILS_BASH_COMMAND_H_

#include <string>

namespace profiler {

// Absolute on-device path to 'run-as'.
const char *const kRunAsExecutable = "/system/bin/run-as";

// Run bash commands.
class BashCommandRunner {
 public:
  // Expected executable_path can be either absolute, relative or even
  // executable name only path.
  explicit BashCommandRunner(const std::string &executable_path)
      : BashCommandRunner(executable_path, false) {}
  explicit BashCommandRunner(const std::string &executable_path,
                             bool log_command)
      : executable_path_(executable_path), log_command_(log_command) {}

  virtual ~BashCommandRunner() = default;
  // Returns true if the command runs successfully (exiting with 0).
  // If |output| is not null, it is populated with stdin and stderr from
  // running command.
  virtual bool Run(const std::string &parameters, std::string *output) const;
  bool RunAs(const std::string &parameters, const std::string &username,
             std::string *output) const;
  static bool IsRunAsCapable();

 private:
  const std::string executable_path_;
  // True writes running commands to logs.
  const bool log_command_;
  virtual bool RunAndReadOutput(const std::string &cmd,
                                std::string *output) const;
};

}  // namespace profiler
#endif  // UTILS_BASH_COMMAND_H_