1
0

whip-output.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. #include "whip-output.h"
  2. #include "whip-utils.h"
  3. #include <obs.hpp>
  4. /*
  5. * Sets the maximum size for a video fragment. Effective range is
  6. * 576-1470, with a lower value equating to more packets created,
  7. * but also better network compatability.
  8. */
  9. static uint16_t MAX_VIDEO_FRAGMENT_SIZE = 1200;
  10. const int signaling_media_id_length = 16;
  11. const char signaling_media_id_valid_char[] = "0123456789"
  12. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  13. "abcdefghijklmnopqrstuvwxyz";
  14. const std::string user_agent = generate_user_agent();
  15. const char *audio_mid = "0";
  16. const uint8_t audio_payload_type = 111;
  17. const char *video_mid = "1";
  18. const uint8_t video_payload_type = 96;
  19. // ~3 seconds of 8.5 Megabit video
  20. const int video_nack_buffer_size = 4000;
  21. WHIPOutput::WHIPOutput(obs_data_t *, obs_output_t *output)
  22. : output(output),
  23. endpoint_url(),
  24. bearer_token(),
  25. resource_url(),
  26. running(false),
  27. start_stop_mutex(),
  28. start_stop_thread(),
  29. base_ssrc(generate_random_u32()),
  30. peer_connection(nullptr),
  31. audio_track(nullptr),
  32. video_track(nullptr),
  33. total_bytes_sent(0),
  34. connect_time_ms(0),
  35. start_time_ns(0),
  36. last_audio_timestamp(0),
  37. last_video_timestamp(0)
  38. {
  39. }
  40. WHIPOutput::~WHIPOutput()
  41. {
  42. Stop();
  43. std::lock_guard<std::mutex> l(start_stop_mutex);
  44. if (start_stop_thread.joinable())
  45. start_stop_thread.join();
  46. }
  47. bool WHIPOutput::Start()
  48. {
  49. std::lock_guard<std::mutex> l(start_stop_mutex);
  50. if (!obs_output_can_begin_data_capture(output, 0))
  51. return false;
  52. if (!obs_output_initialize_encoders(output, 0))
  53. return false;
  54. if (start_stop_thread.joinable())
  55. start_stop_thread.join();
  56. start_stop_thread = std::thread(&WHIPOutput::StartThread, this);
  57. return true;
  58. }
  59. void WHIPOutput::Stop(bool signal)
  60. {
  61. std::lock_guard<std::mutex> l(start_stop_mutex);
  62. if (start_stop_thread.joinable())
  63. start_stop_thread.join();
  64. start_stop_thread = std::thread(&WHIPOutput::StopThread, this, signal);
  65. }
  66. void WHIPOutput::Data(struct encoder_packet *packet)
  67. {
  68. if (!packet) {
  69. Stop(false);
  70. obs_output_signal_stop(output, OBS_OUTPUT_ENCODE_ERROR);
  71. return;
  72. }
  73. if (audio_track && packet->type == OBS_ENCODER_AUDIO) {
  74. int64_t duration = packet->dts_usec - last_audio_timestamp;
  75. Send(packet->data, packet->size, duration, audio_track, audio_sr_reporter);
  76. last_audio_timestamp = packet->dts_usec;
  77. } else if (video_track && packet->type == OBS_ENCODER_VIDEO) {
  78. int64_t duration = packet->dts_usec - last_video_timestamp;
  79. Send(packet->data, packet->size, duration, video_track, video_sr_reporter);
  80. last_video_timestamp = packet->dts_usec;
  81. }
  82. }
  83. void WHIPOutput::ConfigureAudioTrack(std::string media_stream_id, std::string cname)
  84. {
  85. if (!obs_output_get_audio_encoder(output, 0)) {
  86. do_log(LOG_DEBUG, "Not configuring audio track: Audio encoder not assigned");
  87. return;
  88. }
  89. auto media_stream_track_id = std::string(media_stream_id + "-audio");
  90. uint32_t ssrc = base_ssrc;
  91. rtc::Description::Audio audio_description(audio_mid, rtc::Description::Direction::SendOnly);
  92. audio_description.addOpusCodec(audio_payload_type);
  93. audio_description.addSSRC(ssrc, cname, media_stream_id, media_stream_track_id);
  94. audio_track = peer_connection->addTrack(audio_description);
  95. auto rtp_config = std::make_shared<rtc::RtpPacketizationConfig>(ssrc, cname, audio_payload_type,
  96. rtc::OpusRtpPacketizer::DefaultClockRate);
  97. auto packetizer = std::make_shared<rtc::OpusRtpPacketizer>(rtp_config);
  98. audio_sr_reporter = std::make_shared<rtc::RtcpSrReporter>(rtp_config);
  99. auto nack_responder = std::make_shared<rtc::RtcpNackResponder>();
  100. packetizer->addToChain(audio_sr_reporter);
  101. packetizer->addToChain(nack_responder);
  102. audio_track->setMediaHandler(packetizer);
  103. }
  104. void WHIPOutput::ConfigureVideoTrack(std::string media_stream_id, std::string cname)
  105. {
  106. if (!obs_output_get_video_encoder(output)) {
  107. do_log(LOG_DEBUG, "Not configuring video track: Video encoder not assigned");
  108. return;
  109. }
  110. auto media_stream_track_id = std::string(media_stream_id + "-video");
  111. std::shared_ptr<rtc::RtpPacketizer> packetizer;
  112. // More predictable SSRC values between audio and video
  113. uint32_t ssrc = base_ssrc + 1;
  114. rtc::Description::Video video_description(video_mid, rtc::Description::Direction::SendOnly);
  115. video_description.addSSRC(ssrc, cname, media_stream_id, media_stream_track_id);
  116. auto rtp_config = std::make_shared<rtc::RtpPacketizationConfig>(ssrc, cname, video_payload_type,
  117. #if RTC_VERSION_MAJOR == 0 && RTC_VERSION_MINOR > 22 || RTC_VERSION_MAJOR > 0
  118. rtc::H264RtpPacketizer::ClockRate);
  119. #else
  120. rtc::H264RtpPacketizer::defaultClockRate);
  121. #endif
  122. const obs_encoder_t *encoder = obs_output_get_video_encoder2(output, 0);
  123. if (!encoder)
  124. return;
  125. OBSDataAutoRelease settings = obs_encoder_get_settings(encoder);
  126. auto video_bitrate = (int)obs_data_get_int(settings, "bitrate");
  127. const char *codec = obs_encoder_get_codec(encoder);
  128. if (strcmp("h264", codec) == 0) {
  129. video_description.addH264Codec(video_payload_type);
  130. packetizer = std::make_shared<rtc::H264RtpPacketizer>(rtc::H264RtpPacketizer::Separator::StartSequence,
  131. rtp_config, MAX_VIDEO_FRAGMENT_SIZE);
  132. #ifdef ENABLE_HEVC
  133. } else if (strcmp("hevc", codec) == 0) {
  134. video_description.addH265Codec(video_payload_type);
  135. packetizer = std::make_shared<rtc::H265RtpPacketizer>(rtc::H265RtpPacketizer::Separator::StartSequence,
  136. rtp_config, MAX_VIDEO_FRAGMENT_SIZE);
  137. #endif
  138. } else if (strcmp("av1", codec) == 0) {
  139. video_description.addAV1Codec(video_payload_type);
  140. packetizer = std::make_shared<rtc::AV1RtpPacketizer>(rtc::AV1RtpPacketizer::Packetization::TemporalUnit,
  141. rtp_config, MAX_VIDEO_FRAGMENT_SIZE);
  142. } else {
  143. do_log(LOG_ERROR, "Video codec not supported: %s", codec);
  144. return;
  145. }
  146. video_sr_reporter = std::make_shared<rtc::RtcpSrReporter>(rtp_config);
  147. packetizer->addToChain(video_sr_reporter);
  148. packetizer->addToChain(std::make_shared<rtc::RtcpNackResponder>(video_nack_buffer_size));
  149. if (video_bitrate != 0) {
  150. packetizer->addToChain(std::make_shared<rtc::PacingHandler>(static_cast<double>(video_bitrate * 10000),
  151. std::chrono::milliseconds(5)));
  152. }
  153. video_track = peer_connection->addTrack(video_description);
  154. video_track->setMediaHandler(packetizer);
  155. }
  156. /**
  157. * @brief Store connect info provided by the service.
  158. *
  159. * @return bool
  160. */
  161. bool WHIPOutput::Init()
  162. {
  163. obs_service_t *service = obs_output_get_service(output);
  164. if (!service) {
  165. obs_output_signal_stop(output, OBS_OUTPUT_ERROR);
  166. return false;
  167. }
  168. endpoint_url = obs_service_get_connect_info(service, OBS_SERVICE_CONNECT_INFO_SERVER_URL);
  169. if (endpoint_url.empty()) {
  170. obs_output_signal_stop(output, OBS_OUTPUT_BAD_PATH);
  171. return false;
  172. }
  173. bearer_token = obs_service_get_connect_info(service, OBS_SERVICE_CONNECT_INFO_BEARER_TOKEN);
  174. return true;
  175. }
  176. /**
  177. * @brief Set up the PeerConnection and media tracks.
  178. *
  179. * @return bool
  180. */
  181. bool WHIPOutput::Setup()
  182. {
  183. rtc::Configuration cfg;
  184. #if RTC_VERSION_MAJOR == 0 && RTC_VERSION_MINOR > 20 || RTC_VERSION_MAJOR > 0
  185. cfg.disableAutoGathering = true;
  186. #endif
  187. peer_connection = std::make_shared<rtc::PeerConnection>(cfg);
  188. peer_connection->onStateChange([this](rtc::PeerConnection::State state) {
  189. switch (state) {
  190. case rtc::PeerConnection::State::New:
  191. do_log(LOG_INFO, "PeerConnection state is now: New");
  192. break;
  193. case rtc::PeerConnection::State::Connecting:
  194. do_log(LOG_INFO, "PeerConnection state is now: Connecting");
  195. start_time_ns = os_gettime_ns();
  196. break;
  197. case rtc::PeerConnection::State::Connected:
  198. do_log(LOG_INFO, "PeerConnection state is now: Connected");
  199. connect_time_ms = (int)((os_gettime_ns() - start_time_ns) / 1000000.0);
  200. do_log(LOG_INFO, "Connect time: %dms", connect_time_ms.load());
  201. break;
  202. case rtc::PeerConnection::State::Disconnected:
  203. do_log(LOG_INFO, "PeerConnection state is now: Disconnected");
  204. Stop(false);
  205. obs_output_signal_stop(output, OBS_OUTPUT_DISCONNECTED);
  206. break;
  207. case rtc::PeerConnection::State::Failed:
  208. do_log(LOG_INFO, "PeerConnection state is now: Failed");
  209. Stop(false);
  210. obs_output_signal_stop(output, OBS_OUTPUT_ERROR);
  211. break;
  212. case rtc::PeerConnection::State::Closed:
  213. do_log(LOG_INFO, "PeerConnection state is now: Closed");
  214. break;
  215. }
  216. });
  217. std::string media_stream_id, cname;
  218. media_stream_id.reserve(signaling_media_id_length);
  219. cname.reserve(signaling_media_id_length);
  220. for (int i = 0; i < signaling_media_id_length; ++i) {
  221. media_stream_id += signaling_media_id_valid_char[rand() % (sizeof(signaling_media_id_valid_char) - 1)];
  222. cname += signaling_media_id_valid_char[rand() % (sizeof(signaling_media_id_valid_char) - 1)];
  223. }
  224. ConfigureAudioTrack(media_stream_id, cname);
  225. ConfigureVideoTrack(media_stream_id, cname);
  226. peer_connection->setLocalDescription();
  227. return true;
  228. }
  229. // Given a Link header extract URL/Username/Credential and create rtc::IceServer
  230. // <turn:turn.example.net>; username="user"; credential="myPassword";
  231. //
  232. // https://www.ietf.org/archive/id/draft-ietf-wish-whip-13.html#section-4.4
  233. void WHIPOutput::ParseLinkHeader(std::string val, std::vector<rtc::IceServer> &iceServers)
  234. {
  235. std::string url, username, password;
  236. auto extractUrl = [](std::string input) -> std::string {
  237. auto head = input.find("<") + 1;
  238. auto tail = input.find(">");
  239. if (head == std::string::npos || tail == std::string::npos) {
  240. return "";
  241. }
  242. return input.substr(head, tail - head);
  243. };
  244. auto extractValue = [](std::string input) -> std::string {
  245. auto head = input.find("\"") + 1;
  246. auto tail = input.find_last_of("\"");
  247. if (head == std::string::npos || tail == std::string::npos) {
  248. return "";
  249. }
  250. return input.substr(head, tail - head);
  251. };
  252. while (true) {
  253. std::string token = val;
  254. auto pos = token.find(";");
  255. if (pos != std::string::npos) {
  256. token = val.substr(0, pos);
  257. }
  258. if ((token.find("<stun:", 0) == 0) || (token.find("<turn:", 0) == 0)) {
  259. url = extractUrl(token);
  260. } else if (token.find("username=") != std::string::npos) {
  261. username = extractValue(token);
  262. } else if (token.find("credential=") != std::string::npos) {
  263. password = extractValue(token);
  264. }
  265. if (pos == std::string::npos) {
  266. break;
  267. }
  268. val.erase(0, pos + 1);
  269. }
  270. try {
  271. auto iceServer = rtc::IceServer(url);
  272. iceServer.username = username;
  273. iceServer.password = password;
  274. iceServers.push_back(iceServer);
  275. } catch (const std::invalid_argument &err) {
  276. do_log(LOG_WARNING, "Failed to construct ICE Server from %s: %s", val.c_str(), err.what());
  277. }
  278. }
  279. bool WHIPOutput::Connect()
  280. {
  281. struct curl_slist *headers = NULL;
  282. headers = curl_slist_append(headers, "Content-Type: application/sdp");
  283. if (!bearer_token.empty()) {
  284. auto bearer_token_header = std::string("Authorization: Bearer ") + bearer_token;
  285. headers = curl_slist_append(headers, bearer_token_header.c_str());
  286. }
  287. std::string read_buffer;
  288. std::vector<std::string> http_headers;
  289. auto offer_sdp = std::string(peer_connection->localDescription().value());
  290. #ifdef DEBUG_SDP
  291. do_log(LOG_DEBUG, "Offer SDP:\n%s", offer_sdp.c_str());
  292. #endif
  293. // Add user-agent to our requests
  294. headers = curl_slist_append(headers, user_agent.c_str());
  295. char error_buffer[CURL_ERROR_SIZE] = {};
  296. CURL *c = curl_easy_init();
  297. curl_easy_setopt(c, CURLOPT_WRITEFUNCTION, curl_writefunction);
  298. curl_easy_setopt(c, CURLOPT_WRITEDATA, (void *)&read_buffer);
  299. curl_easy_setopt(c, CURLOPT_HEADERFUNCTION, curl_header_function);
  300. curl_easy_setopt(c, CURLOPT_HEADERDATA, (void *)&http_headers);
  301. curl_easy_setopt(c, CURLOPT_HTTPHEADER, headers);
  302. curl_easy_setopt(c, CURLOPT_URL, endpoint_url.c_str());
  303. curl_easy_setopt(c, CURLOPT_POST, 1L);
  304. curl_easy_setopt(c, CURLOPT_COPYPOSTFIELDS, offer_sdp.c_str());
  305. curl_easy_setopt(c, CURLOPT_TIMEOUT, 8L);
  306. curl_easy_setopt(c, CURLOPT_FOLLOWLOCATION, 1L);
  307. curl_easy_setopt(c, CURLOPT_UNRESTRICTED_AUTH, 1L);
  308. curl_easy_setopt(c, CURLOPT_ERRORBUFFER, error_buffer);
  309. auto cleanup = [&]() {
  310. curl_easy_cleanup(c);
  311. curl_slist_free_all(headers);
  312. };
  313. CURLcode res = curl_easy_perform(c);
  314. if (res != CURLE_OK) {
  315. do_log(LOG_ERROR, "Connect failed: %s", error_buffer[0] ? error_buffer : curl_easy_strerror(res));
  316. cleanup();
  317. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  318. return false;
  319. }
  320. long response_code;
  321. curl_easy_getinfo(c, CURLINFO_RESPONSE_CODE, &response_code);
  322. if (response_code != 201) {
  323. do_log(LOG_ERROR, "Connect failed: HTTP endpoint returned response code %ld", response_code);
  324. cleanup();
  325. obs_output_signal_stop(output, OBS_OUTPUT_INVALID_STREAM);
  326. return false;
  327. }
  328. if (read_buffer.empty()) {
  329. do_log(LOG_ERROR, "Connect failed: No data returned from HTTP endpoint request");
  330. cleanup();
  331. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  332. return false;
  333. }
  334. long redirect_count = 0;
  335. curl_easy_getinfo(c, CURLINFO_REDIRECT_COUNT, &redirect_count);
  336. std::string last_location_header;
  337. size_t location_header_count = 0;
  338. for (auto &http_header : http_headers) {
  339. auto value = value_for_header("location", http_header);
  340. if (value.empty())
  341. continue;
  342. location_header_count++;
  343. last_location_header = value;
  344. }
  345. if (location_header_count < static_cast<size_t>(redirect_count) + 1) {
  346. do_log(LOG_ERROR, "WHIP server did not provide a resource URL via the Location header");
  347. cleanup();
  348. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  349. return false;
  350. }
  351. CURLU *url_builder = curl_url();
  352. // Parse Link headers to extract STUN/TURN server configuration URLs
  353. std::vector<rtc::IceServer> iceServers;
  354. for (auto &http_header : http_headers) {
  355. auto value = value_for_header("link", http_header);
  356. if (value.empty())
  357. continue;
  358. // Parse multiple links separated by ','
  359. for (auto end = value.find(","); end != std::string::npos; end = value.find(",")) {
  360. this->ParseLinkHeader(value.substr(0, end), iceServers);
  361. value = value.substr(end + 1);
  362. }
  363. this->ParseLinkHeader(value, iceServers);
  364. }
  365. // If Location header doesn't start with `http` it is a relative URL.
  366. // Construct a absolute URL using the host of the effective URL
  367. if (last_location_header.find("http") != 0) {
  368. char *effective_url = nullptr;
  369. curl_easy_getinfo(c, CURLINFO_EFFECTIVE_URL, &effective_url);
  370. if (effective_url == nullptr) {
  371. do_log(LOG_ERROR, "Failed to build Resource URL");
  372. cleanup();
  373. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  374. return false;
  375. }
  376. curl_url_set(url_builder, CURLUPART_URL, effective_url, 0);
  377. curl_url_set(url_builder, CURLUPART_PATH, last_location_header.c_str(), 0);
  378. curl_url_set(url_builder, CURLUPART_QUERY, "", 0);
  379. } else {
  380. curl_url_set(url_builder, CURLUPART_URL, last_location_header.c_str(), 0);
  381. }
  382. char *url = nullptr;
  383. CURLUcode rc = curl_url_get(url_builder, CURLUPART_URL, &url, CURLU_NO_DEFAULT_PORT);
  384. if (rc) {
  385. do_log(LOG_ERROR, "WHIP server provided a invalid resource URL via the Location header");
  386. cleanup();
  387. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  388. return false;
  389. }
  390. resource_url = url;
  391. curl_free(url);
  392. do_log(LOG_DEBUG, "WHIP Resource URL is: %s", resource_url.c_str());
  393. curl_url_cleanup(url_builder);
  394. #ifdef DEBUG_SDP
  395. do_log(LOG_DEBUG, "Answer SDP:\n%s", read_buffer.c_str());
  396. #endif
  397. auto response = std::string(read_buffer);
  398. response.erase(0, response.find("v=0"));
  399. rtc::Description answer(response, "answer");
  400. try {
  401. peer_connection->setRemoteDescription(answer);
  402. } catch (const std::invalid_argument &err) {
  403. do_log(LOG_ERROR, "WHIP server responded with invalid SDP: %s", err.what());
  404. cleanup();
  405. struct dstr error_message;
  406. dstr_init_copy(&error_message, obs_module_text("Error.InvalidSDP"));
  407. dstr_replace(&error_message, "%1", err.what());
  408. obs_output_set_last_error(output, error_message.array);
  409. dstr_free(&error_message);
  410. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  411. return false;
  412. } catch (const std::exception &err) {
  413. do_log(LOG_ERROR, "Failed to set remote description: %s", err.what());
  414. cleanup();
  415. struct dstr error_message;
  416. dstr_init_copy(&error_message, obs_module_text("Error.NoRemoteDescription"));
  417. dstr_replace(&error_message, "%1", err.what());
  418. obs_output_set_last_error(output, error_message.array);
  419. dstr_free(&error_message);
  420. obs_output_signal_stop(output, OBS_OUTPUT_CONNECT_FAILED);
  421. return false;
  422. }
  423. cleanup();
  424. #if RTC_VERSION_MAJOR == 0 && RTC_VERSION_MINOR > 20 || RTC_VERSION_MAJOR > 0
  425. peer_connection->gatherLocalCandidates(iceServers);
  426. #endif
  427. return true;
  428. }
  429. void WHIPOutput::StartThread()
  430. {
  431. if (!Init())
  432. return;
  433. if (!Setup())
  434. return;
  435. if (!Connect()) {
  436. peer_connection->close();
  437. peer_connection = nullptr;
  438. audio_track = nullptr;
  439. video_track = nullptr;
  440. return;
  441. }
  442. obs_output_begin_data_capture(output, 0);
  443. running = true;
  444. }
  445. void WHIPOutput::SendDelete()
  446. {
  447. if (resource_url.empty()) {
  448. do_log(LOG_DEBUG, "No resource URL available, not sending DELETE");
  449. return;
  450. }
  451. struct curl_slist *headers = NULL;
  452. if (!bearer_token.empty()) {
  453. auto bearer_token_header = std::string("Authorization: Bearer ") + bearer_token;
  454. headers = curl_slist_append(headers, bearer_token_header.c_str());
  455. }
  456. // Add user-agent to our requests
  457. headers = curl_slist_append(headers, user_agent.c_str());
  458. char error_buffer[CURL_ERROR_SIZE] = {};
  459. CURL *c = curl_easy_init();
  460. curl_easy_setopt(c, CURLOPT_HTTPHEADER, headers);
  461. curl_easy_setopt(c, CURLOPT_URL, resource_url.c_str());
  462. curl_easy_setopt(c, CURLOPT_CUSTOMREQUEST, "DELETE");
  463. curl_easy_setopt(c, CURLOPT_TIMEOUT, 8L);
  464. curl_easy_setopt(c, CURLOPT_ERRORBUFFER, error_buffer);
  465. auto cleanup = [&]() {
  466. curl_easy_cleanup(c);
  467. curl_slist_free_all(headers);
  468. };
  469. CURLcode res = curl_easy_perform(c);
  470. if (res != CURLE_OK) {
  471. do_log(LOG_WARNING, "DELETE request for resource URL failed: %s",
  472. error_buffer[0] ? error_buffer : curl_easy_strerror(res));
  473. cleanup();
  474. return;
  475. }
  476. long response_code;
  477. curl_easy_getinfo(c, CURLINFO_RESPONSE_CODE, &response_code);
  478. if (response_code != 200) {
  479. do_log(LOG_WARNING, "DELETE request for resource URL failed. HTTP Code: %ld", response_code);
  480. cleanup();
  481. return;
  482. }
  483. do_log(LOG_DEBUG, "Successfully performed DELETE request for resource URL");
  484. resource_url.clear();
  485. cleanup();
  486. }
  487. void WHIPOutput::StopThread(bool signal)
  488. {
  489. if (peer_connection != nullptr) {
  490. peer_connection->close();
  491. peer_connection = nullptr;
  492. audio_track = nullptr;
  493. video_track = nullptr;
  494. }
  495. SendDelete();
  496. /*
  497. * "signal" exists because we have to preserve the "running" state
  498. * across reconnect attempts. If we don't emit a signal if
  499. * something calls obs_output_stop() and it's reconnecting, you'll
  500. * desync the UI, as the output will be "stopped" and not
  501. * "reconnecting", but the "stop" signal will have never been
  502. * emitted.
  503. */
  504. if (running && signal) {
  505. obs_output_signal_stop(output, OBS_OUTPUT_SUCCESS);
  506. running = false;
  507. }
  508. total_bytes_sent = 0;
  509. connect_time_ms = 0;
  510. start_time_ns = 0;
  511. last_audio_timestamp = 0;
  512. last_video_timestamp = 0;
  513. }
  514. void WHIPOutput::Send(void *data, uintptr_t size, uint64_t duration, std::shared_ptr<rtc::Track> track,
  515. std::shared_ptr<rtc::RtcpSrReporter> rtcp_sr_reporter)
  516. {
  517. if (track == nullptr || !track->isOpen())
  518. return;
  519. std::vector<rtc::byte> sample{(rtc::byte *)data, (rtc::byte *)data + size};
  520. auto rtp_config = rtcp_sr_reporter->rtpConfig;
  521. // Sample time is in microseconds, we need to convert it to seconds
  522. auto elapsed_seconds = double(duration) / (1000.0 * 1000.0);
  523. // Get elapsed time in clock rate
  524. uint32_t elapsed_timestamp = rtp_config->secondsToTimestamp(elapsed_seconds);
  525. // Set new timestamp
  526. rtp_config->timestamp = rtp_config->timestamp + elapsed_timestamp;
  527. #if RTC_VERSION_MAJOR == 0 && RTC_VERSION_MINOR < 23
  528. // Get elapsed time in clock rate from last RTCP sender report
  529. auto report_elapsed_timestamp = rtp_config->timestamp - rtcp_sr_reporter->lastReportedTimestamp();
  530. // Check if last report was at least 1 second ago
  531. if (rtp_config->timestampToSeconds(report_elapsed_timestamp) > 1)
  532. rtcp_sr_reporter->setNeedsToReport();
  533. #endif
  534. try {
  535. track->send(sample);
  536. total_bytes_sent += sample.size();
  537. } catch (const std::exception &e) {
  538. do_log(LOG_ERROR, "error: %s ", e.what());
  539. }
  540. }
  541. void register_whip_output()
  542. {
  543. const uint32_t base_flags = OBS_OUTPUT_ENCODED | OBS_OUTPUT_SERVICE;
  544. const char *audio_codecs = "opus";
  545. #ifdef ENABLE_HEVC
  546. const char *video_codecs = "h264;hevc;av1";
  547. #else
  548. const char *video_codecs = "h264;av1";
  549. #endif
  550. struct obs_output_info info = {};
  551. info.id = "whip_output";
  552. info.flags = OBS_OUTPUT_AV | base_flags;
  553. info.get_name = [](void *) -> const char * {
  554. return obs_module_text("Output.Name");
  555. };
  556. info.create = [](obs_data_t *settings, obs_output_t *output) -> void * {
  557. return new WHIPOutput(settings, output);
  558. };
  559. info.destroy = [](void *priv_data) {
  560. delete static_cast<WHIPOutput *>(priv_data);
  561. };
  562. info.start = [](void *priv_data) -> bool {
  563. return static_cast<WHIPOutput *>(priv_data)->Start();
  564. };
  565. info.stop = [](void *priv_data, uint64_t) {
  566. static_cast<WHIPOutput *>(priv_data)->Stop();
  567. };
  568. info.encoded_packet = [](void *priv_data, struct encoder_packet *packet) {
  569. static_cast<WHIPOutput *>(priv_data)->Data(packet);
  570. };
  571. info.get_defaults = [](obs_data_t *) {
  572. };
  573. info.get_properties = [](void *) -> obs_properties_t * {
  574. return obs_properties_create();
  575. };
  576. info.get_total_bytes = [](void *priv_data) -> uint64_t {
  577. return (uint64_t)static_cast<WHIPOutput *>(priv_data)->GetTotalBytes();
  578. };
  579. info.get_connect_time_ms = [](void *priv_data) -> int {
  580. return static_cast<WHIPOutput *>(priv_data)->GetConnectTime();
  581. };
  582. info.encoded_video_codecs = video_codecs;
  583. info.encoded_audio_codecs = audio_codecs;
  584. info.protocols = "WHIP";
  585. obs_register_output(&info);
  586. info.id = "whip_output_video";
  587. info.flags = OBS_OUTPUT_VIDEO | base_flags;
  588. info.encoded_audio_codecs = nullptr;
  589. obs_register_output(&info);
  590. info.id = "whip_output_audio";
  591. info.flags = OBS_OUTPUT_AUDIO | base_flags;
  592. info.encoded_video_codecs = nullptr;
  593. info.encoded_audio_codecs = audio_codecs;
  594. obs_register_output(&info);
  595. }