ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Linux-NBD/NBD.xs
(Generate patch)

Comparing Linux-NBD/NBD.xs (file contents):
Revision 1.4 by root, Thu Aug 24 12:13:40 2006 UTC vs.
Revision 1.7 by root, Tue Sep 21 11:51:55 2010 UTC

… …
20#define ntohll(netlong) __bswap_64(netlong) 20#define ntohll(netlong) __bswap_64(netlong)
21#else 21#else
22error, you should not exist 22error, you should not exist
23#endif 23#endif
24 24
25struct rstate {
26 struct nbd_request req;
27 u32 req_read; /* how many octets of req are valid */
28 u32 data_read; /* how many octets of the data sv are valid */
29};
30
25MODULE = Linux::NBD PACKAGE = Linux::NBD::Client 31MODULE = Linux::NBD PACKAGE = Linux::NBD::Client
32
33PROTOTYPES: DISABLE
26 34
27void 35void
28_set_sock (int dev, int fd) 36_set_sock (int dev, int fd)
29 CODE: 37 CODE:
30 ioctl (dev, NBD_SET_SOCK, (unsigned long)fd); 38 ioctl (dev, NBD_SET_SOCK, (unsigned long)fd);
… …
63 ioctl (dev, NBD_SET_BLKSIZE, blocksize); 71 ioctl (dev, NBD_SET_BLKSIZE, blocksize);
64 72
65void 73void
66_set_size (int dev, unsigned long size) 74_set_size (int dev, unsigned long size)
67 CODE: 75 CODE:
68 ioctl (dev, NBD_SET_BLKSIZE, size); 76 ioctl (dev, NBD_SET_SIZE, size);
69 77
70void 78void
71_set_size_blocks (int dev, unsigned long nblocks) 79_set_size_blocks (int dev, unsigned long nblocks)
72 CODE: 80 CODE:
73 ioctl (dev, NBD_SET_SIZE_BLOCKS, nblocks); 81 ioctl (dev, NBD_SET_SIZE_BLOCKS, nblocks);
74 82
83void
84_set_timeout (int dev, unsigned long timeout)
85 CODE:
86 ioctl (dev, NBD_SET_TIMEOUT, timeout);
87
88void
89_print_debug (int dev)
90 CODE:
91 ioctl (dev, NBD_PRINT_DEBUG, 0);
92
75MODULE = Linux::NBD PACKAGE = Linux::NBD::Server 93MODULE = Linux::NBD PACKAGE = Linux::NBD::Server
76 94
77void 95void
78_one_request (SV *obj, int fd) 96_one_request (SV *obj, int fd)
79 CODE: 97 CODE:
80{ 98{
81 struct nbd_request req; 99 struct rstate *s;
100 u64 from;
101 u32 len;
102 char *method;
103 MAGIC *mg = mg_find (SvRV (obj), PERL_MAGIC_ext);
82 104
83 if (read (fd, &req, sizeof (req)) == sizeof (req)) 105 if (!mg)
84 { 106 {
107 mg = sv_magicext (SvRV (obj), 0, PERL_MAGIC_ext, 0, 0, 0);
108 mg->mg_len = sizeof (struct rstate);
109 New (0, mg->mg_ptr, mg->mg_len, char);
110
111 ((struct rstate *)mg->mg_ptr)->req_read = 0; /* initialise the state machine */
112 }
113
114 s = (struct rstate *)mg->mg_ptr;
115
116 if (s->req_read < sizeof (struct nbd_request))
117 {
118 int res = read (fd, s->req_read + (char *)&s->req, sizeof (struct nbd_request) - s->req_read);
119
120 if (res > 0)
121 s->req_read += res;
122 else if (res == 0)
123 XSRETURN_UNDEF; /* should req->eof */
124 else if (errno != EAGAIN && errno != EWOULDBLOCK)
125 XSRETURN_UNDEF; /* should req->error */
126
127 s->data_read = 0;
128
129 if (s->req_read < sizeof (struct nbd_request))
130 XSRETURN_NO;
131 }
132
133 /* now we have a full request, so check for data */
85 if (req.magic == htonl (NBD_REQUEST_MAGIC)) 134 if (s->req.magic != htonl (NBD_REQUEST_MAGIC))
135 croak ("Linux::NBD::Server received illegal request magic %08lx - protocol error.\n", ntohl (s->req.magic));
136
137 from = ntohll (s->req.from);
138 len = ntohl (s->req.len);
139
140 switch (ntohl (s->req.type))
86 { 141 {
87 req.type = htonl (req.type); 142 case NBD_CMD_WRITE:
88 143 if (!mg->mg_obj)
89 if (req.type < 2)
90 { 144 {
91 u64 from = ntohll (req.from); 145 mg->mg_flags |= MGf_REFCOUNTED;
146 mg->mg_obj = NEWSV (0, len);
147 SvPOK_only (mg->mg_obj);
148 SvCUR_set (mg->mg_obj, len);
149 }
92 150
93 PUSHMARK (SP); 151 if (s->data_read < len)
94 EXTEND (SP, 3); 152 {
95 PUSHs (obj); 153 int res = read (fd, s->data_read + SvPVX (mg->mg_obj), len - s->data_read);
96 PUSHs (sv_2mortal (newSVpvn (req.handle, sizeof (req.handle))));
97 PUSHs (sv_2mortal (sizeof (UV) < 8 && from > (0xffffffffUL)
98 ? newSVnv (from)
99 : newSVuv (from)));
100 PUSHs (sv_2mortal (newSVuv (ntohl (req.len))));
101 PUTBACK;
102 call_method (req.type ? "req_write" : "req_read", G_DISCARD);
103 SPAGAIN;
104 154
155 if (res > 0)
156 s->data_read += res;
157 else if (res == 0)
158 XSRETURN_UNDEF; /* should req->eof */
159 else if (errno != EAGAIN && errno != EWOULDBLOCK)
160 XSRETURN_UNDEF; /* should req->error */
161
162 if (s->data_read < len)
105 XSRETURN_NO; 163 XSRETURN_NO;
106 } 164 }
165
166 /* fallthrough */
167 case NBD_CMD_READ:
168 s->req_read = 0;
169
170 PUSHMARK (SP);
171 EXTEND (SP, 4);
172 PUSHs (obj);
173 PUSHs (sv_2mortal (newSVpvn (s->req.handle, sizeof (s->req.handle))));
174 PUSHs (sv_2mortal (sizeof (UV) < 8 && from > 0xffffffffUL
175 ? newSVnv (from)
176 : newSVuv (from)));
177
178 if (mg->mg_obj)
179 {
180 PUSHs (sv_2mortal (mg->mg_obj));
181 mg->mg_obj = 0;
182 method = "req_write";
107 } 183 }
184 else
185 {
186 PUSHs (sv_2mortal (newSVuv (len)));
187 method = "req_read";
188 }
189
190 break;
191
192 case NBD_CMD_DISC:
193 s->req_read = 0;
194
195 method = "req_disc";
196
197 PUSHMARK (SP);
198 XPUSHs (obj);
199
200 break;
201
202 default:
203 croak ("Linux::NBD::Server received unsupported request type %d.\n", ntohl (s->req.type));
108 } 204 }
205
206 PUTBACK;
207 call_method (method, G_DISCARD);
208 SPAGAIN;
109 209
110 XSRETURN_YES; 210 XSRETURN_YES;
111} 211}
112 212
113SV * 213SV *
114_format_reply (SV *handle, unsigned int error = 0, SV *data = 0) 214format_reply (SV *unused, SV *handle, unsigned int error = 0, SV *data = 0)
115 CODE: 215 CODE:
116{ 216{
117 struct nbd_reply rep; 217 struct nbd_reply rep;
118 STRLEN len; 218 STRLEN len;
119 char *h = SvPV (handle, len); 219 char *h = SvPV (handle, len);

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines