restore deferred cleanup in rootfs.init

Signed-off-by: Daniel Nephin <dnephin@gmail.com>

gc/scheduler/scheduler.go

@@ -250,7 +250,6 @@ func (s *gcScheduler) run(ctx context.Context) {
 				schedC, nextCollection = schedule(interval)
 				continue
 			}
-			break
 		case e := <-s.eventC:
 			if lastCollection != nil && lastCollection.After(e.ts) {
 				continue

rootfs/init.go

@@ -70,6 +70,14 @@ func createInitLayer(ctx context.Context, parent, initName string, initFn func(s
 		return "", err
 	}
 
+	defer func() {
+		if err != nil {
+			if rerr := snapshotter.Remove(ctx, td); rerr != nil {
+				log.G(ctx).Errorf("Failed to remove snapshot %s: %v", td, rerr)
+			}
+		}
+	}()
+
 	if err = mounter.Mount(td, mounts...); err != nil {
 		return "", err
 	}