From b984dac55bdd0a5b694837fb8fb2ceb07ce9808f Mon Sep 17 00:00:00 2001
From: fujuntang <fujuntang@smartai.com>
Date: 星期二, 07 九月 2021 11:25:57 +0800
Subject: [PATCH] resize the buf allocation to dynamic buf.

---
 src/net/net_mod_socket_wrapper.cpp |   46 +++++++++++++++++++++++++++-------------------
 1 files changed, 27 insertions(+), 19 deletions(-)

diff --git a/src/net/net_mod_socket_wrapper.cpp b/src/net/net_mod_socket_wrapper.cpp
index d22a99a..ab4d59d 100644
--- a/src/net/net_mod_socket_wrapper.cpp
+++ b/src/net/net_mod_socket_wrapper.cpp
@@ -16,8 +16,8 @@
  * 鍏抽棴
  */
 void net_mod_socket_close(void *_socket) {
-	// NetModSocket *sockt = (NetModSocket *)_socket;
-	// delete sockt;
+	NetModSocket *sockt = (NetModSocket *)_socket;
+	delete sockt;
 }
  
 int net_mod_socket_stop(void *_socket) {
@@ -44,6 +44,14 @@
 	// return sockt->bind(key);
 }
 
+int net_mod_socket_reg(void *_socket, void *pData, int len, void **buf, int *size, const int timeout_ms, int flag)
+{
+  NetModSocket *sockt = (NetModSocket *)_socket;
+
+  return sockt->reg(pData, len, buf, size, timeout_ms, flag);
+  
+}
+
 /**
  * 鍙戦�佷俊鎭�
  * @key 鍙戦�佺粰璋�
@@ -51,20 +59,17 @@
  */
 int net_mod_socket_sendto(void *_socket, const void *buf, const int size, const int key) {
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	logger->debug("net_mod_socket_sendto: %d sendto  %d", net_mod_socket_get_key(_socket), key);
 	return sockt->sendto(buf, size, key);
 }
 // 鍙戦�佷俊鎭秴鏃惰繑鍥炪�� @sec 绉� 锛� @nsec 绾崇
 int net_mod_socket_sendto_timeout(void *_socket, const void *buf, const int size, const int key, int sec, int nsec){
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	logger->debug("net_mod_socket_sendto: %d sendto  %d", net_mod_socket_get_key(_socket), key);
 	return sockt->sendto_timeout(buf, size, key, sec, nsec);
 	// return sockt->sendto(buf, size, key);
 }
 // 鍙戦�佷俊鎭珛鍒昏繑鍥炪��
 int net_mod_socket_sendto_nowait(void *_socket, const void *buf, const int size, const int key){
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	logger->debug("net_mod_socket_sendto: %d sendto  %d", net_mod_socket_get_key(_socket), key);
 	return sockt->sendto_nowait(buf, size, key);
 }
 
@@ -77,28 +82,30 @@
 	int rv;
 	NetModSocket *sockt = (NetModSocket *)_socket;
 
-	logger->debug(" %d net_mod_socket_recvfrom before", net_mod_socket_get_key(_socket));
 	rv = sockt->recvfrom(buf, size, key);
-	logger->debug(" %d net_mod_socket_recvfrom after. rv = %d", net_mod_socket_get_key(_socket), rv);
 	return rv;
 }
 
 // 鎺ュ彈淇℃伅瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇
 int net_mod_socket_recvfrom_timeout(void *_socket, void **buf, int *size, int *key, int sec, int nsec){
-	NetModSocket *sockt = (NetModSocket *)_socket;
-	// return sockt->recvfrom(buf, size, key);
-	return sockt->recvfrom_timeout(buf, size, key, sec, nsec);
+  NetModSocket *sockt = (NetModSocket *)_socket;
+  return sockt->recvfrom_timeout(buf, size, key, sec, nsec);
 }
 
 int net_mod_socket_recvfrom_nowait(void *_socket, void **buf, int *size, int *key){
-	NetModSocket *sockt = (NetModSocket *)_socket;
-	return sockt->recvfrom_nowait(buf, size, key);
+  NetModSocket *sockt = (NetModSocket *)_socket;
+  return sockt->recvfrom_nowait(buf, size, key);
 }
 
 int net_mod_socket_sendandrecv(void *_socket, net_node_t *node_arr, int arrlen, void *send_buf, int send_size, 
-  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size){
+  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size,  net_mod_err_t ** err_arr, int *err_arr_size){
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	return sockt->sendandrecv(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size);
+	return sockt->sendandrecv(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size, err_arr, err_arr_size, -1);
+}
+
+int net_mod_socket_bind_proc_id(void * _socket, char *proc_id, int len){
+  NetModSocket *sockt = (NetModSocket *)_socket;
+  return sockt->bind_proc_id(proc_id, len);
 }
 
 /**
@@ -106,16 +113,17 @@
  * @timeout 绛夊緟鏃堕棿锛屽崟浣嶆槸鍗冨垎涔嬩竴绉�
 */
 int net_mod_socket_sendandrecv_timeout(void *_socket, net_node_t *node_arr, int arrlen, void *send_buf, int send_size, 
-  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size,  int timeout){
+  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size, 
+  net_mod_err_t ** err_arr, int *err_arr_size, int timeout){
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	// return sockt->sendandrecv(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size);
-	return sockt->sendandrecv_timeout(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size, timeout);
+	return sockt->sendandrecv(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size, err_arr, err_arr_size, timeout);
 }
 
 int net_mod_socket_sendandrecv_nowait(void *_socket, net_node_t *node_arr, int arrlen, void *send_buf, int send_size, 
-  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size) {
+  net_mod_recv_msg_t ** recv_arr, int *recv_arr_size, net_mod_err_t ** err_arr, int *err_arr_size) {
+
 	NetModSocket *sockt = (NetModSocket *)_socket;
-	return sockt->sendandrecv_nowait(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size);
+	return sockt->sendandrecv(node_arr,  arrlen, send_buf,  send_size, recv_arr, recv_arr_size, err_arr, err_arr_size, 0);
 }
 
 

--
Gitblit v1.8.0