Add roboRIO Team Number Setter tool (#3744)

roborioteamnumbersetter/src/main/native/cpp/SshSession.h

@@ -0,0 +1,81 @@
+// Copyright (c) FIRST and other WPILib contributors.
+// Open Source Software; you can modify and/or share it under the terms of
+// the WPILib BSD license file in the root directory of this project.
+
+#pragma once
+
+#include <libssh/libssh.h>
+
+#include <stdexcept>
+#include <string>
+#include <string_view>
+
+#include <wpi/Logger.h>
+
+namespace sysid {
+/**
+ * This class is a C++ implementation of the SshSessionController in
+ * wpilibsuite/deploy-utils. It handles connecting to an SSH server, running
+ * commands, and transferring files.
+ */
+class SshSession {
+ public:
+  /**
+   * This is the exception that will be thrown by any of the methods in this
+   * class if something goes wrong.
+   */
+  class SshException : public std::runtime_error {
+   public:
+    explicit SshException(const char* msg) : runtime_error(msg) {}
+  };
+
+  /**
+   * Constructs a new session controller.
+   *
+   * @param host The hostname of the server to connect to.
+   * @param port The port that the sshd server is operating on.
+   * @param user The username to login as.
+   * @param pass The password for the given username.
+   * @param logger A reference to a logger to log messages to.
+   */
+  SshSession(std::string_view host, int port, std::string_view user,
+             std::string_view pass, wpi::Logger& logger);
+
+  /**
+   * Destroys the controller object. This also disconnects the session from the
+   * server.
+   */
+  ~SshSession();
+
+  /**
+   * Opens the SSH connection to the given host.
+   */
+  void Open();
+
+  /**
+   * Executes a command and logs the output (if there is any).
+   *
+   * @param cmd The command to execute on the server.
+   */
+  void Execute(std::string_view cmd);
+
+  /**
+   * Puts a file on the server using SFTP.
+   *
+   * @param path The path to the file to put (on the server).
+   * @param contents The contents of the file.
+   */
+  void Put(std::string_view path, std::string_view contents);
+
+ private:
+  ssh_session m_session;
+  std::string m_host;
+
+  int m_port;
+
+  std::string m_username;
+  std::string m_password;
+
+  wpi::Logger& m_logger;
+};
+}  // namespace sysid