summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/extractors/extractor.c3
-rw-r--r--src/process_url.c2
2 files changed, 2 insertions, 3 deletions
diff --git a/src/extractors/extractor.c b/src/extractors/extractor.c
index ccc1ec6..5317945 100644
--- a/src/extractors/extractor.c
+++ b/src/extractors/extractor.c
@@ -6,8 +6,7 @@
#include "haokan.h"
Site_map site_map = {
- {{"www.bilibili.com", SITE_BILIBILI}, {"haokan.baidu.com", SITE_HAOKAN}},
- 2};
+ {{"bilibili.com", SITE_BILIBILI}, {"haokan.baidu.com", SITE_HAOKAN}}, 2};
void options_cleanup(Options *options) {
free_and_nullify(options->URL);
diff --git a/src/process_url.c b/src/process_url.c
index a1204a1..35abf46 100644
--- a/src/process_url.c
+++ b/src/process_url.c
@@ -171,7 +171,7 @@ static int parse_url(const char *URL, const char *outdir, char *fn) {
DEBUG_PRINT("Query: %s\n", query);
for (unsigned short i = 0; i < site_map.size; i++) {
- if (!strcmp(domain, site_map.pairs[i].domain)) {
+ if (strstr(domain, site_map.pairs[i].domain)) {
append_log("Got site: %s\n", domain);
thrd_t t;
options.site = site_map.pairs[i].site;