X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=udpinput.cpp;h=a2915e8333dff36c1b3cc1fa2efcb1cfbf1d27cf;hp=001ef5f6d3db7a5fbf89ec279482b27241eb1a16;hb=4ca2711d4e8f8a3c20e5feda537b30c356a64329;hpb=07569f8e011cd9b064c64bef1ce56f77bf7ddf53 diff --git a/udpinput.cpp b/udpinput.cpp index 001ef5f..a2915e8 100644 --- a/udpinput.cpp +++ b/udpinput.cpp @@ -6,11 +6,11 @@ #include #include #include +#include #include #include "acceptor.h" #include "log.h" -#include "mutexlock.h" #include "serverpool.h" #include "state.pb.h" #include "stream.h" @@ -40,8 +40,9 @@ bool parse_ip_address(const string &ip, sockaddr_storage *addr) ip.c_str()); return false; } - if (inet_pton(AF_INET6, ip.c_str(), &addr6->sin6_addr) != 1) { - log(ERROR, "'%s' is not a valid IPv6 address"); + string raw_ip(ip.begin() + 1, ip.end() - 1); + if (inet_pton(AF_INET6, raw_ip.c_str(), &addr6->sin6_addr) != 1) { + log(ERROR, "'%s' is not a valid IPv6 address", raw_ip.c_str()); return false; } } else { @@ -111,11 +112,8 @@ UDPInput::UDPInput(const string &url) construct_header(); - pthread_mutex_init(&stats_mutex, NULL); stats.url = url; - stats.bytes_received = 0; - stats.data_bytes_received = 0; - stats.connect_time = time(NULL); + stats.connect_time = time(nullptr); } UDPInput::UDPInput(const InputProto &serialized) @@ -129,14 +127,13 @@ UDPInput::UDPInput(const InputProto &serialized) construct_header(); - pthread_mutex_init(&stats_mutex, NULL); stats.url = url; stats.bytes_received = serialized.bytes_received(); stats.data_bytes_received = serialized.data_bytes_received(); if (serialized.has_connect_time()) { stats.connect_time = serialized.connect_time(); } else { - stats.connect_time = time(NULL); + stats.connect_time = time(nullptr); } } @@ -148,6 +145,7 @@ InputProto UDPInput::serialize() const serialized.set_bytes_received(stats.bytes_received); serialized.set_data_bytes_received(stats.data_bytes_received); serialized.set_connect_time(stats.connect_time); + serialized.set_is_metacube_encoded(false); return serialized; } @@ -201,7 +199,7 @@ void UDPInput::do_work() } // Wait for a packet, or a wakeup. - bool activity = wait_for_activity(sock, POLLIN, NULL); + bool activity = wait_for_activity(sock, POLLIN, nullptr); if (!activity) { // Most likely, should_stop was set. continue; @@ -219,19 +217,19 @@ void UDPInput::do_work() } { - MutexLock lock(&stats_mutex); + lock_guard lock(stats_mutex); stats.bytes_received += ret; stats.data_bytes_received += ret; } - - for (size_t i = 0; i < stream_indices.size(); ++i) { - servers->add_data(stream_indices[i], packet_buf, ret, SUITABLE_FOR_STREAM_START); + + for (size_t stream_index : stream_indices) { + servers->add_data(stream_index, packet_buf, ret, /*metacube_flags=*/0, /*pts=*/RationalPTS()); } } } InputStats UDPInput::get_stats() const { - MutexLock lock(&stats_mutex); + lock_guard lock(stats_mutex); return stats; }