From 6c1bdb480814be0671c75faca39f023131ecd2bc Mon Sep 17 00:00:00 2001
From: Roy Shterman <roysh@mellanox.com>
Date: Fri, 27 May 2016 18:47:20 +0300
Subject: [PATCH] Libiscsi: Adding free_pdu function to transport abstraction

Signed-off-by: Roy Shterman <roysh@mellanox.com>
---
 include/iscsi-private.h |  5 +++-
 lib/connect.c           | 12 +++++-----
 lib/discovery.c         |  4 ++--
 lib/init.c              |  6 ++---
 lib/iscsi-command.c     | 12 +++++-----
 lib/login.c             | 52 ++++++++++++++++++++---------------------
 lib/nop.c               |  6 ++---
 lib/pdu.c               | 26 ++++++++++-----------
 lib/socket.c            |  3 ++-
 lib/task_mgmt.c         |  2 +-
 10 files changed, 66 insertions(+), 62 deletions(-)

diff --git a/include/iscsi-private.h b/include/iscsi-private.h
index fef0cfc..f24179a 100644
--- a/include/iscsi-private.h
+++ b/include/iscsi-private.h
@@ -387,17 +387,20 @@ struct iscsi_pdu *iscsi_tcp_new_pdu(struct iscsi_context *iscsi, size_t size);
 
 void iscsi_init_tcp_transport(struct iscsi_context *iscsi);
 
+void iscsi_tcp_free_pdu(struct iscsi_context *iscsi, struct iscsi_pdu *pdu);
+
 struct iscsi_transport {
 	int (*connect)(struct iscsi_context *iscsi, union socket_address *sa, int ai_family);
 	int (*queue_pdu)(struct iscsi_context *iscsi, struct iscsi_pdu *pdu);
 	struct iscsi_pdu* (*new_pdu)(struct iscsi_context *iscsi, size_t size);
 	int (*disconnect)(struct iscsi_context *iscsi);
+	void (*free_pdu)(struct iscsi_context *iscsi, struct iscsi_pdu *pdu);
 };
 
 struct tcp_transport {
 	struct iscsi_transport t;
 };
