patchpanel/system.h

// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef PATCHPANEL_SYSTEM_H_
#define PATCHPANEL_SYSTEM_H_

#include <net/if.h>
#include <net/route.h>
#include <sys/ioctl.h>
#include <sys/types.h>

#include <string>

namespace patchpanel {

// cros lint will yell to force using int16/int64 instead of long here, however
// note that unsigned long IS the correct signature for ioctl in Linux kernel -
// it's 32 bits on 32-bit platform and 64 bits on 64-bit one.
using ioctl_req_t = unsigned long;  // NOLINT(runtime/int)

// Stateless class used for holding all utility functions with side
// effects on the environment. Facilitates mocking these functions in unit
// tests.
class System {
 public:
  // Enum used for restricting the possible paths that SysNetSet can write to.
  enum SysNet {
    // Used for modifying "net.ipv4.ip_forward"
    IPv4Forward = 1,
    // Used for modifying "net.ipv4.ip_local_port_range"
    IPLocalPortRange,
    // Used for modifying "net.ipv4.conf.%s.route_localnet", requires an
    // interface
    // argument
    IPv4RouteLocalnet,
    // Used for modifying "net.ipv6.conf.%s.accept_ra", requires an interface
    // argument
    IPv6AcceptRA,
    // Used for modifying "net.ipv6.conf.all.forwarding"
    IPv6Forward,
    // Used for enabling netfilter connection tracking helper modules.
    ConntrackHelper,
  };

  System() = default;
  System(const System&) = delete;
  System& operator=(const System&) = delete;
  virtual ~System() = default;

  // Write |content| to a "/proc/sys/net/" path as specified by |target|
  virtual bool SysNetSet(SysNet target,
                         const std::string& content,
                         const std::string& iface = "");

  virtual int Ioctl(int fd, ioctl_req_t request, const char* argp);
  int Ioctl(int fd, ioctl_req_t request, uint64_t arg);
  int Ioctl(int fd, ioctl_req_t request, struct ifreq* ifr);
  int Ioctl(int fd, ioctl_req_t request, struct rtentry* route);

  virtual pid_t WaitPid(pid_t pid, int* wstatus, int options = 0);

  static bool Write(const std::string& path, const std::string& content);

 private:
};

}  // namespace patchpanel

#endif  // PATCHPANEL_SYSTEM_H_