mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-22 01:11:42 +00:00
76 lines
2.5 KiB
C++
76 lines
2.5 KiB
C++
/*----------------------------------------------------------------------------*/
|
|
/* 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_MJPEGSERVERIMPL_H_
|
|
#define CAMERASERVER_MJPEGSERVERIMPL_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 MJPEGServerImpl : public SinkImpl {
|
|
public:
|
|
MJPEGServerImpl(llvm::StringRef name, llvm::StringRef description,
|
|
std::unique_ptr<wpi::NetworkAcceptor> acceptor);
|
|
~MJPEGServerImpl() override;
|
|
|
|
llvm::StringRef GetDescription(
|
|
llvm::SmallVectorImpl<char>& buf) const override;
|
|
|
|
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::string m_description;
|
|
|
|
std::unique_ptr<wpi::NetworkAcceptor> m_acceptor;
|
|
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_MJPEGSERVERIMPL_H_
|