implemented handshake retry on the client side
[strongswan.git] / src / libimcv / plugins / imc_test / imc_test_state.c
1 /*
2 * Copyright (C) 2011 Andreas Steffen, HSR Hochschule fuer Technik Rapperswil
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License as published by the
6 * Free Software Foundation; either version 2 of the License, or (at your
7 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * for more details.
13 */
14
15 #include "imc_test_state.h"
16
17 #include <debug.h>
18
19 typedef struct private_imc_test_state_t private_imc_test_state_t;
20
21 /**
22 * Private data of an imc_test_state_t object.
23 */
24 struct private_imc_test_state_t {
25
26 /**
27 * Public members of imc_test_state_t
28 */
29 imc_test_state_t public;
30
31 /**
32 * TNCCS connection ID
33 */
34 TNC_ConnectionID connection_id;
35
36 /**
37 * TNCCS connection state
38 */
39 TNC_ConnectionState state;
40
41 /**
42 * Command to transmit to IMV
43 */
44 char *command;
45
46 /**
47 * Do a handshake retry
48 */
49 bool handshake_retry;
50 };
51
52 METHOD(imc_state_t, get_connection_id, TNC_ConnectionID,
53 private_imc_test_state_t *this)
54 {
55 return this->connection_id;
56 }
57
58 METHOD(imc_state_t, change_state, void,
59 private_imc_test_state_t *this, TNC_ConnectionState new_state)
60 {
61 this->state = new_state;
62 }
63
64 METHOD(imc_state_t, destroy, void,
65 private_imc_test_state_t *this)
66 {
67 free(this->command);
68 free(this);
69 }
70
71 METHOD(imc_test_state_t, get_command, char*,
72 private_imc_test_state_t *this)
73 {
74 return this->command;
75 }
76
77 METHOD(imc_test_state_t, set_command, void,
78 private_imc_test_state_t *this, char* command)
79 {
80 char *old_command;
81
82 old_command = this->command;
83 this->command = strdup(command);
84 free(old_command);
85 }
86
87 METHOD(imc_test_state_t, do_handshake_retry, bool,
88 private_imc_test_state_t *this)
89 {
90 bool retry;
91
92 /* test and reset handshake_retry flag */
93 retry = this->handshake_retry;
94 this->handshake_retry = FALSE;
95 return retry;
96 }
97
98
99 /**
100 * Described in header.
101 */
102 imc_state_t *imc_test_state_create(TNC_ConnectionID connection_id,
103 char *command, bool retry)
104 {
105 private_imc_test_state_t *this;
106
107 INIT(this,
108 .public = {
109 .interface = {
110 .get_connection_id = _get_connection_id,
111 .change_state = _change_state,
112 .destroy = _destroy,
113 },
114 .get_command = _get_command,
115 .set_command = _set_command,
116 .do_handshake_retry = _do_handshake_retry,
117 },
118 .state = TNC_CONNECTION_STATE_CREATE,
119 .connection_id = connection_id,
120 .command = strdup(command),
121 .handshake_retry = retry,
122 );
123
124 return &this->public.interface;
125 }
126
127