whip-output.cpp 21 KB

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