Tube.hpp

#pragma once
 
#include "Common.hpp"
 
namespace pwn::Net
{
class Tube
{
public:
    static constexpr usize PIPE_DEFAULT_SIZE = 1024;
 
    static constexpr u8 LINE_SEPARATOR = '\n';
 
    static constexpr std::string_view INTERACTIVE_PROMPT = "(pwn)> ";
 
 
    PWNAPI Result<usize>
    send(std::vector<u8> const& str);
 
    PWNAPI Result<usize>
    send(std::string const& str);
 
 
    Result<std::vector<u8>>
    recv(_In_ usize size = Tube::PIPE_DEFAULT_SIZE);
 
 
    Result<usize>
    sendline(_In_ std::vector<u8> const& data);
 
    PWNAPI Result<usize>
    sendline(_In_ std::string const& str);
 
 
    PWNAPI Result<std::vector<u8>>
    recvuntil(_In_ std::vector<u8> const& pattern);
 
    PWNAPI Result<std::vector<u8>>
    recvuntil(_In_ std::string const& pattern);
 
 
    PWNAPI Result<std::vector<u8>>
    recvline();
 
 
    PWNAPI Result<usize>
    sendafter(_In_ std::string const& pattern, _In_ std::string const& data);
 
    PWNAPI Result<usize>
    sendafter(_In_ std::vector<u8> const& pattern, _In_ std::vector<u8> const& data);
 
 
    PWNAPI Result<usize>
    sendlineafter(_In_ std::string const& pattern, _In_ std::string const& data);
 
    PWNAPI Result<usize>
    sendlineafter(_In_ std::vector<u8> const& pattern, _In_ std::vector<u8> const& data);
 
 
    PWNAPI Result<usize>
    peek();
 
 
    PWNAPI void
    interactive();
 
 
protected:
    Tube()  = default;
    ~Tube() = default;
 
    virtual Result<usize>
    send_internal(_In_ std::vector<u8> const& data) = 0;
 
    virtual Result<std::vector<u8>>
    recv_internal(_In_ usize size) = 0;
 
    virtual Result<usize>
    peek_internal() = 0;
 
    std::vector<u8> m_receive_buffer;
    std::vector<u8> m_send_buffer;
};
} // namespace Net