| /** @file |
| |
| A brief file description |
| |
| @section license License |
| |
| Licensed to the Apache Software Foundation (ASF) under one |
| or more contributor license agreements. See the NOTICE file |
| distributed with this work for additional information |
| regarding copyright ownership. The ASF licenses this file |
| to you under the Apache License, Version 2.0 (the |
| "License"); you may not use this file except in compliance |
| with the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| /* |
| * replace_header.c: |
| * an example program... |
| * |
| * NOTE: If faced with duplicate headers, this will only detect the |
| * first instance. Operational plugins may need to do more! |
| * |
| * Usage: |
| * |
| * |
| */ |
| #include <stdio.h> |
| #include <string.h> |
| #include <ctype.h> |
| |
| #include "ts/ts.h" |
| #include "tscore/ink_defs.h" |
| |
| #define PLUGIN_NAME "replace_header" |
| |
| static void |
| replace_header(TSHttpTxn txnp) |
| { |
| TSMBuffer resp_bufp; |
| TSMLoc resp_loc; |
| TSMLoc field_loc; |
| |
| if (TSHttpTxnServerRespGet(txnp, &resp_bufp, &resp_loc) != TS_SUCCESS) { |
| TSError("[%s] Couldn't retrieve server response header", PLUGIN_NAME); |
| goto done; |
| } |
| |
| field_loc = TSMimeHdrFieldFind(resp_bufp, resp_loc, TS_MIME_FIELD_ACCEPT_RANGES, TS_MIME_LEN_ACCEPT_RANGES); |
| if (field_loc == TS_NULL_MLOC) { |
| /* field was not found */ |
| |
| /* create a new field in the header */ |
| TSMimeHdrFieldCreate(resp_bufp, resp_loc, &field_loc); /* Probably should check for errors. */ |
| /* set its name */ |
| TSMimeHdrFieldNameSet(resp_bufp, resp_loc, field_loc, TS_MIME_FIELD_ACCEPT_RANGES, TS_MIME_LEN_ACCEPT_RANGES); |
| /* set its value */ |
| TSMimeHdrFieldValueAppend(resp_bufp, resp_loc, field_loc, -1, "none", 4); |
| /* insert it into the header */ |
| TSMimeHdrFieldAppend(resp_bufp, resp_loc, field_loc); |
| TSHandleMLocRelease(resp_bufp, resp_loc, field_loc); |
| TSHandleMLocRelease(resp_bufp, TS_NULL_MLOC, resp_loc); |
| } else { |
| /* clear the field */ |
| TSMimeHdrFieldValuesClear(resp_bufp, resp_loc, field_loc); |
| /* set the value to "none" */ |
| TSMimeHdrFieldValueStringInsert(resp_bufp, resp_loc, field_loc, -1, "none", 4); |
| TSHandleMLocRelease(resp_bufp, resp_loc, field_loc); |
| TSHandleMLocRelease(resp_bufp, TS_NULL_MLOC, resp_loc); |
| } |
| |
| done: |
| TSHttpTxnReenable(txnp, TS_EVENT_HTTP_CONTINUE); |
| } |
| |
| static int |
| replace_header_plugin(TSCont contp ATS_UNUSED, TSEvent event, void *edata) |
| { |
| TSHttpTxn txnp = (TSHttpTxn)edata; |
| |
| switch (event) { |
| case TS_EVENT_HTTP_READ_RESPONSE_HDR: |
| replace_header(txnp); |
| return 0; |
| default: |
| break; |
| } |
| return 0; |
| } |
| |
| void |
| TSPluginInit(int argc ATS_UNUSED, const char *argv[] ATS_UNUSED) |
| { |
| TSPluginRegistrationInfo info; |
| |
| info.plugin_name = PLUGIN_NAME; |
| info.vendor_name = "Apache Software Foundation"; |
| info.support_email = "dev@trafficserver.apache.org"; |
| |
| if (TSPluginRegister(&info) != TS_SUCCESS) { |
| TSError("[%s] Plugin registration failed", PLUGIN_NAME); |
| } |
| |
| TSHttpHookAdd(TS_HTTP_READ_RESPONSE_HDR_HOOK, TSContCreate(replace_header_plugin, NULL)); |
| } |