[ALL] Some cleanup

5 files changed, 41 insertions, 39 deletions

diff --git a/src/kernel/blk.c b/src/kernel/blk.c
index 369627f..2f52cc3 100644
--- a/src/kernel/blk.c
+++ b/src/kernel/blk.c
@@ -99,6 +99,7 @@ struct block_device_operations dnbd3_blk_ops =
 
 int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, unsigned long arg)
 {
+    int result = 0;
     dnbd3_device_t *dev = bdev->bd_disk->private_data;
     struct request_queue *blk_queue = dev->disk->queue;
     dnbd3_ioctl_t *msg = kmalloc(sizeof(dnbd3_ioctl_t), GFP_KERNEL);
@@ -108,33 +109,34 @@ int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, u
     {
     case IOCTL_OPEN:
         strcpy(dev->cur_server.host, msg->host);
-        strcpy(dev->cur_server.port, msg->port);
+        strcpy(dev->cur_server.port, PORTSTR);
         dev->vid = msg->vid;
         dev->rid = msg->rid;
         blk_queue->backing_dev_info.ra_pages = (msg->read_ahead_kb * 1024)/ PAGE_CACHE_SIZE;
-        return dnbd3_net_connect(dev);
+        result =  dnbd3_net_connect(dev);
+        break;
 
     case IOCTL_CLOSE:
         set_capacity(dev->disk, 0);
-        dnbd3_net_disconnect(dev);
+        result = dnbd3_net_disconnect(dev);
         break;
 
     case IOCTL_SWITCH:
         dnbd3_net_disconnect(dev);
         strcpy(dev->cur_server.host, msg->host);
-        return dnbd3_net_connect(dev);
+        result = dnbd3_net_connect(dev);
+        break;
 
     case BLKFLSBUF:
         break;
 
     default:
-        kfree(msg);
-        return -1;
+        result = -EIO;
 
     }
 
     kfree(msg);
-    return 0;
+    return result;
 }
 
 void dnbd3_blk_request(struct request_queue *q)
diff --git a/src/kernel/net.c b/src/kernel/net.c
index ad4884c..840282f 100644
--- a/src/kernel/net.c
+++ b/src/kernel/net.c
@@ -116,7 +116,7 @@ int dnbd3_net_connect(dnbd3_device_t *dev)
     return 0;
 }
 
-void dnbd3_net_disconnect(dnbd3_device_t *dev)
+int dnbd3_net_disconnect(dnbd3_device_t *dev)
 {
     printk("INFO: Disconnecting device %s\n", dev->disk->disk_name);
 
@@ -150,6 +150,8 @@ void dnbd3_net_disconnect(dnbd3_device_t *dev)
         dev->cur_server.sock = NULL;
         memset(dev->cur_server.host, '\0', sizeof(dev->cur_server.host));
     }
+
+    return 0;
 }
 
 void dnbd3_net_heartbeat(unsigned long arg)
@@ -157,6 +159,7 @@ void dnbd3_net_heartbeat(unsigned long arg)
     dnbd3_device_t *dev = (dnbd3_device_t *) arg;
     struct request *req = kmalloc(sizeof(struct request), GFP_ATOMIC);
 
+    // send keepalive
     if (req)
     {
         req->cmd_type = REQ_TYPE_SPECIAL;
@@ -164,7 +167,12 @@ void dnbd3_net_heartbeat(unsigned long arg)
         list_add_tail(&req->queuelist, &dev->request_queue_send);
         wake_up(&dev->process_queue_send);
     }
+    else
+    {
+        printk("ERROR: Couldn't create keepalive request\n");
+    }
 
+    // start discover
     dev->discover = 1;
     wake_up(&dev->process_queue_discover);
 
@@ -255,7 +263,7 @@ int dnbd3_net_discover(void *data)
             if (kernel_sendmsg(sock, &msg, &iov, 1, sizeof(dnbd3_request)) <= 0)
                 goto error;
 
-            // receive net replay
+            // receive net reply
             iov.iov_base = &dnbd3_reply;
             iov.iov_len = sizeof(dnbd3_reply);
             if (kernel_recvmsg(sock, &msg, &iov, 1, sizeof(dnbd3_reply), msg.msg_flags) <= 0)
@@ -319,8 +327,7 @@ int dnbd3_net_discover(void *data)
                                        +dev->alt_servers[i].rtts[3] ) / 4;
 
 