-        
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/lib/connect.c b/lib/connect.c
index 77b7668..f42ec8b 100644
--- a/lib/connect.c
+++ b/lib/connect.c
@@ -277,7 +277,7 @@ void iscsi_defer_reconnect(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED,
 			              NULL, pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 	while ((pdu = iscsi->waitpdu)) {
 		ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
@@ -288,7 +288,7 @@ void iscsi_defer_reconnect(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED,
 			              NULL, pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 }
 
@@ -332,7 +332,7 @@ void iscsi_reconnect_cb(struct iscsi_context *iscsi _U_, int status,
 
 		ISCSI_LIST_REMOVE(&old_iscsi->waitpdu, pdu);
 		if (pdu->itt == 0xffffffff) {
-			iscsi_free_pdu(old_iscsi, pdu);
+			iscsi->t->free_pdu(old_iscsi, pdu);
 			continue;
 		}
 
@@ -346,7 +346,7 @@ void iscsi_reconnect_cb(struct iscsi_context *iscsi _U_, int status,
 				pdu->callback(iscsi, SCSI_STATUS_CANCELLED,
 				              NULL, pdu->private_data);
 			}
-			iscsi_free_pdu(old_iscsi, pdu);
+			iscsi->t->free_pdu(old_iscsi, pdu);
 			continue;
 		}
 
@@ -364,7 +364,7 @@ void iscsi_reconnect_cb(struct iscsi_context *iscsi _U_, int status,
 					     pdu->scsi_cbdata.private_data)) {
 			/* not much we can really do at this point */
 		}
-		iscsi_free_pdu(old_iscsi, pdu);
+		iscsi->t->free_pdu(old_iscsi, pdu);
 	}
 
 	if (old_iscsi->incoming != NULL) {
@@ -375,7 +375,7 @@ void iscsi_reconnect_cb(struct iscsi_context *iscsi _U_, int status,
 	}
 
 	if (old_iscsi->outqueue_current != NULL && old_iscsi->outqueue_current->flags & ISCSI_PDU_DELETE_WHEN_SENT) {
-		iscsi_free_pdu(old_iscsi, old_iscsi->outqueue_current);
+		iscsi->t->free_pdu(old_iscsi, old_iscsi->outqueue_current);
 	}
 
 	for (i = 0; i < old_iscsi->smalloc_free; i++) {
diff --git a/lib/discovery.c b/lib/discovery.c
index 5e803a4..349dbd6 100644
--- a/lib/discovery.c
+++ b/lib/discovery.c
@@ -73,7 +73,7 @@ iscsi_discovery_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	if (iscsi_pdu_add_data(iscsi, pdu, (unsigned char *)str, strlen(str)+1)
 	    != 0) {
 		iscsi_set_error(iscsi, "Out-of-memory: pdu add data failed.");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
@@ -83,7 +83,7 @@ iscsi_discovery_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "Out-of-memory: failed to queue iscsi "
 				"text pdu.");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
diff --git a/lib/init.c b/lib/init.c
index 3ef035b..1349695 100644
--- a/lib/init.c
+++ b/lib/init.c
@@ -350,7 +350,7 @@ iscsi_destroy_context(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 			              pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 	while ((pdu = iscsi->waitpdu)) {
 		ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
@@ -361,11 +361,11 @@ iscsi_destroy_context(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 			              pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 
 	if (iscsi->outqueue_current != NULL && iscsi->outqueue_current->flags & ISCSI_PDU_DELETE_WHEN_SENT) {
-		iscsi_free_pdu(iscsi, iscsi->outqueue_current);
+		iscsi->t->free_pdu(iscsi, iscsi->outqueue_current);
 	}
 
 	if (iscsi->incoming != NULL) {
diff --git a/lib/iscsi-command.c b/lib/iscsi-command.c
index 406fd19..d8349cb 100644
--- a/lib/iscsi-command.c
+++ b/lib/iscsi-command.c
@@ -147,7 +147,7 @@ error:
 		cmd_pdu->callback(iscsi, SCSI_STATUS_ERROR, NULL,
 						  cmd_pdu->private_data);
 	}
-	iscsi_free_pdu(iscsi, cmd_pdu);
+	iscsi->t->free_pdu(iscsi, cmd_pdu);
 	return -1;
 }
 
@@ -283,7 +283,7 @@ iscsi_scsi_command_async(struct iscsi_context *iscsi, int lun,
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "Out-of-memory: failed to queue iscsi "
 				"scsi pdu.");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
@@ -1922,7 +1922,7 @@ iscsi_scsi_cancel_task(struct iscsi_context *iscsi,
 				pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 				      pdu->private_data);
 			}
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 			return 0;
 		}
 	}
@@ -1933,7 +1933,7 @@ iscsi_scsi_cancel_task(struct iscsi_context *iscsi,
 				pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 				      pdu->private_data);
 			}
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 			return 0;
 		}
 	}
@@ -1951,7 +1951,7 @@ iscsi_scsi_cancel_all_tasks(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 				      pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 	while ((pdu = iscsi->outqueue)) {
 		ISCSI_LIST_REMOVE(&iscsi->outqueue, pdu);
@@ -1959,6 +1959,6 @@ iscsi_scsi_cancel_all_tasks(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_CANCELLED, NULL,
 				      pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 }
diff --git a/lib/login.c b/lib/login.c
index 1346f85..896b691 100644
--- a/lib/login.c
+++ b/lib/login.c
@@ -852,14 +852,14 @@ iscsi_login_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 
 	/* initiator name */
 	if (iscsi_login_add_initiatorname(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* optional alias */
 	if (iscsi->alias[0]) {
 		if (iscsi_login_add_alias(iscsi, pdu) != 0) {
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 			return -1;
 		}
 	}
@@ -867,134 +867,134 @@ iscsi_login_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	/* target name */
 	if (iscsi->session_type == ISCSI_SESSION_NORMAL) {
 		if (iscsi_login_add_targetname(iscsi, pdu) != 0) {
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 			return -1;
 		}
 	}
 
 	/* session type */
 	if (iscsi_login_add_sessiontype(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* header digest */
 	if (iscsi_login_add_headerdigest(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* auth method */
 	if (iscsi_login_add_authmethod(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* auth algorithm */
 	if (iscsi_login_add_authalgorithm(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* chap username */
 	if (iscsi_login_add_chap_username(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* chap response */
 	if (iscsi_login_add_chap_response(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* data digest */
 	if (iscsi_login_add_datadigest(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* initial r2t */
 	if (iscsi_login_add_initialr2t(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* immediate data */
 	if (iscsi_login_add_immediatedata(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* max burst length */
 	if (iscsi_login_add_maxburstlength(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* first burst length */
 	if (iscsi_login_add_firstburstlength(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* default time 2 wait */
 	if (iscsi_login_add_defaulttime2wait(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* default time 2 retain */
 	if (iscsi_login_add_defaulttime2retain(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* max outstanding r2t */
 	if (iscsi_login_add_maxoutstandingr2t(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* errorrecoverylevel */
 	if (iscsi_login_add_errorrecoverylevel(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* ifmarker */
 	if (iscsi_login_add_ifmarker(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* ofmarker */
 	if (iscsi_login_add_ofmarker(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* maxconnections */
 	if (iscsi_login_add_maxconnections(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* max recv data segment length */
 	if (iscsi_login_add_maxrecvdatasegmentlength(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* data pdu in order */
 	if (iscsi_login_add_datapduinorder(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
 	/* data sequence in order */
 	if (iscsi_login_add_datasequenceinorder(iscsi, pdu) != 0) {
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
@@ -1005,7 +1005,7 @@ iscsi_login_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "Out-of-memory: failed to queue iscsi "
 				"pdu.");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
@@ -1322,7 +1322,7 @@ iscsi_logout_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "Out-of-memory: failed to queue iscsi "
 				"logout pdu.");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
diff --git a/lib/nop.c b/lib/nop.c
index d5a08eb..639f4c0 100644
--- a/lib/nop.c
+++ b/lib/nop.c
@@ -73,14 +73,14 @@ iscsi_nop_out_async(struct iscsi_context *iscsi, iscsi_command_cb cb,
 	if (data != NULL && len > 0) {
 		if (iscsi_pdu_add_data(iscsi, pdu, data, len) != 0) {
 			iscsi_set_error(iscsi, "Failed to add outdata to nop-out");
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 			return -1;
 		}
 	}
 
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "failed to queue iscsi nop-out pdu");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
@@ -124,7 +124,7 @@ iscsi_send_target_nop_out(struct iscsi_context *iscsi, uint32_t ttt, uint32_t lu
 
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "failed to queue iscsi nop-out pdu");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}
 
diff --git a/lib/pdu.c b/lib/pdu.c
index 5f6f79c..ea61b94 100644
--- a/lib/pdu.c
+++ b/lib/pdu.c
@@ -131,7 +131,7 @@ iscsi_allocate_pdu(struct iscsi_context *iscsi, enum iscsi_opcode opcode,
 }
 
 void
-iscsi_free_pdu(struct iscsi_context *iscsi, struct iscsi_pdu *pdu)
+iscsi_tcp_free_pdu(struct iscsi_context *iscsi, struct iscsi_pdu *pdu)
 {
 	if (pdu == NULL) {
 		iscsi_set_error(iscsi, "trying to free NULL pdu");
@@ -379,7 +379,7 @@ int iscsi_process_reject(struct iscsi_context *iscsi,
 	}
 
 	ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-	iscsi_free_pdu(iscsi, pdu);
+	iscsi->t->free_pdu(iscsi, pdu);
 	return 0;
 }
 
@@ -528,7 +528,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_LOGIN_RESPONSE:
 			if (iscsi_process_login_reply(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi login reply "
 						"failed");
 				return -1;
@@ -537,7 +537,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_TEXT_RESPONSE:
 			if (iscsi_process_text_reply(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi text reply "
 						"failed");
 				return -1;
@@ -546,7 +546,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_LOGOUT_RESPONSE:
 			if (iscsi_process_logout_reply(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi logout reply "
 						"failed");
 				return -1;
@@ -555,7 +555,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_SCSI_RESPONSE:
 			if (iscsi_process_scsi_reply(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi response reply "
 						"failed");
 				return -1;
@@ -565,7 +565,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 			if (iscsi_process_scsi_data_in(iscsi, pdu, in,
 						       &is_finished) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi data in "
 						"failed");
 				return -1;
@@ -574,7 +574,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_NOP_IN:
 			if (iscsi_process_nop_out_reply(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi nop-in failed");
 				return -1;
 			}
@@ -583,7 +583,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 			if (iscsi_process_task_mgmt_reply(iscsi, pdu,
 							  in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi task-mgmt failed");
 				return -1;
 			}
@@ -591,7 +591,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 		case ISCSI_PDU_R2T:
 			if (iscsi_process_r2t(iscsi, pdu, in) != 0) {
 				ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-				iscsi_free_pdu(iscsi, pdu);
+				iscsi->t->free_pdu(iscsi, pdu);
 				iscsi_set_error(iscsi, "iscsi r2t "
 						"failed");
 				return -1;
@@ -606,7 +606,7 @@ iscsi_process_pdu(struct iscsi_context *iscsi, struct iscsi_in_pdu *in)
 
 		if (is_finished) {
 			ISCSI_LIST_REMOVE(&iscsi->waitpdu, pdu);
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 		}
 		return 0;
 	}
@@ -720,7 +720,7 @@ iscsi_timeout_scan(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_TIMEOUT,
 			              NULL, pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 	for (pdu = iscsi->waitpdu; pdu; pdu = next_pdu) {
 		next_pdu = pdu->next;
@@ -740,6 +740,6 @@ iscsi_timeout_scan(struct iscsi_context *iscsi)
 			pdu->callback(iscsi, SCSI_STATUS_TIMEOUT,
 			              NULL, pdu->private_data);
 		}
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 	}
 }
diff --git a/lib/socket.c b/lib/socket.c
index d72c3e8..c4ba1ad 100644
--- a/lib/socket.c
+++ b/lib/socket.c
@@ -796,7 +796,7 @@ iscsi_write_to_socket(struct iscsi_context *iscsi)
 			iscsi->is_corked = 1;
 		}
 		if (pdu->flags & ISCSI_PDU_DELETE_WHEN_SENT) {
-			iscsi_free_pdu(iscsi, pdu);
+			iscsi->t->free_pdu(iscsi, pdu);
 		}
 		iscsi->outqueue_current = NULL;
 	}
@@ -1065,6 +1065,7 @@ void iscsi_init_tcp_transport(struct iscsi_context *iscsi)
 	iscsi->t->queue_pdu = iscsi_tcp_queue_pdu;
 	iscsi->t->new_pdu = iscsi_tcp_new_pdu;
 	iscsi->t->disconnect = iscsi_tcp_disconnect;
+	iscsi->t->free_pdu = iscsi_tcp_free_pdu;
 
 	return;
 }
diff --git a/lib/task_mgmt.c b/lib/task_mgmt.c
index b4d7242..37c2e15 100644
--- a/lib/task_mgmt.c
+++ b/lib/task_mgmt.c
@@ -83,7 +83,7 @@ iscsi_task_mgmt_async(struct iscsi_context *iscsi,
 
 	if (iscsi->t->queue_pdu(iscsi, pdu) != 0) {
 		iscsi_set_error(iscsi, "failed to queue iscsi taskmgmt pdu");
-		iscsi_free_pdu(iscsi, pdu);
+		iscsi->t->free_pdu(iscsi, pdu);
 		return -1;
 	}