From 5d71270d218af104496b504530f954c1988baa96 Mon Sep 17 00:00:00 2001 From: phk Date: Sun, 18 Jan 2009 15:42:29 +0000 Subject: [PATCH] Implement exact matching and negatives for bans. You can now purge using these four conditionals: purge req.url == "/diediedie.html" purge req.url != "/index.html" purge req.url !~ "\.html$" purge req.url ~ "\.jpg$" git-svn-id: svn+ssh://projects.linpro.no/svn/varnish/trunk@3518 d4fa192b-c00b-0410-8231-f00ffab90ce4 --- varnish-cache/bin/varnishd/cache_ban.c | 90 ++++++++++++++----- .../bin/varnishtest/tests/c00021.vtc | 8 ++ 2 files changed, 76 insertions(+), 22 deletions(-) diff --git a/varnish-cache/bin/varnishd/cache_ban.c b/varnish-cache/bin/varnishd/cache_ban.c index 0a0b33ba..72ae21bc 100644 --- a/varnish-cache/bin/varnishd/cache_ban.c +++ b/varnish-cache/bin/varnishd/cache_ban.c @@ -70,7 +70,11 @@ struct ban_test { int cost; char *test; ban_cond_f *func; + int flags; +#define BAN_T_REGEXP (1 << 0) +#define BAN_T_NOT (1 << 1) regex_t re; + char *dst; }; struct ban { @@ -145,7 +149,10 @@ ban_free_ban(struct ban *b) bt = VTAILQ_FIRST(&b->tests); VTAILQ_REMOVE(&b->tests, bt, list); free(bt->test); - regfree(&bt->re); + if (bt->flags & BAN_T_REGEXP) + regfree(&bt->re); + if (bt->dst != NULL) + free(bt->dst); FREE_OBJ(bt); } FREE_OBJ(b); @@ -188,33 +195,67 @@ ban_compare(const struct ban *b1, const struct ban *b2) */ static int -ban_cond_url_regexp(const struct ban_test *bt, const struct object *o, +ban_cond_str(const struct ban_test *bt, const char *p) +{ + int i; + + if (bt->flags & BAN_T_REGEXP) + i = regexec(&bt->re, p, 0, NULL, 0); + else + i = strcmp(bt->dst, p); + if (bt->flags & BAN_T_NOT) + return (i); + return (!i); +} + +static int +ban_cond_url(const struct ban_test *bt, const struct object *o, const struct sess *sp) { (void)o; - return (!regexec(&bt->re, sp->http->hd[HTTP_HDR_URL].b, 0, NULL, 0)); + + CHECK_OBJ_NOTNULL(sp, SESS_MAGIC); + return(ban_cond_str(bt, sp->http->hd[HTTP_HDR_URL].b)); } static int -ban_cond_hash_regexp(const struct ban_test *bt, const struct object *o, +ban_cond_hash(const struct ban_test *bt, const struct object *o, const struct sess *sp) { (void)sp; CHECK_OBJ_NOTNULL(o, OBJECT_MAGIC); CHECK_OBJ_NOTNULL(o->objhead, OBJHEAD_MAGIC); AN(o->objhead->hash); - return (!regexec(&bt->re, o->objhead->hash, 0, NULL, 0)); + return(ban_cond_str(bt, o->objhead->hash)); } /*-------------------------------------------------------------------- * Parse and add a ban test specification */ +static int +ban_parse_regexp(struct cli *cli, struct ban_test *bt, const char *a3) +{ + int i; + char buf[512]; + + i = regcomp(&bt->re, a3, REG_EXTENDED | REG_ICASE | REG_NOSUB); + if (i) { + (void)regerror(i, &bt->re, buf, sizeof buf); + regfree(&bt->re); + VSL(SLT_Debug, 0, "REGEX: <%s>", buf); + cli_out(cli, "%s", buf); + cli_result(cli, CLIS_PARAM); + return (-1); + } + bt->flags |= BAN_T_REGEXP; + return (0); +} + static int ban_parse_test(struct cli *cli, struct ban *b, const char *a1, const char *a2, const char *a3) { struct ban_test *bt; - char buf[512]; struct vsb *sb; int i; @@ -226,27 +267,32 @@ ban_parse_test(struct cli *cli, struct ban *b, const char *a1, const char *a2, c return (-1); } - if (strcmp(a2, "~")) { - /* XXX: Add more conditionals */ - cli_out(cli, "expected \"~\" got \"%s\"", a2); - cli_result(cli, CLIS_PARAM); - return (-1); - } - - i = regcomp(&bt->re, a3, REG_EXTENDED | REG_ICASE | REG_NOSUB); - if (i) { - (void)regerror(i, &bt->re, buf, sizeof buf); - regfree(&bt->re); - VSL(SLT_Debug, 0, "REGEX: <%s>", buf); - cli_out(cli, "%s", buf); + if (!strcmp(a2, "~")) { + i = ban_parse_regexp(cli, bt, a3); + if (i) + return (i); + } else if (!strcmp(a2, "!~")) { + bt->flags |= BAN_T_NOT; + i = ban_parse_regexp(cli, bt, a3); + if (i) + return (i); + } else if (!strcmp(a2, "==")) { + bt->dst = strdup(a3); + XXXAN(bt->dst); + } else if (!strcmp(a2, "!=")) { + bt->flags |= BAN_T_NOT; + bt->dst = strdup(a3); + XXXAN(bt->dst); + } else { + cli_out(cli, + "expected conditional (~, !~, == or !=) got \"%s\"", a2); cli_result(cli, CLIS_PARAM); return (-1); } - if (!strcmp(a1, "req.url")) - bt->func = ban_cond_url_regexp; + bt->func = ban_cond_url; else if (!strcmp(a1, "obj.hash")) - bt->func = ban_cond_hash_regexp; + bt->func = ban_cond_hash; else { cli_out(cli, "unknown or unsupported field \"%s\"", a1); cli_result(cli, CLIS_PARAM); diff --git a/varnish-cache/bin/varnishtest/tests/c00021.vtc b/varnish-cache/bin/varnishtest/tests/c00021.vtc index e4e305d0..680755e7 100644 --- a/varnish-cache/bin/varnishtest/tests/c00021.vtc +++ b/varnish-cache/bin/varnishtest/tests/c00021.vtc @@ -28,6 +28,14 @@ varnish v1 -clierr 104 "purge foo bar" varnish v1 -clierr 106 "purge a b c && a" varnish v1 -clierr 106 "purge a b c && a b" varnish v1 -clierr 106 "purge a b c || a b c" +varnish v1 -cliok "purge req.url == foo" +client c1 { + txreq -url "/foo" + rxresp + expect resp.status == 200 + expect resp.http.content-length == 5 +} + varnish v1 -cliok "purge req.url ~ foo && req.url ~ \"[ o]\"" varnish v1 -cliok "purge.list" -- 2.39.5