obs-module.c 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  1. /******************************************************************************
  2. Copyright (C) 2023 by Lain Bailey <[email protected]>
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 2 of the License, or
  6. (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ******************************************************************************/
  14. #include "util/platform.h"
  15. #include "util/dstr.h"
  16. #include "obs-defs.h"
  17. #include "obs-internal.h"
  18. #include "obs-module.h"
  19. extern const char *get_module_extension(void);
  20. static inline int req_func_not_found(const char *name, const char *path)
  21. {
  22. blog(LOG_DEBUG,
  23. "Required module function '%s' in module '%s' not "
  24. "found, loading of module failed",
  25. name, path);
  26. return MODULE_MISSING_EXPORTS;
  27. }
  28. static int load_module_exports(struct obs_module *mod, const char *path)
  29. {
  30. mod->load = os_dlsym(mod->module, "obs_module_load");
  31. if (!mod->load)
  32. return req_func_not_found("obs_module_load", path);
  33. mod->set_pointer = os_dlsym(mod->module, "obs_module_set_pointer");
  34. if (!mod->set_pointer)
  35. return req_func_not_found("obs_module_set_pointer", path);
  36. mod->ver = os_dlsym(mod->module, "obs_module_ver");
  37. if (!mod->ver)
  38. return req_func_not_found("obs_module_ver", path);
  39. /* optional exports */
  40. mod->unload = os_dlsym(mod->module, "obs_module_unload");
  41. mod->post_load = os_dlsym(mod->module, "obs_module_post_load");
  42. mod->set_locale = os_dlsym(mod->module, "obs_module_set_locale");
  43. mod->free_locale = os_dlsym(mod->module, "obs_module_free_locale");
  44. mod->name = os_dlsym(mod->module, "obs_module_name");
  45. mod->description = os_dlsym(mod->module, "obs_module_description");
  46. mod->author = os_dlsym(mod->module, "obs_module_author");
  47. mod->get_string = os_dlsym(mod->module, "obs_module_get_string");
  48. return MODULE_SUCCESS;
  49. }
  50. bool obs_module_get_locale_string(const obs_module_t *mod,
  51. const char *lookup_string,
  52. const char **translated_string)
  53. {
  54. if (mod->get_string) {
  55. return mod->get_string(lookup_string, translated_string);
  56. }
  57. return false;
  58. }
  59. const char *obs_module_get_locale_text(const obs_module_t *mod,
  60. const char *text)
  61. {
  62. const char *str = text;
  63. obs_module_get_locale_string(mod, text, &str);
  64. return str;
  65. }
  66. static inline char *get_module_name(const char *file)
  67. {
  68. static size_t ext_len = 0;
  69. struct dstr name = {0};
  70. if (ext_len == 0) {
  71. const char *ext = get_module_extension();
  72. ext_len = strlen(ext);
  73. }
  74. dstr_copy(&name, file);
  75. dstr_resize(&name, name.len - ext_len);
  76. return name.array;
  77. }
  78. #ifdef _WIN32
  79. extern void reset_win32_symbol_paths(void);
  80. #endif
  81. int obs_open_module(obs_module_t **module, const char *path,
  82. const char *data_path)
  83. {
  84. struct obs_module mod = {0};
  85. int errorcode;
  86. if (!module || !path || !obs)
  87. return MODULE_ERROR;
  88. #ifdef __APPLE__
  89. /* HACK: Do not load obsolete obs-browser build on macOS; the
  90. * obs-browser plugin used to live in the Application Support
  91. * directory. */
  92. if (astrstri(path, "Library/Application Support/obs-studio") != NULL &&
  93. astrstri(path, "obs-browser") != NULL) {
  94. blog(LOG_WARNING, "Ignoring old obs-browser.so version");
  95. return MODULE_HARDCODED_SKIP;
  96. }
  97. #endif
  98. blog(LOG_DEBUG, "---------------------------------");
  99. mod.module = os_dlopen(path);
  100. if (!mod.module) {
  101. blog(LOG_WARNING, "Module '%s' not loaded", path);
  102. return MODULE_FILE_NOT_FOUND;
  103. }
  104. errorcode = load_module_exports(&mod, path);
  105. if (errorcode != MODULE_SUCCESS)
  106. return errorcode;
  107. mod.bin_path = bstrdup(path);
  108. mod.file = strrchr(mod.bin_path, '/');
  109. mod.file = (!mod.file) ? mod.bin_path : (mod.file + 1);
  110. mod.mod_name = get_module_name(mod.file);
  111. mod.data_path = bstrdup(data_path);
  112. mod.next = obs->first_module;
  113. if (mod.file) {
  114. blog(LOG_DEBUG, "Loading module: %s", mod.file);
  115. }
  116. *module = bmemdup(&mod, sizeof(mod));
  117. obs->first_module = (*module);
  118. mod.set_pointer(*module);
  119. if (mod.set_locale)
  120. mod.set_locale(obs->locale);
  121. return MODULE_SUCCESS;
  122. }
  123. bool obs_init_module(obs_module_t *module)
  124. {
  125. if (!module || !obs)
  126. return false;
  127. if (module->loaded)
  128. return true;
  129. const char *profile_name =
  130. profile_store_name(obs_get_profiler_name_store(),
  131. "obs_init_module(%s)", module->file);
  132. profile_start(profile_name);
  133. module->loaded = module->load();
  134. if (!module->loaded)
  135. blog(LOG_WARNING, "Failed to initialize module '%s'",
  136. module->file);
  137. profile_end(profile_name);
  138. return module->loaded;
  139. }
  140. void obs_log_loaded_modules(void)
  141. {
  142. blog(LOG_INFO, " Loaded Modules:");
  143. for (obs_module_t *mod = obs->first_module; !!mod; mod = mod->next)
  144. blog(LOG_INFO, " %s", mod->file);
  145. }
  146. const char *obs_get_module_file_name(obs_module_t *module)
  147. {
  148. return module ? module->file : NULL;
  149. }
  150. const char *obs_get_module_name(obs_module_t *module)
  151. {
  152. return (module && module->name) ? module->name() : NULL;
  153. }
  154. const char *obs_get_module_author(obs_module_t *module)
  155. {
  156. return (module && module->author) ? module->author() : NULL;
  157. }
  158. const char *obs_get_module_description(obs_module_t *module)
  159. {
  160. return (module && module->description) ? module->description() : NULL;
  161. }
  162. const char *obs_get_module_binary_path(obs_module_t *module)
  163. {
  164. return module ? module->bin_path : NULL;
  165. }
  166. const char *obs_get_module_data_path(obs_module_t *module)
  167. {
  168. return module ? module->data_path : NULL;
  169. }
  170. obs_module_t *obs_get_module(const char *name)
  171. {
  172. obs_module_t *module = obs->first_module;
  173. while (module) {
  174. if (strcmp(module->mod_name, name) == 0) {
  175. return module;
  176. }
  177. module = module->next;
  178. }
  179. return NULL;
  180. }
  181. void *obs_get_module_lib(obs_module_t *module)
  182. {
  183. return module ? module->module : NULL;
  184. }
  185. char *obs_find_module_file(obs_module_t *module, const char *file)
  186. {
  187. struct dstr output = {0};
  188. if (!file)
  189. file = "";
  190. if (!module)
  191. return NULL;
  192. dstr_copy(&output, module->data_path);
  193. if (!dstr_is_empty(&output) && dstr_end(&output) != '/' && *file)
  194. dstr_cat_ch(&output, '/');
  195. dstr_cat(&output, file);
  196. if (!os_file_exists(output.array))
  197. dstr_free(&output);
  198. return output.array;
  199. }
  200. char *obs_module_get_config_path(obs_module_t *module, const char *file)
  201. {
  202. struct dstr output = {0};
  203. dstr_copy(&output, obs->module_config_path);
  204. if (!dstr_is_empty(&output) && dstr_end(&output) != '/')
  205. dstr_cat_ch(&output, '/');
  206. dstr_cat(&output, module->mod_name);
  207. dstr_cat_ch(&output, '/');
  208. dstr_cat(&output, file);
  209. return output.array;
  210. }
  211. void obs_add_module_path(const char *bin, const char *data)
  212. {
  213. struct obs_module_path omp;
  214. if (!obs || !bin || !data)
  215. return;
  216. omp.bin = bstrdup(bin);
  217. omp.data = bstrdup(data);
  218. da_push_back(obs->module_paths, &omp);
  219. }
  220. void obs_add_safe_module(const char *name)
  221. {
  222. if (!obs || !name)
  223. return;
  224. char *item = bstrdup(name);
  225. da_push_back(obs->safe_modules, &item);
  226. }
  227. extern void get_plugin_info(const char *path, bool *is_obs_plugin,
  228. bool *can_load);
  229. struct fail_info {
  230. struct dstr fail_modules;
  231. size_t fail_count;
  232. };
  233. static bool is_safe_module(const char *name)
  234. {
  235. if (!obs->safe_modules.num)
  236. return true;
  237. for (size_t i = 0; i < obs->safe_modules.num; i++) {
  238. if (strcmp(name, obs->safe_modules.array[i]) == 0)
  239. return true;
  240. }
  241. return false;
  242. }
  243. static void load_all_callback(void *param, const struct obs_module_info2 *info)
  244. {
  245. struct fail_info *fail_info = param;
  246. obs_module_t *module;
  247. bool is_obs_plugin;
  248. bool can_load_obs_plugin;
  249. get_plugin_info(info->bin_path, &is_obs_plugin, &can_load_obs_plugin);
  250. if (!is_obs_plugin) {
  251. blog(LOG_WARNING, "Skipping module '%s', not an OBS plugin",
  252. info->bin_path);
  253. return;
  254. }
  255. if (!is_safe_module(info->name)) {
  256. blog(LOG_WARNING, "Skipping module '%s', not on safe list",
  257. info->name);
  258. return;
  259. }
  260. if (!can_load_obs_plugin) {
  261. blog(LOG_WARNING,
  262. "Skipping module '%s' due to possible "
  263. "import conflicts",
  264. info->bin_path);
  265. goto load_failure;
  266. }
  267. int code = obs_open_module(&module, info->bin_path, info->data_path);
  268. switch (code) {
  269. case MODULE_MISSING_EXPORTS:
  270. blog(LOG_DEBUG,
  271. "Failed to load module file '%s', not an OBS plugin",
  272. info->bin_path);
  273. return;
  274. case MODULE_FILE_NOT_FOUND:
  275. blog(LOG_DEBUG,
  276. "Failed to load module file '%s', file not found",
  277. info->bin_path);
  278. return;
  279. case MODULE_ERROR:
  280. blog(LOG_DEBUG, "Failed to load module file '%s'",
  281. info->bin_path);
  282. goto load_failure;
  283. case MODULE_INCOMPATIBLE_VER:
  284. blog(LOG_DEBUG,
  285. "Failed to load module file '%s', incompatible version",
  286. info->bin_path);
  287. goto load_failure;
  288. case MODULE_HARDCODED_SKIP:
  289. return;
  290. }
  291. if (!obs_init_module(module))
  292. free_module(module);
  293. UNUSED_PARAMETER(param);
  294. return;
  295. load_failure:
  296. if (fail_info) {
  297. dstr_cat(&fail_info->fail_modules, info->name);
  298. dstr_cat(&fail_info->fail_modules, ";");
  299. fail_info->fail_count++;
  300. }
  301. }
  302. static const char *obs_load_all_modules_name = "obs_load_all_modules";
  303. #ifdef _WIN32
  304. static const char *reset_win32_symbol_paths_name = "reset_win32_symbol_paths";
  305. #endif
  306. void obs_load_all_modules(void)
  307. {
  308. profile_start(obs_load_all_modules_name);
  309. obs_find_modules2(load_all_callback, NULL);
  310. #ifdef _WIN32
  311. profile_start(reset_win32_symbol_paths_name);
  312. reset_win32_symbol_paths();
  313. profile_end(reset_win32_symbol_paths_name);
  314. #endif
  315. profile_end(obs_load_all_modules_name);
  316. }
  317. static const char *obs_load_all_modules2_name = "obs_load_all_modules2";
  318. void obs_load_all_modules2(struct obs_module_failure_info *mfi)
  319. {
  320. struct fail_info fail_info = {0};
  321. memset(mfi, 0, sizeof(*mfi));
  322. profile_start(obs_load_all_modules2_name);
  323. obs_find_modules2(load_all_callback, &fail_info);
  324. #ifdef _WIN32
  325. profile_start(reset_win32_symbol_paths_name);
  326. reset_win32_symbol_paths();
  327. profile_end(reset_win32_symbol_paths_name);
  328. #endif
  329. profile_end(obs_load_all_modules2_name);
  330. mfi->count = fail_info.fail_count;
  331. mfi->failed_modules =
  332. strlist_split(fail_info.fail_modules.array, ';', false);
  333. dstr_free(&fail_info.fail_modules);
  334. }
  335. void obs_module_failure_info_free(struct obs_module_failure_info *mfi)
  336. {
  337. if (mfi->failed_modules) {
  338. bfree(mfi->failed_modules);
  339. mfi->failed_modules = NULL;
  340. }
  341. }
  342. void obs_post_load_modules(void)
  343. {
  344. for (obs_module_t *mod = obs->first_module; !!mod; mod = mod->next)
  345. if (mod->post_load)
  346. mod->post_load();
  347. }
  348. static inline void make_data_dir(struct dstr *parsed_data_dir,
  349. const char *data_dir, const char *name)
  350. {
  351. dstr_copy(parsed_data_dir, data_dir);
  352. dstr_replace(parsed_data_dir, "%module%", name);
  353. if (dstr_end(parsed_data_dir) == '/')
  354. dstr_resize(parsed_data_dir, parsed_data_dir->len - 1);
  355. }
  356. static char *make_data_directory(const char *module_name, const char *data_dir)
  357. {
  358. struct dstr parsed_data_dir = {0};
  359. bool found = false;
  360. make_data_dir(&parsed_data_dir, data_dir, module_name);
  361. found = os_file_exists(parsed_data_dir.array);
  362. if (!found && astrcmpi_n(module_name, "lib", 3) == 0)
  363. make_data_dir(&parsed_data_dir, data_dir, module_name + 3);
  364. return parsed_data_dir.array;
  365. }
  366. static bool parse_binary_from_directory(struct dstr *parsed_bin_path,
  367. const char *bin_path, const char *file)
  368. {
  369. struct dstr directory = {0};
  370. bool found = true;
  371. dstr_copy(&directory, bin_path);
  372. dstr_replace(&directory, "%module%", file);
  373. if (dstr_end(&directory) != '/')
  374. dstr_cat_ch(&directory, '/');
  375. dstr_copy_dstr(parsed_bin_path, &directory);
  376. dstr_cat(parsed_bin_path, file);
  377. #ifdef __APPLE__
  378. if (!os_file_exists(parsed_bin_path->array)) {
  379. dstr_cat(parsed_bin_path, ".so");
  380. }
  381. #else
  382. dstr_cat(parsed_bin_path, get_module_extension());
  383. #endif
  384. if (!os_file_exists(parsed_bin_path->array)) {
  385. /* Legacy fallback: Check for plugin with .so suffix*/
  386. dstr_cat(parsed_bin_path, ".so");
  387. /* if the file doesn't exist, check with 'lib' prefix */
  388. dstr_copy_dstr(parsed_bin_path, &directory);
  389. dstr_cat(parsed_bin_path, "lib");
  390. dstr_cat(parsed_bin_path, file);
  391. dstr_cat(parsed_bin_path, get_module_extension());
  392. /* if neither exist, don't include this as a library */
  393. if (!os_file_exists(parsed_bin_path->array)) {
  394. dstr_free(parsed_bin_path);
  395. found = false;
  396. }
  397. }
  398. dstr_free(&directory);
  399. return found;
  400. }
  401. static void process_found_module(struct obs_module_path *omp, const char *path,
  402. bool directory,
  403. obs_find_module_callback2_t callback,
  404. void *param)
  405. {
  406. struct obs_module_info2 info;
  407. struct dstr name = {0};
  408. struct dstr parsed_bin_path = {0};
  409. const char *file;
  410. char *parsed_data_dir;
  411. bool bin_found = true;
  412. file = strrchr(path, '/');
  413. file = file ? (file + 1) : path;
  414. if (strcmp(file, ".") == 0 || strcmp(file, "..") == 0)
  415. return;
  416. dstr_copy(&name, file);
  417. char *ext = strrchr(name.array, '.');
  418. if (ext)
  419. dstr_resize(&name, ext - name.array);
  420. if (!directory) {
  421. dstr_copy(&parsed_bin_path, path);
  422. } else {
  423. bin_found = parse_binary_from_directory(&parsed_bin_path,
  424. omp->bin, name.array);
  425. }
  426. parsed_data_dir = make_data_directory(name.array, omp->data);
  427. if (parsed_data_dir && bin_found) {
  428. info.bin_path = parsed_bin_path.array;
  429. info.data_path = parsed_data_dir;
  430. info.name = name.array;
  431. callback(param, &info);
  432. }
  433. bfree(parsed_data_dir);
  434. dstr_free(&name);
  435. dstr_free(&parsed_bin_path);
  436. }
  437. static void find_modules_in_path(struct obs_module_path *omp,
  438. obs_find_module_callback2_t callback,
  439. void *param)
  440. {
  441. struct dstr search_path = {0};
  442. char *module_start;
  443. bool search_directories = false;
  444. os_glob_t *gi;
  445. dstr_copy(&search_path, omp->bin);
  446. module_start = strstr(search_path.array, "%module%");
  447. if (module_start) {
  448. dstr_resize(&search_path, module_start - search_path.array);
  449. search_directories = true;
  450. }
  451. if (!dstr_is_empty(&search_path) && dstr_end(&search_path) != '/')
  452. dstr_cat_ch(&search_path, '/');
  453. dstr_cat_ch(&search_path, '*');
  454. if (!search_directories)
  455. dstr_cat(&search_path, get_module_extension());
  456. if (os_glob(search_path.array, 0, &gi) == 0) {
  457. for (size_t i = 0; i < gi->gl_pathc; i++) {
  458. if (search_directories == gi->gl_pathv[i].directory)
  459. process_found_module(omp, gi->gl_pathv[i].path,
  460. search_directories,
  461. callback, param);
  462. }
  463. os_globfree(gi);
  464. }
  465. dstr_free(&search_path);
  466. }
  467. void obs_find_modules2(obs_find_module_callback2_t callback, void *param)
  468. {
  469. if (!obs)
  470. return;
  471. for (size_t i = 0; i < obs->module_paths.num; i++) {
  472. struct obs_module_path *omp = obs->module_paths.array + i;
  473. find_modules_in_path(omp, callback, param);
  474. }
  475. }
  476. void obs_find_modules(obs_find_module_callback_t callback, void *param)
  477. {
  478. /* the structure is ABI compatible so we can just cast the callback */
  479. obs_find_modules2((obs_find_module_callback2_t)callback, param);
  480. }
  481. void obs_enum_modules(obs_enum_module_callback_t callback, void *param)
  482. {
  483. struct obs_module *module;
  484. if (!obs)
  485. return;
  486. module = obs->first_module;
  487. while (module) {
  488. callback(param, module);
  489. module = module->next;
  490. }
  491. }
  492. void free_module(struct obs_module *mod)
  493. {
  494. if (!mod)
  495. return;
  496. if (mod->module) {
  497. if (mod->free_locale)
  498. mod->free_locale();
  499. if (mod->loaded && mod->unload)
  500. mod->unload();
  501. /* there is no real reason to close the dynamic libraries,
  502. * and sometimes this can cause issues. */
  503. /* os_dlclose(mod->module); */
  504. }
  505. for (obs_module_t *m = obs->first_module; !!m; m = m->next) {
  506. if (m->next == mod) {
  507. m->next = mod->next;
  508. break;
  509. }
  510. }
  511. if (obs->first_module == mod)
  512. obs->first_module = mod->next;
  513. bfree(mod->mod_name);
  514. bfree(mod->bin_path);
  515. bfree(mod->data_path);
  516. bfree(mod);
  517. }
  518. lookup_t *obs_module_load_locale(obs_module_t *module,
  519. const char *default_locale, const char *locale)
  520. {
  521. struct dstr str = {0};
  522. lookup_t *lookup = NULL;
  523. if (!module || !default_locale || !locale) {
  524. blog(LOG_WARNING, "obs_module_load_locale: Invalid parameters");
  525. return NULL;
  526. }
  527. dstr_copy(&str, "locale/");
  528. dstr_cat(&str, default_locale);
  529. dstr_cat(&str, ".ini");
  530. char *file = obs_find_module_file(module, str.array);
  531. if (file)
  532. lookup = text_lookup_create(file);
  533. bfree(file);
  534. if (!lookup) {
  535. blog(LOG_WARNING, "Failed to load '%s' text for module: '%s'",
  536. default_locale, module->file);
  537. goto cleanup;
  538. }
  539. if (astrcmpi(locale, default_locale) == 0)
  540. goto cleanup;
  541. dstr_copy(&str, "/locale/");
  542. dstr_cat(&str, locale);
  543. dstr_cat(&str, ".ini");
  544. file = obs_find_module_file(module, str.array);
  545. if (!text_lookup_add(lookup, file))
  546. blog(LOG_WARNING, "Failed to load '%s' text for module: '%s'",
  547. locale, module->file);
  548. bfree(file);
  549. cleanup:
  550. dstr_free(&str);
  551. return lookup;
  552. }
  553. #define REGISTER_OBS_DEF(size_var, structure, dest, info) \
  554. do { \
  555. struct structure data = {0}; \
  556. if (!size_var) { \
  557. blog(LOG_ERROR, "Tried to register " #structure \
  558. " outside of obs_module_load"); \
  559. return; \
  560. } \
  561. \
  562. if (size_var > sizeof(data)) { \
  563. blog(LOG_ERROR, \
  564. "Tried to register " #structure \
  565. " with size %llu which is more " \
  566. "than libobs currently supports " \
  567. "(%llu)", \
  568. (long long unsigned)size_var, \
  569. (long long unsigned)sizeof(data)); \
  570. goto error; \
  571. } \
  572. \
  573. memcpy(&data, info, size_var); \
  574. da_push_back(dest, &data); \
  575. } while (false)
  576. #define HAS_VAL(type, info, val) \
  577. ((offsetof(type, val) + sizeof(info->val) <= size) && info->val)
  578. #define CHECK_REQUIRED_VAL(type, info, val, func) \
  579. do { \
  580. if (!HAS_VAL(type, info, val)) { \
  581. blog(LOG_ERROR, \
  582. "Required value '" #val "' for " \
  583. "'%s' not found. " #func " failed.", \
  584. info->id); \
  585. goto error; \
  586. } \
  587. } while (false)
  588. #define CHECK_REQUIRED_VAL_EITHER(type, info, val1, val2, func) \
  589. do { \
  590. if (!HAS_VAL(type, info, val1) && \
  591. !HAS_VAL(type, info, val2)) { \
  592. blog(LOG_ERROR, \
  593. "Neither '" #val1 "' nor '" #val2 "' " \
  594. "for '%s' found. " #func " failed.", \
  595. info->id); \
  596. goto error; \
  597. } \
  598. } while (false)
  599. #define HANDLE_ERROR(size_var, structure, info) \
  600. do { \
  601. struct structure data = {0}; \
  602. if (!size_var) \
  603. return; \
  604. \
  605. memcpy(&data, info, \
  606. sizeof(data) < size_var ? sizeof(data) : size_var); \
  607. \
  608. if (data.type_data && data.free_type_data) \
  609. data.free_type_data(data.type_data); \
  610. } while (false)
  611. #define source_warn(format, ...) \
  612. blog(LOG_WARNING, "obs_register_source: " format, ##__VA_ARGS__)
  613. #define output_warn(format, ...) \
  614. blog(LOG_WARNING, "obs_register_output: " format, ##__VA_ARGS__)
  615. #define encoder_warn(format, ...) \
  616. blog(LOG_WARNING, "obs_register_encoder: " format, ##__VA_ARGS__)
  617. #define service_warn(format, ...) \
  618. blog(LOG_WARNING, "obs_register_service: " format, ##__VA_ARGS__)
  619. void obs_register_source_s(const struct obs_source_info *info, size_t size)
  620. {
  621. struct obs_source_info data = {0};
  622. obs_source_info_array_t *array = NULL;
  623. if (info->type == OBS_SOURCE_TYPE_INPUT) {
  624. array = &obs->input_types;
  625. } else if (info->type == OBS_SOURCE_TYPE_FILTER) {
  626. array = &obs->filter_types;
  627. } else if (info->type == OBS_SOURCE_TYPE_TRANSITION) {
  628. array = &obs->transition_types;
  629. } else if (info->type != OBS_SOURCE_TYPE_SCENE) {
  630. source_warn("Tried to register unknown source type: %u",
  631. info->type);
  632. goto error;
  633. }
  634. if (get_source_info2(info->id, info->version)) {
  635. source_warn("Source '%s' already exists! "
  636. "Duplicate library?",
  637. info->id);
  638. goto error;
  639. }
  640. if (size > sizeof(data)) {
  641. source_warn("Tried to register obs_source_info with size "
  642. "%llu which is more than libobs currently "
  643. "supports (%llu)",
  644. (long long unsigned)size,
  645. (long long unsigned)sizeof(data));
  646. goto error;
  647. }
  648. memcpy(&data, info, size);
  649. /* mark audio-only filters as an async filter categorically */
  650. if (data.type == OBS_SOURCE_TYPE_FILTER) {
  651. if ((data.output_flags & OBS_SOURCE_VIDEO) == 0)
  652. data.output_flags |= OBS_SOURCE_ASYNC;
  653. }
  654. if (data.type == OBS_SOURCE_TYPE_TRANSITION) {
  655. if (data.get_width)
  656. source_warn("get_width ignored registering "
  657. "transition '%s'",
  658. data.id);
  659. if (data.get_height)
  660. source_warn("get_height ignored registering "
  661. "transition '%s'",
  662. data.id);
  663. data.output_flags |= OBS_SOURCE_COMPOSITE | OBS_SOURCE_VIDEO |
  664. OBS_SOURCE_CUSTOM_DRAW;
  665. }
  666. if ((data.output_flags & OBS_SOURCE_COMPOSITE) != 0) {
  667. if ((data.output_flags & OBS_SOURCE_AUDIO) != 0) {
  668. source_warn("Source '%s': Composite sources "
  669. "cannot be audio sources",
  670. info->id);
  671. goto error;
  672. }
  673. if ((data.output_flags & OBS_SOURCE_ASYNC) != 0) {
  674. source_warn("Source '%s': Composite sources "
  675. "cannot be async sources",
  676. info->id);
  677. goto error;
  678. }
  679. }
  680. #define CHECK_REQUIRED_VAL_(info, val, func) \
  681. CHECK_REQUIRED_VAL(struct obs_source_info, info, val, func)
  682. CHECK_REQUIRED_VAL_(info, get_name, obs_register_source);
  683. if (info->type != OBS_SOURCE_TYPE_FILTER &&
  684. info->type != OBS_SOURCE_TYPE_TRANSITION &&
  685. (info->output_flags & OBS_SOURCE_VIDEO) != 0 &&
  686. (info->output_flags & OBS_SOURCE_ASYNC) == 0) {
  687. CHECK_REQUIRED_VAL_(info, get_width, obs_register_source);
  688. CHECK_REQUIRED_VAL_(info, get_height, obs_register_source);
  689. }
  690. if ((data.output_flags & OBS_SOURCE_COMPOSITE) != 0) {
  691. CHECK_REQUIRED_VAL_(info, audio_render, obs_register_source);
  692. }
  693. #undef CHECK_REQUIRED_VAL_
  694. /* version-related stuff */
  695. data.unversioned_id = data.id;
  696. if (data.version) {
  697. struct dstr versioned_id = {0};
  698. dstr_printf(&versioned_id, "%s_v%d", data.id,
  699. (int)data.version);
  700. data.id = versioned_id.array;
  701. } else {
  702. data.id = bstrdup(data.id);
  703. }
  704. if (array)
  705. da_push_back(*array, &data);
  706. da_push_back(obs->source_types, &data);
  707. return;
  708. error:
  709. HANDLE_ERROR(size, obs_source_info, info);
  710. }
  711. void obs_register_output_s(const struct obs_output_info *info, size_t size)
  712. {
  713. if (find_output(info->id)) {
  714. output_warn("Output id '%s' already exists! "
  715. "Duplicate library?",
  716. info->id);
  717. goto error;
  718. }
  719. #define CHECK_REQUIRED_VAL_(info, val, func) \
  720. CHECK_REQUIRED_VAL(struct obs_output_info, info, val, func)
  721. CHECK_REQUIRED_VAL_(info, get_name, obs_register_output);
  722. CHECK_REQUIRED_VAL_(info, create, obs_register_output);
  723. CHECK_REQUIRED_VAL_(info, destroy, obs_register_output);
  724. CHECK_REQUIRED_VAL_(info, start, obs_register_output);
  725. CHECK_REQUIRED_VAL_(info, stop, obs_register_output);
  726. if (info->flags & OBS_OUTPUT_SERVICE)
  727. CHECK_REQUIRED_VAL_(info, protocols, obs_register_output);
  728. if (info->flags & OBS_OUTPUT_ENCODED) {
  729. CHECK_REQUIRED_VAL_(info, encoded_packet, obs_register_output);
  730. } else {
  731. if (info->flags & OBS_OUTPUT_VIDEO)
  732. CHECK_REQUIRED_VAL_(info, raw_video,
  733. obs_register_output);
  734. if (info->flags & OBS_OUTPUT_AUDIO) {
  735. if (info->flags & OBS_OUTPUT_MULTI_TRACK) {
  736. CHECK_REQUIRED_VAL_(info, raw_audio2,
  737. obs_register_output);
  738. } else {
  739. CHECK_REQUIRED_VAL_(info, raw_audio,
  740. obs_register_output);
  741. }
  742. }
  743. }
  744. #undef CHECK_REQUIRED_VAL_
  745. REGISTER_OBS_DEF(size, obs_output_info, obs->output_types, info);
  746. if (info->flags & OBS_OUTPUT_SERVICE) {
  747. char **protocols = strlist_split(info->protocols, ';', false);
  748. for (char **protocol = protocols; *protocol; ++protocol) {
  749. bool skip = false;
  750. for (size_t i = 0; i < obs->data.protocols.num; i++) {
  751. if (strcmp(*protocol,
  752. obs->data.protocols.array[i]) == 0)
  753. skip = true;
  754. }
  755. if (skip)
  756. continue;
  757. char *new_prtcl = bstrdup(*protocol);
  758. da_push_back(obs->data.protocols, &new_prtcl);
  759. }
  760. strlist_free(protocols);
  761. }
  762. return;
  763. error:
  764. HANDLE_ERROR(size, obs_output_info, info);
  765. }
  766. void obs_register_encoder_s(const struct obs_encoder_info *info, size_t size)
  767. {
  768. if (find_encoder(info->id)) {
  769. encoder_warn("Encoder id '%s' already exists! "
  770. "Duplicate library?",
  771. info->id);
  772. goto error;
  773. }
  774. #define CHECK_REQUIRED_VAL_(info, val, func) \
  775. CHECK_REQUIRED_VAL(struct obs_encoder_info, info, val, func)
  776. CHECK_REQUIRED_VAL_(info, get_name, obs_register_encoder);
  777. CHECK_REQUIRED_VAL_(info, create, obs_register_encoder);
  778. CHECK_REQUIRED_VAL_(info, destroy, obs_register_encoder);
  779. if ((info->caps & OBS_ENCODER_CAP_PASS_TEXTURE) != 0)
  780. CHECK_REQUIRED_VAL_EITHER(struct obs_encoder_info, info,
  781. encode_texture, encode_texture2,
  782. obs_register_encoder);
  783. else
  784. CHECK_REQUIRED_VAL_(info, encode, obs_register_encoder);
  785. if (info->type == OBS_ENCODER_AUDIO)
  786. CHECK_REQUIRED_VAL_(info, get_frame_size, obs_register_encoder);
  787. #undef CHECK_REQUIRED_VAL_
  788. REGISTER_OBS_DEF(size, obs_encoder_info, obs->encoder_types, info);
  789. return;
  790. error:
  791. HANDLE_ERROR(size, obs_encoder_info, info);
  792. }
  793. void obs_register_service_s(const struct obs_service_info *info, size_t size)
  794. {
  795. if (find_service(info->id)) {
  796. service_warn("Service id '%s' already exists! "
  797. "Duplicate library?",
  798. info->id);
  799. goto error;
  800. }
  801. #define CHECK_REQUIRED_VAL_(info, val, func) \
  802. CHECK_REQUIRED_VAL(struct obs_service_info, info, val, func)
  803. CHECK_REQUIRED_VAL_(info, get_name, obs_register_service);
  804. CHECK_REQUIRED_VAL_(info, create, obs_register_service);
  805. CHECK_REQUIRED_VAL_(info, destroy, obs_register_service);
  806. CHECK_REQUIRED_VAL_(info, get_protocol, obs_register_service);
  807. #undef CHECK_REQUIRED_VAL_
  808. REGISTER_OBS_DEF(size, obs_service_info, obs->service_types, info);
  809. return;
  810. error:
  811. HANDLE_ERROR(size, obs_service_info, info);
  812. }