receive_msg: receive custom message flags

Author: razvancrainea

modules/proto_hep/proto_hep.c

@@ -977,7 +977,7 @@ static inline int hep_handle_req(struct tcp_req *req,
 
 		/* skip receive msg if we were told so from at least one callback */
 		if ( ret != HEP_SCRIPT_SKIP ) {
-			if ( receive_msg(msg_buf, msg_len, &local_rcv, ctx) <0 ) {
+			if ( receive_msg(msg_buf, msg_len, &local_rcv, ctx, 0) <0 ) {
 				LM_ERR("receive_msg failed \n");
 			}
 		} else {
@@ -1312,7 +1312,7 @@ static int hep_udp_read_req(struct socket_info *si, int* bytes_read)
 
 	if (ret != HEP_SCRIPT_SKIP) {
 		/* receive_msg must free buf too!*/
-		receive_msg( msg.s, msg.len, &ri, ctx);
+		receive_msg( msg.s, msg.len, &ri, ctx, 0);
 	} else {
 		if ( ctx ) {
 			context_free( ctx );

modules/proto_sctp/sctp_server.c

@@ -182,7 +182,7 @@ int proto_sctp_read(struct socket_info *si, int* bytes_read)
 	}
 
 	/* receive_msg must free buf too!*/
-	receive_msg(buf, len, &ri, NULL);
+	receive_msg(buf, len, &ri, NULL, 0);
 
 	return 0;
 }

modules/proto_ws/ws_common.h

@@ -528,7 +528,7 @@ static int ws_process(struct tcp_connection *con)
 					"keeping connection \n");
 			}
 
-			if (receive_msg(msg_buf, msg_len, &local_rcv, NULL) <0)
+			if (receive_msg(msg_buf, msg_len, &local_rcv, NULL, 0) <0)
 					LM_ERR("receive_msg failed \n");
 
 			*req->tcp.parsed = bk;

modules/sipcapture/sipcapture.c

@@ -5546,7 +5546,7 @@ int raw_capture_rcv_loop(int rsock, int port1, int port2, int ipip) {
 		if((!port1 && !port2)
 		        || (src_port >= port1 && src_port <= port2) || (dst_port >= port1 && dst_port <= port2)
 		        || (!port2 && (src_port == port1 || dst_port == port1)))
-		                          receive_msg(buf+offset, len, &ri, NULL);
+		                          receive_msg(buf+offset, len, &ri, NULL, 0);
 	}
 
 	return 0;

modules/tm/cluster.c

@@ -78,7 +78,7 @@ static void tm_repl_received(bin_packet_t *packet)
 	/* TODO: mark the packet as replicated, to make sure we don't replicate it again */
 
 	/* all set up - process it */
-	receive_msg(tmp.s, tmp.len, &ri, NULL);
+	receive_msg(tmp.s, tmp.len, &ri, NULL, FL_TM_REPLICATED);
 }
 #undef TM_BIN_POP
 

net/proto_tcp/tcp_common.h

@@ -409,7 +409,7 @@ static inline int tcp_handle_req(struct tcp_req *req,
 			}
 
 			if (receive_msg(msg_buf, msg_len,
-				&local_rcv, NULL) <0)
+				&local_rcv, NULL, 0) <0)
 					LM_ERR("receive_msg failed \n");
 
 			if (!size && req != &_tcp_common_current_req) {

net/proto_udp/proto_udp.c

@@ -178,7 +178,7 @@ static int udp_read_req(struct socket_info *si, int* bytes_read)
 	}
 
 	/* receive_msg must free buf too!*/
-	receive_msg( msg.s, msg.len, &ri, NULL);
+	receive_msg( msg.s, msg.len, &ri, NULL, 0);
 
 	return 0;
 }

receive.c

@@ -97,7 +97,7 @@ unsigned int get_next_msg_no(void)
  * break (e.g.: modules/textops)
  */
 int receive_msg(char* buf, unsigned int len, struct receive_info* rcv_info,
-		context_p existing_context)
+		context_p existing_context, unsigned int flags)
 {
 	static context_p ctx = NULL;
 	struct sip_msg* msg;
@@ -139,6 +139,7 @@ int receive_msg(char* buf, unsigned int len, struct receive_info* rcv_info,
 	msg->len=len;
 	msg->rcv=*rcv_info;
 	msg->id=msg_no;
+	msg->flags=flags;
 	msg->ruri_q = Q_UNSPECIFIED;
 
 	if (parse_msg(in_buff.s,len, msg)!=0){

receive.h

@@ -31,7 +31,7 @@
 #include "context.h"
 
 int receive_msg(char* buf, unsigned int len, struct receive_info *ri,
-		context_p existing_context);
+		context_p existing_context, unsigned int flags);
 
 unsigned int get_next_msg_no(void);