X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fsptps.h;h=96edc366ab659cd4e837753e6e3ad8c841ded0ca;hb=8c10a4131f19dc27b0bc56ef48ae481d1fa03cb0;hp=a2633bd1814352ae2de12a7796925c8a4a3d1065;hpb=266afc6c63d3d02584feb24b69063f97057daac8;p=tinc

diff --git a/src/sptps.h b/src/sptps.h
index a2633bd1..96edc366 100644
--- a/src/sptps.h
+++ b/src/sptps.h
@@ -1,3 +1,6 @@
+#ifndef TINC_SPTPS_H
+#define TINC_SPTPS_H
+
 /*
     sptps.h -- Simple Peer-to-Peer Security
     Copyright (C) 2011-2014 Guus Sliepen <guus@tinc-vpn.org>
@@ -17,9 +20,6 @@
     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
-#ifndef __SPTPS_H__
-#define __SPTPS_H__
-
 #include "system.h"
 
 #include "chacha-poly1305/chacha-poly1305.h"
@@ -33,24 +33,26 @@
 #define SPTPS_ALERT 129       // Warning or error messages
 #define SPTPS_CLOSE 130       // Application closed the connection
 
-// Key exchange states
-#define SPTPS_KEX 1           // Waiting for the first Key EXchange record
-#define SPTPS_SECONDARY_KEX 2 // Ready to receive a secondary Key EXchange record
-#define SPTPS_SIG 3           // Waiting for a SIGnature record
-#define SPTPS_ACK 4           // Waiting for an ACKnowledgement record
-
 // Overhead for datagrams
 #define SPTPS_DATAGRAM_OVERHEAD 21
 
 typedef bool (*send_data_t)(void *handle, uint8_t type, const void *data, size_t len);
 typedef bool (*receive_record_t)(void *handle, uint8_t type, const void *data, uint16_t len);
 
+// Key exchange states
+typedef enum sptps_state_t {
+	SPTPS_KEX = 1,           // Waiting for the first Key EXchange record
+	SPTPS_SECONDARY_KEX = 2, // Ready to receive a secondary Key EXchange record
+	SPTPS_SIG = 3,           // Waiting for a SIGnature record
+	SPTPS_ACK = 4,           // Waiting for an ACKnowledgement record
+} sptps_state_t;
+
 typedef struct sptps {
 	bool initiator;
 	bool datagram;
-	int state;
+	sptps_state_t state;
 
-	char *inbuf;
+	uint8_t *inbuf;
 	size_t buflen;
 	uint16_t reclen;
 
@@ -60,7 +62,7 @@ typedef struct sptps {
 	uint32_t received;
 	unsigned int replaywin;
 	unsigned int farfuture;
-	char *late;
+	uint8_t *late;
 
 	bool outstate;
 	chacha_poly1305_ctx_t *outcipher;
@@ -70,10 +72,10 @@ typedef struct sptps {
 	ecdsa_t *hiskey;
 	ecdh_t *ecdh;
 
-	char *mykex;
-	char *hiskex;
-	char *key;
-	char *label;
+	uint8_t *mykex;
+	uint8_t *hiskex;
+	uint8_t *key;
+	uint8_t *label;
 	size_t labellen;
 
 	void *handle;
@@ -88,7 +90,7 @@ extern void (*sptps_log)(sptps_t *s, int s_errno, const char *format, va_list ap
 extern bool sptps_start(sptps_t *s, void *handle, bool initiator, bool datagram, ecdsa_t *mykey, ecdsa_t *hiskey, const void *label, size_t labellen, send_data_t send_data, receive_record_t receive_record);
 extern bool sptps_stop(sptps_t *s);
 extern bool sptps_send_record(sptps_t *s, uint8_t type, const void *data, uint16_t len);
-extern bool sptps_receive_data(sptps_t *s, const void *data, size_t len);
+extern size_t sptps_receive_data(sptps_t *s, const void *data, size_t len);
 extern bool sptps_force_kex(sptps_t *s);
 extern bool sptps_verify_datagram(sptps_t *s, const void *data, size_t len);