runtime_probe/functions/network.h

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

#ifndef RUNTIME_PROBE_FUNCTIONS_NETWORK_H_
#define RUNTIME_PROBE_FUNCTIONS_NETWORK_H_

#include <optional>
#include <string>

#include "runtime_probe/probe_function.h"
#include "runtime_probe/probe_function_argument.h"

namespace runtime_probe {

class NetworkFunction : public PrivilegedProbeFunction {
  using PrivilegedProbeFunction::PrivilegedProbeFunction;

 public:
  NAME_PROBE_FUNCTION("network");

 protected:
  virtual std::optional<std::string> GetNetworkType() const;

 private:
  // PrivilegedProbeFunction overrides.
  bool PostParseArguments() final;
  DataType EvalImpl() const final;
  void PostHelperEvalImpl(DataType* helper_results) const final;

  // Type of the network device. Accepts all types if it is omitted.
  PROBE_FUNCTION_ARG_DEF(std::string, device_type, (std::string("")));
};

}  // namespace runtime_probe

#endif  // RUNTIME_PROBE_FUNCTIONS_NETWORK_H_