File: o_direct.diff

package info (click to toggle)
vblade 16-1
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 192 kB
  • ctags: 154
  • sloc: ansic: 941; makefile: 70; sh: 6
file content (135 lines) | stat: -rw-r--r-- 3,052 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
diff -uprN vblade-14rc1/aoe.c vblade-14rc2/aoe.c
--- vblade-14rc1/aoe.c	2006-11-20 12:48:05.000000000 -0500
+++ vblade-14rc2/aoe.c	2006-11-20 13:14:04.000000000 -0500
@@ -185,15 +185,42 @@ doaoe(Aoehdr *p)
 	}
 }
 
+// allocate the buffer so that the ata data area
+// is page aligned for o_direct on linux
+
+void *
+bufalloc(uchar **buf, long len)
+{
+	long psize;
+	unsigned long n;
+
+	psize = sysconf(_SC_PAGESIZE);
+	if (psize == -1) {
+		perror("sysconf");
+		exit(EXIT_FAILURE);
+	}
+	n = len/psize + 3;
+	*buf = malloc(psize * n);
+	if (!*buf) {
+		perror("malloc");
+		exit(EXIT_FAILURE);
+	}
+	n = (unsigned long) *buf;
+	n += psize * 2;
+	n &= ~(psize - 1);
+	return (void *) (n - sizeof (Ata));
+}
+
 void
 aoe(void)
 {
 	Aoehdr *p;
+	uchar *freeme;
 	uchar *buf;
 	int n, sh;
 	enum { bufsz = 1<<16, };
 
-	buf = malloc(bufsz);
+	buf = bufalloc(&freeme, bufsz);
 	aoead(sfd);
 
 	for (;;) {
@@ -218,7 +245,7 @@ aoe(void)
 			continue;
 		doaoe(p);
 	}
-	free(buf);
+	free(freeme);
 }
 
 void
@@ -310,7 +337,7 @@ main(int argc, char **argv)
 	}
 	if (s.st_mode & (S_IWUSR|S_IWGRP|S_IWOTH))
 		omode = O_RDWR;
-	bfd = open(argv[3], omode);
+	bfd = opendisk(argv[3], omode);
 	if (bfd == -1) {
 		perror("open");
 		exit(1);
diff -uprN vblade-14rc1/fns.h vblade-14rc2/fns.h
--- vblade-14rc1/fns.h	2006-11-13 16:12:31.000000000 -0500
+++ vblade-14rc2/fns.h	2006-11-20 13:13:44.000000000 -0500
@@ -21,6 +21,7 @@ int	atacmd(Ataregs *, uchar *, int);
 
 int	dial(char *);
 int	getea(int, char *, uchar *);
+int	opendisk(const char *, int);
 int	putsec(int, uchar *, vlong, int);
 int	getsec(int, uchar *, vlong, int);
 int	putpkt(int, uchar *, int);
diff -uprN vblade-14rc1/freebsd.c vblade-14rc2/freebsd.c
--- vblade-14rc1/freebsd.c	2006-11-13 13:57:34.000000000 -0500
+++ vblade-14rc2/freebsd.c	2006-11-20 13:13:44.000000000 -0500
@@ -241,6 +241,11 @@ getea(int s, char *eth, uchar *ea)
 	return(0);
 }
 
+int
+opendisk(const char *disk, int omode)
+{
+	return open(disk, omode);
+}
 
 int
 getsec(int fd, uchar *place, vlong lba, int nsec)
diff -uprN vblade-14rc1/linux.c vblade-14rc2/linux.c
--- vblade-14rc1/linux.c	2006-11-13 13:57:34.000000000 -0500
+++ vblade-14rc2/linux.c	2006-11-20 13:13:44.000000000 -0500
@@ -1,5 +1,6 @@
 // linux.c: low level access routines for Linux
 #include "config.h"
+#define _GNU_SOURCE
 #include <sys/socket.h>
 #include <stdio.h>
 #include <string.h>
@@ -22,6 +23,7 @@
 #include <netinet/in.h>
 #include <linux/fs.h>
 #include <sys/stat.h>
+#include <fcntl.h>
 
 #include "dat.h"
 #include "fns.h"
@@ -102,17 +104,21 @@ getmtu(int s, char *name)
 }
 
 int
+opendisk(const char *disk, int omode)
+{
+	return open(disk, omode|O_DIRECT);
+}
+
+int
 getsec(int fd, uchar *place, vlong lba, int nsec)
 {
-	lseek(fd, lba * 512, 0);
-	return read(fd, place, nsec * 512);
+	return pread(fd, place, nsec * 512, lba * 512);
 }
 
 int
 putsec(int fd, uchar *place, vlong lba, int nsec)
 {
-	lseek(fd, lba * 512, 0);
-	return write(fd, place, nsec * 512);
+	return pwrite(fd, place, nsec * 512, lba * 512);
 }
 
 int