From: Razvan Deaconescu <razvan.deaconescu@cs.pub.ro>
Date: Sat, 21 May 2011 09:35:51 +0000 (+0300)
Subject: raw: Add sw_shutdown "syscall".
X-Git-Url: http://p2p-next.cs.pub.ro/gitweb/?a=commitdiff_plain;h=9febfa2a7d846f9bbd2e7d546907d7c711607594;p=swifty.git

raw: Add sw_shutdown "syscall".
---

diff --git a/src/raw/swift_raw.c b/src/raw/swift_raw.c
index 186731b..50cd1df 100644
--- a/src/raw/swift_raw.c
+++ b/src/raw/swift_raw.c
@@ -306,6 +306,49 @@ int sw_setsockopt (int __fd, int __level, int __optname,
  */
 int sw_shutdown (int __fd, int __how)
 {
+	struct sock_list *list;
+
+	/* Find socket in management structure. */
+	list = list_elem_from_socket(__fd);
+	if (list == NULL) {
+		errno = EBADF;
+		goto list_elem_err;
+	}
+
+	/* Check and update socket state. */
+	if (__how == STATE_SHUT_RDWR)
+		list->rw_state = STATE_SHUT_RDWR;
+	else if (__how == STATE_SHUT_WR) {
+		if (list->rw_state == STATE_SHUT_RD)
+			list->rw_state = STATE_SHUT_RDWR;
+		else if (list->rw_state == STATE_SHUT_WR) {
+			errno = ENOTCONN;
+			goto not_conn_err;
+		}
+	}
+	else if (__how == STATE_SHUT_RD) {
+		if (list->rw_state == STATE_SHUT_WR)
+			list->rw_state = STATE_SHUT_RDWR;
+		else if (list->rw_state == STATE_SHUT_RD) {
+			errno = ENOTCONN;
+			goto not_conn_err;
+		}
+	}
+
+	/* Remove socket from socket management structure. */
+	if (list->rw_state == STATE_SHUT_RDWR) {
+		list = list_unlink_socket(__fd);
+		if (list == NULL) {
+			errno = EBADF;
+			goto list_unlink_err;
+		}
+	}
+
 	/* Call classical interface of shutdown(2). */
 	return shutdown(__fd, __how);
+
+not_conn_err:
+list_elem_err:
+list_unlink_err:
+	return -1;
 }