Add forgotten VRT_RemoveVFP call

420d0edd · Nils Goroll · acacd065 · 420d0edd
Commit 420d0edd authored Apr 17, 2019 by Nils Goroll
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 3 deletions

vmod_etag.c src/vmod_etag.c +10 -3

No files found.
--- a/src/vmod_etag.c
+++ b/src/vmod_etag.c
@@ -9,9 +9,16 @@
 int
 vmod_event_function(VRT_CTX, struct vmod_priv *priv, enum vcl_event_e e)
 {
-	if (e != VCL_EVENT_LOAD)
-		return (0);
+	switch (e) {
+	case VCL_EVENT_LOAD:
+		VRT_AddVFP(ctx, &VFP_etag);
+		break;
+	case VCL_EVENT_DISCARD:
+		VRT_RemoveVFP(ctx, &VFP_etag);
+		break;
+	default:
+		break;
+	}

-	VRT_AddVFP(ctx, &VFP_etag);
 	return (0);
 }