-
-            if ( best_rtt > dev->alt_servers[i].rtt)
+            if (best_rtt > dev->alt_servers[i].rtt)
             {
             	best_rtt = dev->alt_servers[i].rtt;
             	strcpy(best_server, current_server);
@@ -346,7 +353,8 @@ int dnbd3_net_discover(void *data)
             continue;
 
         // take server with lowest rtt
-        if (ready && num > 1 && strcmp(dev->cur_server.host, best_server) && !kthread_should_stop() && dev->cur_server.rtt > best_rtt + RTT_THRESHOLD)
+        if (ready && num > 1 && strcmp(dev->cur_server.host, best_server) && !kthread_should_stop()
+                                                 && dev->cur_server.rtt > best_rtt + RTT_THRESHOLD)
         {
             printk("INFO: Server %s on %s is faster (%lluus)\n", best_server, dev->disk->disk_name, best_rtt);
         	kfree(buf);
@@ -357,10 +365,6 @@ int dnbd3_net_discover(void *data)
             dnbd3_net_connect(dev);
             return 0;
         }
-        else
-        {
-            dev->cur_server.rtt = best_rtt;
-        }
 
         turn = (turn +1) % 4;
         if (turn == 3)
@@ -382,6 +386,9 @@ int dnbd3_net_send(void *data)
 
     init_msghdr(msg);
 
+    dnbd3_request.vid = dev->vid;
+    dnbd3_request.rid = dev->rid;
+
     set_user_nice(current, -20);
 
     while (!kthread_should_stop() || !list_empty(&dev->request_queue_send))
@@ -402,8 +409,6 @@ int dnbd3_net_send(void *data)
         {
         case REQ_TYPE_FS:
             dnbd3_request.cmd = CMD_GET_BLOCK;
-            dnbd3_request.vid = dev->vid;
-            dnbd3_request.rid = dev->rid;
             dnbd3_request.offset = blk_rq_pos(blk_request) << 9; // *512
             dnbd3_request.size = blk_rq_bytes(blk_request); // bytes left to complete entire request
             break;
@@ -413,13 +418,9 @@ int dnbd3_net_send(void *data)
             {
             case REQ_GET_FILESIZE:
                 dnbd3_request.cmd = CMD_GET_SIZE;
-                dnbd3_request.vid = dev->vid;
-                dnbd3_request.rid = dev->rid;
                 break;
             case REQ_GET_SERVERS:
                 dnbd3_request.cmd = CMD_GET_SERVERS;
-                dnbd3_request.vid = dev->vid;
-                dnbd3_request.rid = dev->rid;
                 break;
             }
             break;
@@ -487,7 +488,7 @@ int dnbd3_net_receive(void *data)
         if (list_empty(&dev->request_queue_receive))
             continue;
 
-        // receive net replay
+        // receive net reply
         iov.iov_base = &dnbd3_reply;
         iov.iov_len = sizeof(dnbd3_reply);
         if (kernel_recvmsg(dev->cur_server.sock, &msg, &iov, 1, sizeof(dnbd3_reply), msg.msg_flags) <= 0)
diff --git a/src/kernel/net.h b/src/kernel/net.h
index 3dfca41..64816bc 100644
--- a/src/kernel/net.h
+++ b/src/kernel/net.h
@@ -32,7 +32,7 @@
 
 int dnbd3_net_connect(dnbd3_device_t *lo);
 
-void dnbd3_net_disconnect(dnbd3_device_t *lo);
+int dnbd3_net_disconnect(dnbd3_device_t *lo);
 
 int dnbd3_net_send(void *data);
 
diff --git a/src/kernel/sysfs.c b/src/kernel/sysfs.c
index fe4321e..a455bd2 100644
--- a/src/kernel/sysfs.c
+++ b/src/kernel/sysfs.c
@@ -23,21 +23,6 @@
 #include "sysfs.h"
 #include "utils.h"
 
-typedef struct
-{
-    struct attribute attr;
-    ssize_t (*show)(char *, dnbd3_device_t *);
-    ssize_t (*store)(const char *, size_t, dnbd3_device_t *);
-} device_attr_t;
-
-typedef struct
-{
-    struct attribute attr;
-    ssize_t (*show)(char *, dnbd3_server_t *);
-    ssize_t (*store)(const char *, size_t, dnbd3_server_t *);
-} server_attr_t;
-
-
 ssize_t show_cur_server_ip(char *buf, dnbd3_device_t *dev)
 {
     return sprintf(buf, "%s\n", dev->cur_server.host);
diff --git a/src/kernel/sysfs.h b/src/kernel/sysfs.h
index 09a053f..ab5ceb5 100644
--- a/src/kernel/sysfs.h
+++ b/src/kernel/sysfs.h
@@ -27,5 +27,19 @@ void dnbd3_sysfs_init(dnbd3_device_t *dev);
 
 void dnbd3_sysfs_exit(dnbd3_device_t *dev);
 
+typedef struct
+{
+    struct attribute attr;
+    ssize_t (*show)(char *, dnbd3_device_t *);
+    ssize_t (*store)(const char *, size_t, dnbd3_device_t *);
+} device_attr_t;
+
+typedef struct
+{
+    struct attribute attr;
+    ssize_t (*show)(char *, dnbd3_server_t *);
+    ssize_t (*store)(const char *, size_t, dnbd3_server_t *);
+} server_attr_t;
+
 
 #endif /* SYSFS_H_ */