[ALL] Bugs, bugs, bugs...

3 files changed, 30 insertions, 30 deletions

diff --git a/src/kernel/blk.c b/src/kernel/blk.c
index 70d4e1f..1023743 100644
--- a/src/kernel/blk.c
+++ b/src/kernel/blk.c
@@ -95,8 +95,6 @@ struct block_device_operations dnbd3_blk_ops =
 int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, unsigned long arg)
 {
     dnbd3_device_t *dev = bdev->bd_disk->private_data;
-    int minor = dev->disk->first_minor;
-
     dnbd3_ioctl_t *msg = kmalloc(sizeof(dnbd3_ioctl_t), GFP_KERNEL);
     copy_from_user((char *)msg, (char *)arg, sizeof(*msg));
 
@@ -111,9 +109,8 @@ int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, u
         break;
 
     case IOCTL_CLOSE:
+        set_capacity(dev->disk, 0);
         dnbd3_net_disconnect(dev);
-        dnbd3_blk_del_device(dev);
-        dnbd3_blk_add_device(dev, minor);
         break;
 
     case IOCTL_SWITCH:
diff --git a/src/kernel/net.c b/src/kernel/net.c
index bdab8c2..a820a04 100644
--- a/src/kernel/net.c
+++ b/src/kernel/net.c
@@ -36,17 +36,17 @@ void dnbd3_net_connect(dnbd3_device_t *dev)
     // do some checks before connecting
     if (!req)
     {
-        printk("ERROR: Kmalloc failed.\n");
+        printk("FATAL: Kmalloc failed.\n");
         return;
     }
     if (!dev->cur_server.host || !dev->cur_server.port || (dev->vid == 0))
     {
-        printk("ERROR: Host, port or vid not set.\n");
+        printk("FATAL: Host, port or vid not set.\n");
         return;
     }
     if (dev->cur_server.sock)
     {
-        printk("ERROR: Device %s already connected to %s.\n", dev->disk->disk_name, dev->cur_server.host);
+        printk("ERROR: Device %s is already connected to %s.\n", dev->disk->disk_name, dev->cur_server.host);
         return;
     }
 
@@ -73,6 +73,7 @@ void dnbd3_net_connect(dnbd3_device_t *dev)
     }
 
     dev->panic = 0;
+    dev->alt_servers_num = 0;
 
     // enqueue request to request_queue_send (ask file size)
     req->cmd_type = REQ_TYPE_SPECIAL;
@@ -122,10 +123,12 @@ void dnbd3_net_disconnect(dnbd3_device_t *dev)
 
     // clear socket
     if (dev->cur_server.sock)
-    {
         sock_release(dev->cur_server.sock);
-        dev->cur_server.sock = NULL;
-    }
+
+    dev->thread_send = NULL;
+    dev->thread_receive = NULL;
+    dev->thread_discover = NULL;
+    dev->cur_server.sock = NULL;
 }
 
 void dnbd3_net_heartbeat(unsigned long arg)
@@ -320,6 +323,7 @@ int dnbd3_net_discover(void *data)
 
     }
     kfree(buf);
+    printk("INFO: dnbd3_net_discover exit\n");
     return 0;
 }
 
@@ -354,6 +358,8 @@ 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;
@@ -368,6 +374,8 @@ int dnbd3_net_send(void *data)
                 break;
             case REQ_GET_SERVERS:
                 dnbd3_request.cmd = CMD_GET_SERVERS;
+                dnbd3_request.vid = dev->vid;
+                dnbd3_request.rid = dev->rid;
                 break;
             }
             break;
@@ -394,6 +402,8 @@ int dnbd3_net_send(void *data)
         spin_unlock_irq(&dev->blk_lock);
         wake_up(&dev->process_queue_receive);
     }
+
+    printk("INFO: dnbd3_net_send exit\n");
     return 0;
 
     error:
@@ -450,29 +460,19 @@ int dnbd3_net_receive(void *data)
         }
         spin_unlock_irq(&dev->blk_lock);
 
-        // check if server send error
-        switch (dnbd3_reply.error)
+        // check error
+        if (dnbd3_reply.size == 0)
         {
-        case ERROR_SIZE:
             printk("FATAL: Requested image does't exist\n");
+            spin_lock_irq(&dev->blk_lock);
             list_del_init(&blk_request->queuelist);
-            kthread_stop(dev->thread_send);
-            del_timer(&dev->hb_timer);
-            sock_release(dev->cur_server.sock);
-            kfree(blk_request);
-            dev->cur_server.sock = NULL;
-            return -1;
-
-        case ERROR_RELOAD:
-            list_del_init(&blk_request->queuelist);
-            blk_request->cmd_type = REQ_TYPE_SPECIAL;
-            blk_request->cmd_flags = REQ_GET_FILESIZE;
-            list_add(&blk_request->queuelist, &dev->request_queue_send);
-            wake_up(&dev->process_queue_send);
+            spin_unlock_irq(&dev->blk_lock);
+            if ( (dnbd3_reply.cmd == CMD_GET_SIZE) || (dnbd3_reply.cmd == CMD_GET_SERVERS) )
+                kfree(blk_request);
             continue;
-
         }
 
+
         // what to do?
         switch (dnbd3_reply.cmd)
         {
@@ -543,6 +543,8 @@ int dnbd3_net_receive(void *data)
 
         }
     }
+
+    printk("INFO: dnbd3_net_receive exit\n");
     return 0;
 
     error:
diff --git a/src/kernel/sysfs.c b/src/kernel/sysfs.c
index e363824..177cb8c 100644
--- a/src/kernel/sysfs.c
+++ b/src/kernel/sysfs.c
@@ -163,7 +163,10 @@ struct sysfs_ops server_ops =
     .show = server_show,
 };
 
-void release(struct kobject *kobj) {}
+void release(struct kobject *kobj)
+{
+    kobj->state_initialized = 0;
+}
 
 struct kobj_type device_ktype =
 {
@@ -202,11 +205,9 @@ void dnbd3_sysfs_init(dnbd3_device_t *dev)
 void dnbd3_sysfs_exit(dnbd3_device_t *dev)
 {
     int i;
-
     for (i = 0; i < NUMBER_SERVERS; i++)
     {
         kobject_put(&dev->alt_servers[i].kobj);
     }
-
     kobject_put(&dev->kobj);
 }