Added add_segment() method to IETF attributes

[strongswan.git] / src / libimcv / plugins / imc_test / imc_test_state.h

diff --git a/src/libimcv/plugins/imc_test/imc_test_state.h b/src/libimcv/plugins/imc_test/imc_test_state.h
index 384285a..5f9ee25 100644 (file)
--- a/src/libimcv/plugins/imc_test/imc_test_state.h
+++ b/src/libimcv/plugins/imc_test/imc_test_state.h
@@ -13,14 +13,17 @@
  */
 
 /**
+ * @defgroup imc_test imc_test
+ * @ingroup libimcv_plugins
  *
  * @defgroup imc_test_state_t imc_test_state
- * @{ @ingroup imc_test_state
+ * @{ @ingroup imc_test
  */
 
 #ifndef IMC_TEST_STATE_H_
 #define IMC_TEST_STATE_H_
 
+#include <tncifimc.h>
 #include <imc/imc_state.h>
 #include <library.h>
 
@@ -51,6 +54,13 @@ struct imc_test_state_t {
        void (*set_command)(imc_test_state_t *this, char *command);
 
        /**
+        * get the value size of a dummy attribute to send to IMV
+        *
+        * @return                              size of the dummy attribute value to send to IMV
+        */
+       int (*get_dummy_size)(imc_test_state_t *this);
+
+       /**
         * Test and reset the first handshake flag
         *
         * @return                              TRUE if first handshake
@@ -63,16 +73,18 @@ struct imc_test_state_t {
         * @return                              TRUE if a handshake retry should be done
         */
        bool (*do_handshake_retry)(imc_test_state_t *this);
+
 };
 
 /**
  * Create an imc_test_state_t instance
  *
- * @param id           connection ID
- * @param command      command to send to IMV
- * @param retry                TRUE if a handshake retry should be done
+ * @param id                   connection ID
+ * @param command              command to send to IMV
+ * @param dummy_size   size of the dummy attribute to send (only if > 0)
+ * @param retry                        TRUE if a handshake retry should be done
  */
 imc_state_t* imc_test_state_create(TNC_ConnectionID id, char* command,
-                                                                  bool retry);
+                                                                  int dummy_size, bool retry);
 
 #endif /** IMC_TEST_STATE_H_ @}*/