looks like the disk_lock method actually does work

1 files changed, 18 insertions, 20 deletions

diff --git a/sys/dev/vnd.c b/sys/dev/vnd.c
index 6f392994f9f..385ea386994 100644
--- a/sys/dev/vnd.c
+++ b/sys/dev/vnd.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: vnd.c,v 1.132 2011/06/20 22:14:17 tedu Exp $	*/
+/*	$OpenBSD: vnd.c,v 1.133 2011/06/21 01:47:15 deraadt Exp $	*/
 /*	$NetBSD: vnd.c,v 1.26 1996/03/30 23:06:11 christos Exp $	*/
 
 /*
@@ -61,7 +61,6 @@
 #include <sys/mount.h>
 #include <sys/vnode.h>
 #include <sys/file.h>
-#include <sys/rwlock.h>
 #include <sys/uio.h>
 #include <sys/conf.h>
 #include <sys/dkio.h>
@@ -105,7 +104,6 @@ struct vnd_softc {
 	struct vnode	*sc_vp;			/* vnode */
 	struct ucred	*sc_cred;		/* credentials */
 	blf_ctx		*sc_keyctx;		/* key context */
-	struct rwlock	 sc_rwlock;
 };
 
 /* sc_flags */
@@ -169,10 +167,10 @@ vndattach(int num)
 	for (i = 0; i < num; i++) {
 		struct vnd_softc *sc = &vnd_softc[i];
 
-		rw_init(&sc->sc_rwlock, "vndlock");
 		sc->sc_dev.dv_unit = i;
 		snprintf(sc->sc_dev.dv_xname, sizeof(sc->sc_dev.dv_xname),
 		    "vnd%d", i);
+		disk_construct(&sc->sc_dk);
 		device_ref(&sc->sc_dev);
 	}
 	numvnd = num;
@@ -191,7 +189,7 @@ vndopen(dev_t dev, int flags, int mode, struct proc *p)
 		return (ENXIO);
 	sc = &vnd_softc[unit];
 
-	if ((error = rw_enter(&sc->sc_rwlock, RW_WRITE|RW_INTR)) != 0)
+	if ((error = disk_lock(&sc->sc_dk)) != 0)
 		return (error);
 
 	if ((flags & FWRITE) && (sc->sc_flags & VNF_READONLY)) {
@@ -231,7 +229,7 @@ vndopen(dev_t dev, int flags, int mode, struct proc *p)
 	    sc->sc_dk.dk_copenmask | sc->sc_dk.dk_bopenmask;
 
 bad:
-	rw_exit_write(&sc->sc_rwlock);
+	disk_unlock(&sc->sc_dk);
 	return (error);
 }
 
@@ -278,7 +276,7 @@ vndclose(dev_t dev, int flags, int mode, struct proc *p)
 		return (ENXIO);
 	sc = &vnd_softc[unit];
 
-	rw_enter(&sc->sc_rwlock, RW_WRITE);
+	disk_lock_nointr(&sc->sc_dk);
 
 	part = DISKPART(dev);
 
@@ -295,7 +293,7 @@ vndclose(dev_t dev, int flags, int mode, struct proc *p)
 	sc->sc_dk.dk_openmask =
 	    sc->sc_dk.dk_copenmask | sc->sc_dk.dk_bopenmask;
 
-	rw_exit_write(&sc->sc_rwlock);
+	disk_unlock(&sc->sc_dk);
 	return (0);
 }
 
@@ -483,12 +481,12 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		if (sc->sc_flags & VNF_INITED)
 			return (EBUSY);
 
-		if ((error = rw_enter(&sc->sc_rwlock, RW_WRITE|RW_INTR)) != 0)
+		if ((error = disk_lock(&sc->sc_dk)) != 0)
 			return (error);
 
 		if ((error = copyinstr(vio->vnd_file, sc->sc_file,
 		    sizeof(sc->sc_file), NULL))) {
-			rw_exit_write(&sc->sc_rwlock);
+			disk_unlock(&sc->sc_dk);
 			return (error);
 		}
 
@@ -510,7 +508,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 			error = vn_open(&nd, FREAD, 0);
 		}
 		if (error) {
-			rw_exit_write(&sc->sc_rwlock);
+			disk_unlock(&sc->sc_dk);
 			return (error);
 		}
 
@@ -521,7 +519,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 			if (error) {
 				VOP_UNLOCK(nd.ni_vp, 0, p);
 				vn_close(nd.ni_vp, VNDRW(sc), p->p_ucred, p);
-				rw_exit_write(&sc->sc_rwlock);
+				disk_unlock(&sc->sc_dk);
 				return (error);
 			}
 			sc->sc_size = vattr.va_size / sc->sc_secsize;
@@ -530,7 +528,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		sc->sc_vp = nd.ni_vp;
 		if ((error = vndsetcred(sc, p->p_ucred)) != 0) {
 			(void) vn_close(nd.ni_vp, VNDRW(sc), p->p_ucred, p);
-			rw_exit_write(&sc->sc_rwlock);
+			disk_unlock(&sc->sc_dk);
 			return (error);
 		}
 
@@ -544,7 +542,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 			    vio->vnd_keylen)) != 0) {
 				(void) vn_close(nd.ni_vp, VNDRW(sc),
 				    p->p_ucred, p);
-				rw_exit_write(&sc->sc_rwlock);
+				disk_unlock(&sc->sc_dk);
 				return (error);
 			}
 
@@ -565,7 +563,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		sc->sc_dk.dk_name = sc->sc_dev.dv_xname;
 		disk_attach(&sc->sc_dev, &sc->sc_dk);
 
-		rw_exit_write(&sc->sc_rwlock);
+		disk_unlock(&sc->sc_dk);
 
 		break;
 
@@ -573,7 +571,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		if ((sc->sc_flags & VNF_INITED) == 0)
 			return (ENXIO);
 
-		if ((error = rw_enter(&sc->sc_rwlock, RW_WRITE|RW_INTR)) != 0)
+		if ((error = disk_lock(&sc->sc_dk)) != 0)
 			return (error);
 
 		/*
@@ -586,7 +584,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		if ((sc->sc_dk.dk_openmask & ~pmask) ||
 		    ((sc->sc_dk.dk_bopenmask & pmask) &&
 		    (sc->sc_dk.dk_copenmask & pmask))) {
-			rw_exit_write(&sc->sc_rwlock);
+			disk_unlock(&sc->sc_dk);
 			return (EBUSY);
 		}
 
@@ -601,7 +599,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 
 		/* Detach the disk. */
 		disk_detach(&sc->sc_dk);
-		rw_exit_write(&sc->sc_rwlock);
+		disk_unlock(&sc->sc_dk);
 		break;
 
 	case VNDIOCGET:
@@ -668,7 +666,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 		if ((flag & FWRITE) == 0)
 			return (EBADF);
 
-		if ((error = rw_enter(&sc->sc_rwlock, RW_WRITE|RW_INTR)) != 0)
+		if ((error = disk_lock(&sc->sc_dk)) != 0)
 			return (error);
 
 		error = setdisklabel(sc->sc_dk.dk_label,
@@ -679,7 +677,7 @@ vndioctl(dev_t dev, u_long cmd, caddr_t addr, int flag, struct proc *p)
 				    vndstrategy, sc->sc_dk.dk_label);
 		}
 
-		rw_exit_write(&sc->sc_rwlock);
+		disk_unlock(&sc->sc_dk);
 		return (error);
 
 	default: