diff options
Diffstat (limited to 'src/extractors')
| -rw-r--r-- | src/extractors/haokan.c | 59 |
1 files changed, 30 insertions, 29 deletions
diff --git a/src/extractors/haokan.c b/src/extractors/haokan.c index d2d0379..204e539 100644 --- a/src/extractors/haokan.c +++ b/src/extractors/haokan.c @@ -8,39 +8,40 @@ void haokan_extract(Options *options) { char *resp, *title, *videoURL; - get(options->URL, &resp); - const char *patterns_str[2] = {"<div class='ssr-video-title\'>(.*?)</div>", - "\"playurl\":\"(http.+?)\""}; - const str_array_t patterns = {(char **)patterns_str, 2}; - str_array_t results = create_str_array(0); - int r = regex_match(resp, patterns, &results); - // Should match exactly two results in HTML, otherwise error out. - if (!r && results.n == 2) { - // for (unsigned short i = 0; i < results.n; i++) { - // DEBUG_PRINT("%s\n", results.str[i]); - // } - title = results.str[0]; - substitute_str(results.str[1], "\\\\/", "/", &videoURL); - DEBUG_PRINT("title: %s\n", title); - DEBUG_PRINT("videoURL: %s\n", videoURL); + if (!get(options->URL, &resp)) { + const char *patterns_str[2] = {"<div class='ssr-video-title\'>(.*?)</div>", + "\"playurl\":\"(http.+?)\""}; + const str_array_t patterns = {(char **)patterns_str, 2}; + str_array_t results = create_str_array(0); + int r = regex_match(resp, patterns, &results); + // Should match exactly two results in HTML, otherwise error out. + if (!r && results.n == 2) { + // for (unsigned short i = 0; i < results.n; i++) { + // DEBUG_PRINT("%s\n", results.str[i]); + // } + title = results.str[0]; + substitute_str(results.str[1], "\\\\/", "/", &videoURL); + DEBUG_PRINT("title: %s\n", title); + DEBUG_PRINT("videoURL: %s\n", videoURL); - char *ct = NULL; - get_info(videoURL, NULL, &ct); - if (ct == NULL) { - goto end; - } - const char *ext = mimeType2ext(ct); - char *filename = malloc(strlen(title) + strlen(ct) + 2); - sprintf(filename, "%s.%s", title, ext); + char *ct = NULL; + get_info(videoURL, NULL, &ct); + if (ct == NULL) { + goto end; + } + const char *ext = mimeType2ext(ct); + char *filename = malloc(strlen(title) + strlen(ct) + 2); + sprintf(filename, "%s.%s", title, ext); - add_url(videoURL, NULL, filename, "https://haokan.baidu.com", NULL, NULL); + add_url(videoURL, NULL, filename, "https://haokan.baidu.com", NULL, NULL); - end: + end: + free_str_array(&results); + free_and_nullify(filename); + free_and_nullify(videoURL); + return; + } free_str_array(&results); - free_and_nullify(filename); - free_and_nullify(videoURL); - return; } LOG("haokan", "Download failed.\n"); - free_str_array(&results); } |