media_server.cpp

#include "rtmp_server.hpp"
#include "ws_server.hpp"
#include "httpflv_server.hpp"
#include "net/webrtc/webrtc_pub.hpp"
#include "net/webrtc/rtc_dtls.hpp"
#include "net/webrtc/srtp_session.hpp"
#include "utils/byte_crypto.hpp"
#include "logger.hpp"
#include <stdint.h>
#include <stddef.h>

websocket_server* wss_p = nullptr;
websocket_server* ws_p  = nullptr;

const std::string ssl_pem_file = "../certs/server.key";
const std::string cert_file = "../certs/server.crt";
boost::asio::io_context io_context;

void create_wss_server(boost::asio::io_context& io_context, uint16_t ws_port) {


    log_infof("websocket https server is starting, port:%d", ws_port);
    wss_p = new websocket_server(io_context, ws_port, WEBSOCKET_IMPLEMENT_PROTOO_TYPE, cert_file, ssl_pem_file);
}

void create_ws_server(boost::asio::io_context& io_context, uint16_t ws_port) {
    ws_p = new websocket_server(io_context, ws_port, WEBSOCKET_IMPLEMENT_PROTOO_TYPE);
    log_infof("websocket http server is starting, port:%d", ws_port);
}

boost::asio::io_context& get_global_io_context() {
    return io_context;
}

int main(int argn, char** argv) {
    const uint16_t rtmp_def_port = 1935;
    const uint16_t ws_def_port = 1900;
    const uint16_t ws_webrtc_port = 9110;
    const uint16_t httpflv_port = 8080;
    const uint16_t webrtc_media_port = 7000;
    const std::string host_ip = "172.29.78.76";

    
    boost::asio::io_service::work work(io_context);

    Logger::get_instance()->set_filename("server.log");

    try {
        byte_crypto::init();
        rtc_dtls::dtls_init(ssl_pem_file, cert_file);
        srtp_session::init();
        init_single_udp_server(io_context, host_ip, webrtc_media_port);

        rtmp_server server(io_context, rtmp_def_port);
        websocket_server ws(io_context, ws_def_port, WEBSOCKET_IMPLEMENT_FLV_TYPE);
        httpflv_server httpflv_serv(io_context, httpflv_port);

        create_ws_server(io_context, ws_webrtc_port);
        log_infof("rtmp server start:%d", rtmp_def_port);
        log_infof("websocket server start:%d", ws_def_port);
        io_context.run();
    }
    catch(const std::exception& e) {
        std::cerr << e.what() << '\n';
    }
    return 0;
}