123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- /*
- * Copyright (C) 2010 Martin Willi
- * Copyright (C) 2010 revosec AG
- *
- * This program is free software; you can redistribute it and/or modify it
- * under the terms of the GNU General Public License as published by the
- * Free Software Foundation; either version 2 of the License, or (at your
- * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
- *
- * This program is distributed in the hope that it will be useful, but
- * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
- * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
- * for more details.
- */
- #include "hook.h"
- #include <encoding/payloads/unknown_payload.h>
- typedef struct private_set_ike_initiator_t private_set_ike_initiator_t;
- /**
- * Private data of an set_ike_initiator_t object.
- */
- struct private_set_ike_initiator_t {
- /**
- * Implements the hook_t interface.
- */
- hook_t hook;
- /**
- * Alter requests or responses?
- */
- bool req;
- /**
- * ID of message to alter.
- */
- int id;
- };
- METHOD(listener_t, message, bool,
- private_set_ike_initiator_t *this, ike_sa_t *ike_sa, message_t *message,
- bool incoming, bool plain)
- {
- if (!incoming && plain &&
- message->get_request(message) == this->req &&
- message->get_message_id(message) == this->id)
- {
- ike_sa_id_t *id;
- DBG1(DBG_CFG, "toggling IKE message initiator flag");
- id = message->get_ike_sa_id(message);
- id->switch_initiator(id);
- }
- return TRUE;
- }
- METHOD(hook_t, destroy, void,
- private_set_ike_initiator_t *this)
- {
- free(this);
- }
- /**
- * Create the IKE_AUTH fill hook
- */
- hook_t *set_ike_initiator_hook_create(char *name)
- {
- private_set_ike_initiator_t *this;
- INIT(this,
- .hook = {
- .listener = {
- .message = _message,
- },
- .destroy = _destroy,
- },
- .req = conftest->test->get_bool(conftest->test,
- "hooks.%s.request", TRUE, name),
- .id = conftest->test->get_int(conftest->test,
- "hooks.%s.id", 0, name),
- );
- return &this->hook;
- }
|