From 770fc69429e6b61ca648b15816fd003819abe3e2 Mon Sep 17 00:00:00 2001
From: Uros Majstorovic <majstor@majstor.org>
Date: Sat, 12 Aug 2017 23:09:10 +0200
Subject: conn_close handler added

---
 code/core/core.c | 5 ++++-
 code/core/core.h | 2 ++
 code/core/rbuf.c | 2 +-
 3 files changed, 7 insertions(+), 2 deletions(-)

(limited to 'code/core')

diff --git a/code/core/core.c b/code/core/core.c
index 069c910..a7ef3de 100644
--- a/code/core/core.c
+++ b/code/core/core.c
@@ -580,7 +580,10 @@ int ecp_conn_close(ECPConnection *conn, unsigned int timeout) {
     if (refcount) return ECP_ERR_TIMEOUT;
 #endif
 
-    if (!conn->out) {
+    if (conn->out) {
+        ecp_conn_close_t *handler = conn->sock->ctx->handler[conn->type] ? conn->sock->ctx->handler[conn->type]->conn_close : NULL;
+        if (handler) handler(conn);
+    } else {
         ecp_conn_destroy_t *handler = conn->sock->ctx->handler[conn->type] ? conn->sock->ctx->handler[conn->type]->conn_destroy : NULL;
         if (handler) handler(conn);
         if (conn->proxy) {
diff --git a/code/core/core.h b/code/core/core.h
index 382b2c1..b0694b2 100644
--- a/code/core/core.h
+++ b/code/core/core.h
@@ -117,6 +117,7 @@ typedef void ecp_conn_free_t (struct ECPConnection *c);
 typedef int ecp_conn_create_t (struct ECPConnection *c, unsigned char *msg, size_t sz);
 typedef void ecp_conn_destroy_t (struct ECPConnection *c);
 typedef ssize_t ecp_conn_open_t (struct ECPConnection *c);
+typedef void ecp_conn_close_t (struct ECPConnection *c);
 
 typedef struct ECPCryptoIface {
     int init;
@@ -196,6 +197,7 @@ typedef struct ECPConnHandler {
     ecp_conn_create_t *conn_create;
     ecp_conn_destroy_t *conn_destroy;
     ecp_conn_open_t *conn_open;
+    ecp_conn_close_t *conn_close;
 } ECPConnHandler;
 
 typedef struct ECPSockCTable {
diff --git a/code/core/rbuf.c b/code/core/rbuf.c
index bb110b9..d92c3f1 100644
--- a/code/core/rbuf.c
+++ b/code/core/rbuf.c
@@ -51,7 +51,7 @@ int ecp_conn_rbuf_start(ECPConnection *conn, ecp_seq_t seq) {
     return ECP_OK;
 }
 
-ssize_t ecp_conn_rbuf_pkt_send(struct ECPConnection *conn, ECPNetAddr *addr, unsigned char *packet, size_t pkt_size, ecp_seq_t seq, int idx) {
+ssize_t ecp_conn_rbuf_pkt_send(ECPConnection *conn, ECPNetAddr *addr, unsigned char *packet, size_t pkt_size, ecp_seq_t seq, int idx) {
     int do_send;
     ECPRBSend *buf = conn->rbuf.send;
     
-- 
cgit v1.2.3