a series of fixups for nydusd

service/src/daemon.rs

@@ -422,13 +422,16 @@ impl DaemonController {
         self.fs_service.lock().unwrap().clone()
     }
 
-    /// Shutdown all services managed by the controller.
-    pub fn shutdown(&self) {
+    /// Notify controller shutdown
+    pub fn notify_shutdown(&self) {
         // Marking exiting state.
         self.active.store(false, Ordering::Release);
         // Signal the `run_loop()` working thread to exit.
         let _ = self.waker.wake();
+    }
 
+    /// Shutdown all services managed by the controller.
+    pub fn shutdown(&self) {
         let daemon = self.daemon.lock().unwrap().take();
         if let Some(d) = daemon {
             if let Err(e) = d.trigger_stop() {

src/bin/nydusd/main.rs

@@ -720,7 +720,7 @@ mod nbd {
 }
 
 extern "C" fn sig_exit(_sig: std::os::raw::c_int) {
-    DAEMON_CONTROLLER.shutdown();
+    DAEMON_CONTROLLER.notify_shutdown();
 }
 
 fn main() -> Result<()> {

storage/src/cache/cachedfile.rs

@@ -1662,6 +1662,20 @@ impl FileIoMergeState {
         tag: BlobIoTag,
         chunk: Option<Arc<dyn BlobChunkInfo>>,
     ) -> Result<()> {
+        // Make sure user io of same region continuous
+        if !self.regions.is_empty() && self.joinable(region_type) {
+            let region = &self.regions[self.regions.len() - 1];
+            if !region.seg.is_empty() && tag.is_user_io() {
+                if let BlobIoTag::User(ref seg) = tag {
+                    if seg.offset as u64 + start
+                        != region.blob_address + region.seg.offset as u64 + region.seg.len as u64
+                    {
+                        self.commit();
+                    }
+                }
+            }
+        }
+
         if self.regions.is_empty() || !self.joinable(region_type) {
             self.regions.push(Region::new(region_type));
             self.last_region_joinable = true;
@@ -1793,7 +1807,7 @@ mod tests {
 
         let tag = BlobIoTag::User(BlobIoSegment {
             offset: 0x1800,
-            len: 0x1800,
+            len: 0x800,
         });
         state
             .push(RegionType::CacheFast, 0x1000, 0x2000, tag, None)
@@ -1810,8 +1824,8 @@ mod tests {
         assert_eq!(state.regions.len(), 1);
 
         let tag = BlobIoTag::User(BlobIoSegment {
-            offset: 0x0000,
-            len: 0x2000,
+            offset: 0x0001,
+            len: 0x1fff,
         });
         state
             .push(RegionType::CacheSlow, 0x5000, 0x2000, tag, None)

storage/src/cache/filecache/mod.rs

@@ -257,7 +257,7 @@ impl FileCacheEntry {
             } else {
                 blob_info.uncompressed_size()
             };
-            if file_size == 0 {
+            if file_size == 0 || file_size < cached_file_size {
                 file.set_len(cached_file_size)?;
             } else if cached_file_size != 0 && file_size != cached_file_size {
                 let msg = format!(