1
0

obs-module.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950
  1. /******************************************************************************
  2. Copyright (C) 2013-2014 by Hugh 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. extern void get_plugin_info(const char *path, bool *is_obs_plugin,
  221. bool *can_load);
  222. struct fail_info {
  223. struct dstr fail_modules;
  224. size_t fail_count;
  225. };
  226. static void load_all_callback(void *param, const struct obs_module_info2 *info)
  227. {
  228. struct fail_info *fail_info = param;
  229. obs_module_t *module;
  230. bool is_obs_plugin;
  231. bool can_load_obs_plugin;
  232. get_plugin_info(info->bin_path, &is_obs_plugin, &can_load_obs_plugin);
  233. if (!is_obs_plugin) {
  234. blog(LOG_WARNING, "Skipping module '%s', not an OBS plugin",
  235. info->bin_path);
  236. return;
  237. }
  238. if (!can_load_obs_plugin) {
  239. blog(LOG_WARNING,
  240. "Skipping module '%s' due to possible "
  241. "import conflicts",
  242. info->bin_path);
  243. goto load_failure;
  244. }
  245. int code = obs_open_module(&module, info->bin_path, info->data_path);
  246. switch (code) {
  247. case MODULE_MISSING_EXPORTS:
  248. blog(LOG_DEBUG,
  249. "Failed to load module file '%s', not an OBS plugin",
  250. info->bin_path);
  251. return;
  252. case MODULE_FILE_NOT_FOUND:
  253. blog(LOG_DEBUG,
  254. "Failed to load module file '%s', file not found",
  255. info->bin_path);
  256. return;
  257. case MODULE_ERROR:
  258. blog(LOG_DEBUG, "Failed to load module file '%s'",
  259. info->bin_path);
  260. goto load_failure;
  261. case MODULE_INCOMPATIBLE_VER:
  262. blog(LOG_DEBUG,
  263. "Failed to load module file '%s', incompatible version",
  264. info->bin_path);
  265. goto load_failure;
  266. case MODULE_HARDCODED_SKIP:
  267. return;
  268. }
  269. if (!obs_init_module(module))
  270. free_module(module);
  271. UNUSED_PARAMETER(param);
  272. return;
  273. load_failure:
  274. if (fail_info) {
  275. dstr_cat(&fail_info->fail_modules, info->name);
  276. dstr_cat(&fail_info->fail_modules, ";");
  277. fail_info->fail_count++;
  278. }
  279. }
  280. static const char *obs_load_all_modules_name = "obs_load_all_modules";
  281. #ifdef _WIN32
  282. static const char *reset_win32_symbol_paths_name = "reset_win32_symbol_paths";
  283. #endif
  284. void obs_load_all_modules(void)
  285. {
  286. profile_start(obs_load_all_modules_name);
  287. obs_find_modules2(load_all_callback, NULL);
  288. #ifdef _WIN32
  289. profile_start(reset_win32_symbol_paths_name);
  290. reset_win32_symbol_paths();
  291. profile_end(reset_win32_symbol_paths_name);
  292. #endif
  293. profile_end(obs_load_all_modules_name);
  294. }
  295. static const char *obs_load_all_modules2_name = "obs_load_all_modules2";
  296. void obs_load_all_modules2(struct obs_module_failure_info *mfi)
  297. {
  298. struct fail_info fail_info = {0};
  299. memset(mfi, 0, sizeof(*mfi));
  300. profile_start(obs_load_all_modules2_name);
  301. obs_find_modules2(load_all_callback, &fail_info);
  302. #ifdef _WIN32
  303. profile_start(reset_win32_symbol_paths_name);
  304. reset_win32_symbol_paths();
  305. profile_end(reset_win32_symbol_paths_name);
  306. #endif
  307. profile_end(obs_load_all_modules2_name);
  308. mfi->count = fail_info.fail_count;
  309. mfi->failed_modules =
  310. strlist_split(fail_info.fail_modules.array, ';', false);
  311. dstr_free(&fail_info.fail_modules);
  312. }
  313. void obs_module_failure_info_free(struct obs_module_failure_info *mfi)
  314. {
  315. if (mfi->failed_modules) {
  316. bfree(mfi->failed_modules);
  317. mfi->failed_modules = NULL;
  318. }
  319. }
  320. void obs_post_load_modules(void)
  321. {
  322. for (obs_module_t *mod = obs->first_module; !!mod; mod = mod->next)
  323. if (mod->post_load)
  324. mod->post_load();
  325. }
  326. static inline void make_data_dir(struct dstr *parsed_data_dir,
  327. const char *data_dir, const char *name)
  328. {
  329. dstr_copy(parsed_data_dir, data_dir);
  330. dstr_replace(parsed_data_dir, "%module%", name);
  331. if (dstr_end(parsed_data_dir) == '/')
  332. dstr_resize(parsed_data_dir, parsed_data_dir->len - 1);
  333. }
  334. static char *make_data_directory(const char *module_name, const char *data_dir)
  335. {
  336. struct dstr parsed_data_dir = {0};
  337. bool found = false;
  338. make_data_dir(&parsed_data_dir, data_dir, module_name);
  339. found = os_file_exists(parsed_data_dir.array);
  340. if (!found && astrcmpi_n(module_name, "lib", 3) == 0)
  341. make_data_dir(&parsed_data_dir, data_dir, module_name + 3);
  342. return parsed_data_dir.array;
  343. }
  344. static bool parse_binary_from_directory(struct dstr *parsed_bin_path,
  345. const char *bin_path, const char *file)
  346. {
  347. struct dstr directory = {0};
  348. bool found = true;
  349. dstr_copy(&directory, bin_path);
  350. dstr_replace(&directory, "%module%", file);
  351. if (dstr_end(&directory) != '/')
  352. dstr_cat_ch(&directory, '/');
  353. dstr_copy_dstr(parsed_bin_path, &directory);
  354. dstr_cat(parsed_bin_path, file);
  355. #ifdef __APPLE__
  356. if (!os_file_exists(parsed_bin_path->array)) {
  357. dstr_cat(parsed_bin_path, ".so");
  358. }
  359. #else
  360. dstr_cat(parsed_bin_path, get_module_extension());
  361. #endif
  362. if (!os_file_exists(parsed_bin_path->array)) {
  363. /* Legacy fallback: Check for plugin with .so suffix*/
  364. dstr_cat(parsed_bin_path, ".so");
  365. /* if the file doesn't exist, check with 'lib' prefix */
  366. dstr_copy_dstr(parsed_bin_path, &directory);
  367. dstr_cat(parsed_bin_path, "lib");
  368. dstr_cat(parsed_bin_path, file);
  369. dstr_cat(parsed_bin_path, get_module_extension());
  370. /* if neither exist, don't include this as a library */
  371. if (!os_file_exists(parsed_bin_path->array)) {
  372. dstr_free(parsed_bin_path);
  373. found = false;
  374. }
  375. }
  376. dstr_free(&directory);
  377. return found;
  378. }
  379. static void process_found_module(struct obs_module_path *omp, const char *path,
  380. bool directory,
  381. obs_find_module_callback2_t callback,
  382. void *param)
  383. {
  384. struct obs_module_info2 info;
  385. struct dstr name = {0};
  386. struct dstr parsed_bin_path = {0};
  387. const char *file;
  388. char *parsed_data_dir;
  389. bool bin_found = true;
  390. file = strrchr(path, '/');
  391. file = file ? (file + 1) : path;
  392. if (strcmp(file, ".") == 0 || strcmp(file, "..") == 0)
  393. return;
  394. dstr_copy(&name, file);
  395. char *ext = strrchr(name.array, '.');
  396. if (ext)
  397. dstr_resize(&name, ext - name.array);
  398. if (!directory) {
  399. dstr_copy(&parsed_bin_path, path);
  400. } else {
  401. bin_found = parse_binary_from_directory(&parsed_bin_path,
  402. omp->bin, name.array);
  403. }
  404. parsed_data_dir = make_data_directory(name.array, omp->data);
  405. if (parsed_data_dir && bin_found) {
  406. info.bin_path = parsed_bin_path.array;
  407. info.data_path = parsed_data_dir;
  408. info.name = name.array;
  409. callback(param, &info);
  410. }
  411. bfree(parsed_data_dir);
  412. dstr_free(&name);
  413. dstr_free(&parsed_bin_path);
  414. }
  415. static void find_modules_in_path(struct obs_module_path *omp,
  416. obs_find_module_callback2_t callback,
  417. void *param)
  418. {
  419. struct dstr search_path = {0};
  420. char *module_start;
  421. bool search_directories = false;
  422. os_glob_t *gi;
  423. dstr_copy(&search_path, omp->bin);
  424. module_start = strstr(search_path.array, "%module%");
  425. if (module_start) {
  426. dstr_resize(&search_path, module_start - search_path.array);
  427. search_directories = true;
  428. }
  429. if (!dstr_is_empty(&search_path) && dstr_end(&search_path) != '/')
  430. dstr_cat_ch(&search_path, '/');
  431. dstr_cat_ch(&search_path, '*');
  432. if (!search_directories)
  433. dstr_cat(&search_path, get_module_extension());
  434. if (os_glob(search_path.array, 0, &gi) == 0) {
  435. for (size_t i = 0; i < gi->gl_pathc; i++) {
  436. if (search_directories == gi->gl_pathv[i].directory)
  437. process_found_module(omp, gi->gl_pathv[i].path,
  438. search_directories,
  439. callback, param);
  440. }
  441. os_globfree(gi);
  442. }
  443. dstr_free(&search_path);
  444. }
  445. void obs_find_modules2(obs_find_module_callback2_t callback, void *param)
  446. {
  447. if (!obs)
  448. return;
  449. for (size_t i = 0; i < obs->module_paths.num; i++) {
  450. struct obs_module_path *omp = obs->module_paths.array + i;
  451. find_modules_in_path(omp, callback, param);
  452. }
  453. }
  454. void obs_find_modules(obs_find_module_callback_t callback, void *param)
  455. {
  456. /* the structure is ABI compatible so we can just cast the callback */
  457. obs_find_modules2((obs_find_module_callback2_t)callback, param);
  458. }
  459. void obs_enum_modules(obs_enum_module_callback_t callback, void *param)
  460. {
  461. struct obs_module *module;
  462. if (!obs)
  463. return;
  464. module = obs->first_module;
  465. while (module) {
  466. callback(param, module);
  467. module = module->next;
  468. }
  469. }
  470. void free_module(struct obs_module *mod)
  471. {
  472. if (!mod)
  473. return;
  474. if (mod->module) {
  475. if (mod->free_locale)
  476. mod->free_locale();
  477. if (mod->loaded && mod->unload)
  478. mod->unload();
  479. /* there is no real reason to close the dynamic libraries,
  480. * and sometimes this can cause issues. */
  481. /* os_dlclose(mod->module); */
  482. }
  483. for (obs_module_t *m = obs->first_module; !!m; m = m->next) {
  484. if (m->next == mod) {
  485. m->next = mod->next;
  486. break;
  487. }
  488. }
  489. if (obs->first_module == mod)
  490. obs->first_module = mod->next;
  491. bfree(mod->mod_name);
  492. bfree(mod->bin_path);
  493. bfree(mod->data_path);
  494. bfree(mod);
  495. }
  496. lookup_t *obs_module_load_locale(obs_module_t *module,
  497. const char *default_locale, const char *locale)
  498. {
  499. struct dstr str = {0};
  500. lookup_t *lookup = NULL;
  501. if (!module || !default_locale || !locale) {
  502. blog(LOG_WARNING, "obs_module_load_locale: Invalid parameters");
  503. return NULL;
  504. }
  505. dstr_copy(&str, "locale/");
  506. dstr_cat(&str, default_locale);
  507. dstr_cat(&str, ".ini");
  508. char *file = obs_find_module_file(module, str.array);
  509. if (file)
  510. lookup = text_lookup_create(file);
  511. bfree(file);
  512. if (!lookup) {
  513. blog(LOG_WARNING, "Failed to load '%s' text for module: '%s'",
  514. default_locale, module->file);
  515. goto cleanup;
  516. }
  517. if (astrcmpi(locale, default_locale) == 0)
  518. goto cleanup;
  519. dstr_copy(&str, "/locale/");
  520. dstr_cat(&str, locale);
  521. dstr_cat(&str, ".ini");
  522. file = obs_find_module_file(module, str.array);
  523. if (!text_lookup_add(lookup, file))
  524. blog(LOG_WARNING, "Failed to load '%s' text for module: '%s'",
  525. locale, module->file);
  526. bfree(file);
  527. cleanup:
  528. dstr_free(&str);
  529. return lookup;
  530. }
  531. #define REGISTER_OBS_DEF(size_var, structure, dest, info) \
  532. do { \
  533. struct structure data = {0}; \
  534. if (!size_var) { \
  535. blog(LOG_ERROR, "Tried to register " #structure \
  536. " outside of obs_module_load"); \
  537. return; \
  538. } \
  539. \
  540. if (size_var > sizeof(data)) { \
  541. blog(LOG_ERROR, \
  542. "Tried to register " #structure \
  543. " with size %llu which is more " \
  544. "than libobs currently supports " \
  545. "(%llu)", \
  546. (long long unsigned)size_var, \
  547. (long long unsigned)sizeof(data)); \
  548. goto error; \
  549. } \
  550. \
  551. memcpy(&data, info, size_var); \
  552. da_push_back(dest, &data); \
  553. } while (false)
  554. #define CHECK_REQUIRED_VAL(type, info, val, func) \
  555. do { \
  556. if ((offsetof(type, val) + sizeof(info->val) > size) || \
  557. !info->val) { \
  558. blog(LOG_ERROR, \
  559. "Required value '" #val "' for " \
  560. "'%s' not found. " #func " failed.", \
  561. info->id); \
  562. goto error; \
  563. } \
  564. } while (false)
  565. #define HANDLE_ERROR(size_var, structure, info) \
  566. do { \
  567. struct structure data = {0}; \
  568. if (!size_var) \
  569. return; \
  570. \
  571. memcpy(&data, info, \
  572. sizeof(data) < size_var ? sizeof(data) : size_var); \
  573. \
  574. if (data.type_data && data.free_type_data) \
  575. data.free_type_data(data.type_data); \
  576. } while (false)
  577. #define source_warn(format, ...) \
  578. blog(LOG_WARNING, "obs_register_source: " format, ##__VA_ARGS__)
  579. #define output_warn(format, ...) \
  580. blog(LOG_WARNING, "obs_register_output: " format, ##__VA_ARGS__)
  581. #define encoder_warn(format, ...) \
  582. blog(LOG_WARNING, "obs_register_encoder: " format, ##__VA_ARGS__)
  583. #define service_warn(format, ...) \
  584. blog(LOG_WARNING, "obs_register_service: " format, ##__VA_ARGS__)
  585. void obs_register_source_s(const struct obs_source_info *info, size_t size)
  586. {
  587. struct obs_source_info data = {0};
  588. struct darray *array = NULL;
  589. if (info->type == OBS_SOURCE_TYPE_INPUT) {
  590. array = &obs->input_types.da;
  591. } else if (info->type == OBS_SOURCE_TYPE_FILTER) {
  592. array = &obs->filter_types.da;
  593. } else if (info->type == OBS_SOURCE_TYPE_TRANSITION) {
  594. array = &obs->transition_types.da;
  595. } else if (info->type != OBS_SOURCE_TYPE_SCENE) {
  596. source_warn("Tried to register unknown source type: %u",
  597. info->type);
  598. goto error;
  599. }
  600. if (get_source_info2(info->id, info->version)) {
  601. source_warn("Source '%s' already exists! "
  602. "Duplicate library?",
  603. info->id);
  604. goto error;
  605. }
  606. if (size > sizeof(data)) {
  607. source_warn("Tried to register obs_source_info with size "
  608. "%llu which is more than libobs currently "
  609. "supports (%llu)",
  610. (long long unsigned)size,
  611. (long long unsigned)sizeof(data));
  612. goto error;
  613. }
  614. memcpy(&data, info, size);
  615. /* mark audio-only filters as an async filter categorically */
  616. if (data.type == OBS_SOURCE_TYPE_FILTER) {
  617. if ((data.output_flags & OBS_SOURCE_VIDEO) == 0)
  618. data.output_flags |= OBS_SOURCE_ASYNC;
  619. }
  620. if (data.type == OBS_SOURCE_TYPE_TRANSITION) {
  621. if (data.get_width)
  622. source_warn("get_width ignored registering "
  623. "transition '%s'",
  624. data.id);
  625. if (data.get_height)
  626. source_warn("get_height ignored registering "
  627. "transition '%s'",
  628. data.id);
  629. data.output_flags |= OBS_SOURCE_COMPOSITE | OBS_SOURCE_VIDEO |
  630. OBS_SOURCE_CUSTOM_DRAW;
  631. }
  632. if ((data.output_flags & OBS_SOURCE_COMPOSITE) != 0) {
  633. if ((data.output_flags & OBS_SOURCE_AUDIO) != 0) {
  634. source_warn("Source '%s': Composite sources "
  635. "cannot be audio sources",
  636. info->id);
  637. goto error;
  638. }
  639. if ((data.output_flags & OBS_SOURCE_ASYNC) != 0) {
  640. source_warn("Source '%s': Composite sources "
  641. "cannot be async sources",
  642. info->id);
  643. goto error;
  644. }
  645. }
  646. #define CHECK_REQUIRED_VAL_(info, val, func) \
  647. CHECK_REQUIRED_VAL(struct obs_source_info, info, val, func)
  648. CHECK_REQUIRED_VAL_(info, get_name, obs_register_source);
  649. if (info->type != OBS_SOURCE_TYPE_FILTER &&
  650. info->type != OBS_SOURCE_TYPE_TRANSITION &&
  651. (info->output_flags & OBS_SOURCE_VIDEO) != 0 &&
  652. (info->output_flags & OBS_SOURCE_ASYNC) == 0) {
  653. CHECK_REQUIRED_VAL_(info, get_width, obs_register_source);
  654. CHECK_REQUIRED_VAL_(info, get_height, obs_register_source);
  655. }
  656. if ((data.output_flags & OBS_SOURCE_COMPOSITE) != 0) {
  657. CHECK_REQUIRED_VAL_(info, audio_render, obs_register_source);
  658. }
  659. #undef CHECK_REQUIRED_VAL_
  660. /* version-related stuff */
  661. data.unversioned_id = data.id;
  662. if (data.version) {
  663. struct dstr versioned_id = {0};
  664. dstr_printf(&versioned_id, "%s_v%d", data.id,
  665. (int)data.version);
  666. data.id = versioned_id.array;
  667. } else {
  668. data.id = bstrdup(data.id);
  669. }
  670. if (array)
  671. darray_push_back(sizeof(struct obs_source_info), array, &data);
  672. da_push_back(obs->source_types, &data);
  673. return;
  674. error:
  675. HANDLE_ERROR(size, obs_source_info, info);
  676. }
  677. void obs_register_output_s(const struct obs_output_info *info, size_t size)
  678. {
  679. if (find_output(info->id)) {
  680. output_warn("Output id '%s' already exists! "
  681. "Duplicate library?",
  682. info->id);
  683. goto error;
  684. }
  685. #define CHECK_REQUIRED_VAL_(info, val, func) \
  686. CHECK_REQUIRED_VAL(struct obs_output_info, info, val, func)
  687. CHECK_REQUIRED_VAL_(info, get_name, obs_register_output);
  688. CHECK_REQUIRED_VAL_(info, create, obs_register_output);
  689. CHECK_REQUIRED_VAL_(info, destroy, obs_register_output);
  690. CHECK_REQUIRED_VAL_(info, start, obs_register_output);
  691. CHECK_REQUIRED_VAL_(info, stop, obs_register_output);
  692. if (info->flags & OBS_OUTPUT_ENCODED) {
  693. CHECK_REQUIRED_VAL_(info, encoded_packet, obs_register_output);
  694. } else {
  695. if (info->flags & OBS_OUTPUT_VIDEO)
  696. CHECK_REQUIRED_VAL_(info, raw_video,
  697. obs_register_output);
  698. if (info->flags & OBS_OUTPUT_AUDIO) {
  699. if (info->flags & OBS_OUTPUT_MULTI_TRACK) {
  700. CHECK_REQUIRED_VAL_(info, raw_audio2,
  701. obs_register_output);
  702. } else {
  703. CHECK_REQUIRED_VAL_(info, raw_audio,
  704. obs_register_output);
  705. }
  706. }
  707. }
  708. #undef CHECK_REQUIRED_VAL_
  709. REGISTER_OBS_DEF(size, obs_output_info, obs->output_types, info);
  710. return;
  711. error:
  712. HANDLE_ERROR(size, obs_output_info, info);
  713. }
  714. void obs_register_encoder_s(const struct obs_encoder_info *info, size_t size)
  715. {
  716. if (find_encoder(info->id)) {
  717. encoder_warn("Encoder id '%s' already exists! "
  718. "Duplicate library?",
  719. info->id);
  720. goto error;
  721. }
  722. #define CHECK_REQUIRED_VAL_(info, val, func) \
  723. CHECK_REQUIRED_VAL(struct obs_encoder_info, info, val, func)
  724. CHECK_REQUIRED_VAL_(info, get_name, obs_register_encoder);
  725. CHECK_REQUIRED_VAL_(info, create, obs_register_encoder);
  726. CHECK_REQUIRED_VAL_(info, destroy, obs_register_encoder);
  727. if ((info->caps & OBS_ENCODER_CAP_PASS_TEXTURE) != 0)
  728. CHECK_REQUIRED_VAL_(info, encode_texture, obs_register_encoder);
  729. else
  730. CHECK_REQUIRED_VAL_(info, encode, obs_register_encoder);
  731. if (info->type == OBS_ENCODER_AUDIO)
  732. CHECK_REQUIRED_VAL_(info, get_frame_size, obs_register_encoder);
  733. #undef CHECK_REQUIRED_VAL_
  734. REGISTER_OBS_DEF(size, obs_encoder_info, obs->encoder_types, info);
  735. return;
  736. error:
  737. HANDLE_ERROR(size, obs_encoder_info, info);
  738. }
  739. void obs_register_service_s(const struct obs_service_info *info, size_t size)
  740. {
  741. if (find_service(info->id)) {
  742. service_warn("Service id '%s' already exists! "
  743. "Duplicate library?",
  744. info->id);
  745. goto error;
  746. }
  747. #define CHECK_REQUIRED_VAL_(info, val, func) \
  748. CHECK_REQUIRED_VAL(struct obs_service_info, info, val, func)
  749. CHECK_REQUIRED_VAL_(info, get_name, obs_register_service);
  750. CHECK_REQUIRED_VAL_(info, create, obs_register_service);
  751. CHECK_REQUIRED_VAL_(info, destroy, obs_register_service);
  752. #undef CHECK_REQUIRED_VAL_
  753. REGISTER_OBS_DEF(size, obs_service_info, obs->service_types, info);
  754. return;
  755. error:
  756. HANDLE_ERROR(size, obs_service_info, info);
  757. }
  758. void obs_register_modal_ui_s(const struct obs_modal_ui *info, size_t size)
  759. {
  760. #define CHECK_REQUIRED_VAL_(info, val, func) \
  761. CHECK_REQUIRED_VAL(struct obs_modal_ui, info, val, func)
  762. CHECK_REQUIRED_VAL_(info, task, obs_register_modal_ui);
  763. CHECK_REQUIRED_VAL_(info, target, obs_register_modal_ui);
  764. CHECK_REQUIRED_VAL_(info, exec, obs_register_modal_ui);
  765. #undef CHECK_REQUIRED_VAL_
  766. REGISTER_OBS_DEF(size, obs_modal_ui, obs->modal_ui_callbacks, info);
  767. return;
  768. error:
  769. HANDLE_ERROR(size, obs_modal_ui, info);
  770. }
  771. void obs_register_modeless_ui_s(const struct obs_modeless_ui *info, size_t size)
  772. {
  773. #define CHECK_REQUIRED_VAL_(info, val, func) \
  774. CHECK_REQUIRED_VAL(struct obs_modeless_ui, info, val, func)
  775. CHECK_REQUIRED_VAL_(info, task, obs_register_modeless_ui);
  776. CHECK_REQUIRED_VAL_(info, target, obs_register_modeless_ui);
  777. CHECK_REQUIRED_VAL_(info, create, obs_register_modeless_ui);
  778. #undef CHECK_REQUIRED_VAL_
  779. REGISTER_OBS_DEF(size, obs_modeless_ui, obs->modeless_ui_callbacks,
  780. info);
  781. return;
  782. error:
  783. HANDLE_ERROR(size, obs_modeless_ui, info);
  784. }