walkero
/
libpsl
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

Add --no-star-rule to psl utility

This commit is contained in:
Tim Rühsen 2018-02-26 11:23:01 +01:00
parent 35d96246e7
commit 337e189979
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tools/psl.c
View File

@ -47,6 +47,8 @@ static void usage(int err, FILE* f)
fprintf(f, " --version show library version information\n"); fprintf(f, " --version show library version information\n");
fprintf(f, " --use-latest-data use the latest PSL data available [default]\n"); fprintf(f, " --use-latest-data use the latest PSL data available [default]\n");
fprintf(f, " --use-builtin-data use the builtin PSL data\n"); fprintf(f, " --use-builtin-data use the builtin PSL data\n");
fprintf(f, " --no-star-rule do not apply the prevailing star rule\n");
fprintf(f, " (only applies to --is-public-suffix)\n");
fprintf(f, " --load-psl-file <filename> load PSL data from file\n"); fprintf(f, " --load-psl-file <filename> load PSL data from file\n");
fprintf(f, " --is-public-suffix check if domains are public suffixes [default]\n"); fprintf(f, " --is-public-suffix check if domains are public suffixes [default]\n");
fprintf(f, " --is-cookie-domain-acceptable <cookie-domain>\n"); fprintf(f, " --is-cookie-domain-acceptable <cookie-domain>\n");
@ -71,7 +73,7 @@ static const char *time2str(time_t t)
int main(int argc, const char *const *argv) int main(int argc, const char *const *argv)
{ {
int mode = 1; int mode = 1, no_star_rule = 0;
const char *const *arg, *psl_file = NULL, *cookie_domain = NULL; const char *const *arg, *psl_file = NULL, *cookie_domain = NULL;
psl_ctx_t *psl = (psl_ctx_t *) psl_latest(NULL); psl_ctx_t *psl = (psl_ctx_t *) psl_latest(NULL);
@ -111,6 +113,9 @@ int main(int argc, const char *const *argv)
if (!(psl = (psl_ctx_t *) psl_builtin())) if (!(psl = (psl_ctx_t *) psl_builtin()))
printf("No builtin PSL data available\n"); printf("No builtin PSL data available\n");
} }
else if (!strcmp(*arg, "--no-star-rule")) {
no_star_rule = 1;
}
else if (!strcmp(*arg, "--load-psl-file") && arg < argv + argc - 1) { else if (!strcmp(*arg, "--load-psl-file") && arg < argv + argc - 1) {
psl_free(psl); psl_free(psl);
if (psl_file) { if (psl_file) {
@ -147,6 +152,10 @@ int main(int argc, const char *const *argv)
} }
if (mode != 99) { if (mode != 99) {
if (mode != 1 && no_star_rule) {
fprintf(stderr, "--no-star-rule only combines with --is-public-suffix\n");
usage(1, stderr);
}
if (!psl) { if (!psl) {
fprintf(stderr, "No PSL data available - aborting\n"); fprintf(stderr, "No PSL data available - aborting\n");
exit(2); exit(2);
@ -165,8 +174,12 @@ int main(int argc, const char *const *argv)
if ((rc = psl_str_to_utf8lower(domain, NULL, NULL, &lower)) != PSL_SUCCESS) if ((rc = psl_str_to_utf8lower(domain, NULL, NULL, &lower)) != PSL_SUCCESS)
fprintf(stderr, "%s: Failed to convert to lowercase UTF-8 (%d)\n", domain, rc); fprintf(stderr, "%s: Failed to convert to lowercase UTF-8 (%d)\n", domain, rc);
else if (mode == 1) else if (mode == 1) {
if (no_star_rule)
printf("%s: %d (%s)\n", domain, psl_is_public_suffix2(psl, lower, PSL_TYPE_ANY|PSL_TYPE_NO_STAR_RULE), lower);
else
printf("%s: %d (%s)\n", domain, psl_is_public_suffix(psl, lower), lower); printf("%s: %d (%s)\n", domain, psl_is_public_suffix(psl, lower), lower);
}
else if (mode == 2) else if (mode == 2)
printf("%s: %s\n", domain, psl_unregistrable_domain(psl, lower)); printf("%s: %s\n", domain, psl_unregistrable_domain(psl, lower));
else if (mode == 3) else if (mode == 3)
@ -191,9 +204,13 @@ int main(int argc, const char *const *argv)
} }
if (mode == 1) { if (mode == 1) {
for (; arg < argv + argc; arg++) for (; arg < argv + argc; arg++) {
if (no_star_rule)
printf("%s: %d\n", *arg, psl_is_public_suffix2(psl, *arg, PSL_TYPE_ANY|PSL_TYPE_NO_STAR_RULE));
else
printf("%s: %d\n", *arg, psl_is_public_suffix(psl, *arg)); printf("%s: %d\n", *arg, psl_is_public_suffix(psl, *arg));
} }
}
else if (mode == 2) { else if (mode == 2) {
for (; arg < argv + argc; arg++) for (; arg < argv + argc; arg++)
printf("%s: %s\n", *arg, psl_unregistrable_domain(psl, *arg)); printf("%s: %s\n", *arg, psl_unregistrable_domain(psl, *arg));