Initial HTTPSink implementation. Untested.

2026-06-27 02:01:42 +00:00 · 2016-09-08 00:05:10 -07:00
parent 85be299da1
commit e415ca66b6
4 changed files with 544 additions and 22 deletions
--- a/src/HTTPSinkImpl.h
+++ b/src/HTTPSinkImpl.h
@@ -0,0 +1,75 @@
+/*----------------------------------------------------------------------------*/
+/* Copyright (c) FIRST 2016. All Rights Reserved.                             */
+/* Open Source Software - may be modified and shared by FRC teams. The code   */
+/* must be accompanied by the FIRST BSD license file in the root directory of */
+/* the project.                                                               */
+/*----------------------------------------------------------------------------*/
+
+#ifndef CAMERASERVER_HTTPSINKIMPL_H_
+#define CAMERASERVER_HTTPSINKIMPL_H_
+
+#include <atomic>
+#include <condition_variable>
+#include <memory>
+#include <mutex>
+#include <thread>
+#include <vector>
+
+#include "llvm/raw_ostream.h"
+#include "llvm/SmallVector.h"
+#include "llvm/StringRef.h"
+#include "support/raw_istream.h"
+#include "support/raw_socket_ostream.h"
+#include "tcpsockets/NetworkAcceptor.h"
+#include "tcpsockets/NetworkStream.h"
+
+#include "SinkImpl.h"
+
+namespace cs {
+
+class SourceImpl;
+
+class HTTPSinkImpl : public SinkImpl {
+ public:
+  HTTPSinkImpl(llvm::StringRef name,
+               std::unique_ptr<wpi::NetworkAcceptor> acceptor);
+  ~HTTPSinkImpl() override;
+
+  void GetDescription(llvm::SmallVectorImpl<char>& desc) const override;
+
+  void SetSourceChannel(int channel) { m_sourceChannel.store(channel); }
+
+  void Stop();
+
+  static void SendHeader(llvm::raw_ostream& os, int code,
+                         llvm::StringRef codeText, llvm::StringRef contentType,
+                         llvm::StringRef extra = llvm::StringRef{});
+  static void SendError(llvm::raw_ostream& os, int code,
+                        llvm::StringRef message);
+  static bool ReadLine(wpi::raw_istream& istream,
+                       llvm::SmallVectorImpl<char>& buffer, int maxLen);
+  static bool UnescapeURI(llvm::StringRef str,
+                          llvm::SmallVectorImpl<char>& out);
+  static bool ProcessCommand(llvm::raw_ostream& os, SourceImpl& source,
+                             llvm::StringRef parameters, bool respond);
+  static void SendJSON(llvm::raw_ostream& os, SourceImpl& source, bool header);
+
+  void SendStream(wpi::raw_socket_ostream& os);
+
+ private:
+  void ServerThreadMain();
+  void ConnThreadMain(wpi::NetworkStream* stream);
+
+  std::unique_ptr<wpi::NetworkAcceptor> m_acceptor;
+  std::atomic_int m_sourceChannel;
+  std::atomic_bool m_active;  // set to false to terminate threads
+  std::thread m_serverThread;
+
+  std::mutex m_mutex;
+  std::vector<std::thread> m_connThreads;
+  std::vector<std::unique_ptr<wpi::NetworkStream>> m_connStreams;
+};
+
+}  // namespace cs
+
+#endif  // CAMERASERVER_HTTPSINKIMPL_H_