51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
+#ifndef __SPTPS_H__
+#define __SPTPS_H__
+
#include "system.h"
#include "cipher.h"
#include "ecdh.h"
#include "ecdsa.h"
-#define SPTPS_KEX 0
-#define SPTPS_SECONDARY_KEX 1 // Waiting for peer's ECDHE pubkey
-#define SPTPS_SIG 2 // Waiting for peer's ECDHE pubkey
-#define SPTPS_ACK 3 // Waiting for peer's acknowledgement of pubkey reception
+#define SPTPS_VERSION 0
+
+// Record types
+#define SPTPS_HANDSHAKE 128 // Key exchange and authentication
+#define SPTPS_ALERT 129 // Warning or error messages
+#define SPTPS_CLOSE 130 // Application closed the connection
-#define SPTPS_HANDSHAKE 128
-#define SPTPS_VERSION 128
+// Key exchange states
+#define SPTPS_KEX 0 // Waiting for the first Key EXchange record
+#define SPTPS_SECONDARY_KEX 1 // Ready to receive a secondary Key EXchange record
+#define SPTPS_SIG 2 // Waiting for a SIGnature record
+#define SPTPS_ACK 3 // Waiting for an ACKnowledgement record
-typedef bool (*send_data_t)(void *handle, const char *data, size_t len);
+typedef bool (*send_data_t)(void *handle, uint8_t type, const char *data, size_t len);
typedef bool (*receive_record_t)(void *handle, uint8_t type, const char *data, uint16_t len);
typedef struct sptps {
bool initiator;
+ bool datagram;
int state;
char *inbuf;
size_t buflen;
+ uint16_t reclen;
bool instate;
cipher_t incipher;
receive_record_t receive_record;
} sptps_t;
-extern bool start_sptps(sptps_t *s, void *handle, bool initiator, ecdsa_t mykey, ecdsa_t hiskey, const char *label, size_t labellen, send_data_t send_data, receive_record_t receive_record);
-extern bool stop_sptps(sptps_t *s);
-extern bool send_record(sptps_t *s, uint8_t type, const char *data, uint16_t len);
-extern bool receive_data(sptps_t *s, const char *data, size_t len);
-extern bool force_kex(sptps_t *s);
+extern bool sptps_start(sptps_t *s, void *handle, bool initiator, bool datagram, ecdsa_t mykey, ecdsa_t hiskey, const char *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 char *data, uint16_t len);
+extern bool sptps_receive_data(sptps_t *s, const char *data, size_t len);
+extern bool sptps_force_kex(sptps_t *s);
+extern bool sptps_verify_datagram(sptps_t *s, const char *data, size_t len);
+
+#endif