1*22028508SToomas Soome /* $NetBSD: tftp.c,v 1.4 1997/09/17 16:57:07 drochner Exp $ */
2*22028508SToomas Soome
3*22028508SToomas Soome /*
4*22028508SToomas Soome * Copyright (c) 1996
5*22028508SToomas Soome * Matthias Drochner. All rights reserved.
6*22028508SToomas Soome *
7*22028508SToomas Soome * Redistribution and use in source and binary forms, with or without
8*22028508SToomas Soome * modification, are permitted provided that the following conditions
9*22028508SToomas Soome * are met:
10*22028508SToomas Soome * 1. Redistributions of source code must retain the above copyright
11*22028508SToomas Soome * notice, this list of conditions and the following disclaimer.
12*22028508SToomas Soome * 2. Redistributions in binary form must reproduce the above copyright
13*22028508SToomas Soome * notice, this list of conditions and the following disclaimer in the
14*22028508SToomas Soome * documentation and/or other materials provided with the distribution.
15*22028508SToomas Soome * 3. All advertising materials mentioning features or use of this software
16*22028508SToomas Soome * must display the following acknowledgement:
17*22028508SToomas Soome * This product includes software developed for the NetBSD Project
18*22028508SToomas Soome * by Matthias Drochner.
19*22028508SToomas Soome * 4. The name of the author may not be used to endorse or promote products
20*22028508SToomas Soome * derived from this software without specific prior written permission.
21*22028508SToomas Soome *
22*22028508SToomas Soome * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
23*22028508SToomas Soome * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
24*22028508SToomas Soome * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
25*22028508SToomas Soome * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
26*22028508SToomas Soome * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
27*22028508SToomas Soome * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28*22028508SToomas Soome * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29*22028508SToomas Soome * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30*22028508SToomas Soome * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
31*22028508SToomas Soome * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32*22028508SToomas Soome */
33*22028508SToomas Soome
34*22028508SToomas Soome #include <sys/cdefs.h>
35*22028508SToomas Soome
36*22028508SToomas Soome /*
37*22028508SToomas Soome * Simple TFTP implementation for libsa.
38*22028508SToomas Soome * Assumes:
39*22028508SToomas Soome * - socket descriptor (int) at dev->d_opendata, dev stored at
40*22028508SToomas Soome * open_file->f_devdata
41*22028508SToomas Soome * - server host IP in global rootip
42*22028508SToomas Soome * Restrictions:
43*22028508SToomas Soome * - read only
44*22028508SToomas Soome * - lseek only with SEEK_SET or SEEK_CUR
45*22028508SToomas Soome * - no big time differences between transfers (<tftp timeout)
46*22028508SToomas Soome */
47*22028508SToomas Soome
48*22028508SToomas Soome #include <sys/types.h>
49*22028508SToomas Soome #include <sys/stat.h>
50*22028508SToomas Soome #include <netinet/in.h>
51*22028508SToomas Soome #include <netinet/udp.h>
52*22028508SToomas Soome #include <netinet/in_systm.h>
53*22028508SToomas Soome #include <arpa/tftp.h>
54*22028508SToomas Soome
55*22028508SToomas Soome #include <string.h>
56*22028508SToomas Soome
57*22028508SToomas Soome #include "stand.h"
58*22028508SToomas Soome #include "net.h"
59*22028508SToomas Soome #include "netif.h"
60*22028508SToomas Soome
61*22028508SToomas Soome #include "tftp.h"
62*22028508SToomas Soome
63*22028508SToomas Soome struct tftp_handle;
64*22028508SToomas Soome struct tftprecv_extra;
65*22028508SToomas Soome
66*22028508SToomas Soome static ssize_t recvtftp(struct iodesc *, void **, void **, time_t, void *);
67*22028508SToomas Soome static int tftp_open(const char *, struct open_file *);
68*22028508SToomas Soome static int tftp_close(struct open_file *);
69*22028508SToomas Soome static int tftp_parse_oack(struct tftp_handle *, char *, size_t);
70*22028508SToomas Soome static int tftp_read(struct open_file *, void *, size_t, size_t *);
71*22028508SToomas Soome static off_t tftp_seek(struct open_file *, off_t, int);
72*22028508SToomas Soome static int tftp_set_blksize(struct tftp_handle *, const char *);
73*22028508SToomas Soome static int tftp_stat(struct open_file *, struct stat *);
74*22028508SToomas Soome
75*22028508SToomas Soome struct fs_ops tftp_fsops = {
76*22028508SToomas Soome .fs_name = "tftp",
77*22028508SToomas Soome .fo_open = tftp_open,
78*22028508SToomas Soome .fo_close = tftp_close,
79*22028508SToomas Soome .fo_read = tftp_read,
80*22028508SToomas Soome .fo_write = null_write,
81*22028508SToomas Soome .fo_seek = tftp_seek,
82*22028508SToomas Soome .fo_stat = tftp_stat,
83*22028508SToomas Soome .fo_readdir = null_readdir
84*22028508SToomas Soome };
85*22028508SToomas Soome
86*22028508SToomas Soome static int tftpport = 2000;
87*22028508SToomas Soome static int is_open = 0;
88*22028508SToomas Soome
89*22028508SToomas Soome /*
90*22028508SToomas Soome * The legacy TFTP_BLKSIZE value was SEGSIZE(512).
91*22028508SToomas Soome * TFTP_REQUESTED_BLKSIZE of 1428 is (Ethernet MTU, less the TFTP, UDP and
92*22028508SToomas Soome * IP header lengths).
93*22028508SToomas Soome */
94*22028508SToomas Soome #define TFTP_REQUESTED_BLKSIZE 1428
95*22028508SToomas Soome
96*22028508SToomas Soome /*
97*22028508SToomas Soome * Choose a blksize big enough so we can test with Ethernet
98*22028508SToomas Soome * Jumbo frames in the future.
99*22028508SToomas Soome */
100*22028508SToomas Soome #define TFTP_MAX_BLKSIZE 9008
101*22028508SToomas Soome
102*22028508SToomas Soome struct tftp_handle {
103*22028508SToomas Soome struct iodesc *iodesc;
104*22028508SToomas Soome int currblock; /* contents of lastdata */
105*22028508SToomas Soome int islastblock; /* flag */
106*22028508SToomas Soome int validsize;
107*22028508SToomas Soome int off;
108*22028508SToomas Soome char *path; /* saved for re-requests */
109*22028508SToomas Soome unsigned int tftp_blksize;
110*22028508SToomas Soome unsigned long tftp_tsize;
111*22028508SToomas Soome void *pkt;
112*22028508SToomas Soome struct tftphdr *tftp_hdr;
113*22028508SToomas Soome };
114*22028508SToomas Soome
115*22028508SToomas Soome struct tftprecv_extra {
116*22028508SToomas Soome struct tftp_handle *tftp_handle;
117*22028508SToomas Soome unsigned short rtype; /* Received type */
118*22028508SToomas Soome };
119*22028508SToomas Soome
120*22028508SToomas Soome #define TFTP_MAX_ERRCODE EOPTNEG
121*22028508SToomas Soome static const int tftperrors[TFTP_MAX_ERRCODE + 1] = {
122*22028508SToomas Soome 0, /* ??? */
123*22028508SToomas Soome ENOENT,
124*22028508SToomas Soome EPERM,
125*22028508SToomas Soome ENOSPC,
126*22028508SToomas Soome EINVAL, /* ??? */
127*22028508SToomas Soome EINVAL, /* ??? */
128*22028508SToomas Soome EEXIST,
129*22028508SToomas Soome EINVAL, /* ??? */
130*22028508SToomas Soome EINVAL, /* Option negotiation failed. */
131*22028508SToomas Soome };
132*22028508SToomas Soome
133*22028508SToomas Soome static int tftp_getnextblock(struct tftp_handle *h);
134*22028508SToomas Soome
135*22028508SToomas Soome /* send error message back. */
136*22028508SToomas Soome static void
tftp_senderr(struct tftp_handle * h,ushort_t errcode,const char * msg)137*22028508SToomas Soome tftp_senderr(struct tftp_handle *h, ushort_t errcode, const char *msg)
138*22028508SToomas Soome {
139*22028508SToomas Soome struct {
140*22028508SToomas Soome uchar_t header[HEADER_SIZE];
141*22028508SToomas Soome struct tftphdr t;
142*22028508SToomas Soome uchar_t space[63]; /* +1 from t */
143*22028508SToomas Soome } __packed __aligned(4) wbuf;
144*22028508SToomas Soome char *wtail;
145*22028508SToomas Soome int len;
146*22028508SToomas Soome
147*22028508SToomas Soome len = strlen(msg);
148*22028508SToomas Soome if (len > sizeof (wbuf.space))
149*22028508SToomas Soome len = sizeof (wbuf.space);
150*22028508SToomas Soome
151*22028508SToomas Soome wbuf.t.th_opcode = htons((ushort_t)ERROR);
152*22028508SToomas Soome wbuf.t.th_code = htons(errcode);
153*22028508SToomas Soome
154*22028508SToomas Soome wtail = wbuf.t.th_msg;
155*22028508SToomas Soome bcopy(msg, wtail, len);
156*22028508SToomas Soome wtail[len] = '\0';
157*22028508SToomas Soome wtail += len + 1;
158*22028508SToomas Soome
159*22028508SToomas Soome sendudp(h->iodesc, &wbuf.t, wtail - (char *)&wbuf.t);
160*22028508SToomas Soome }
161*22028508SToomas Soome
162*22028508SToomas Soome static void
tftp_sendack(struct tftp_handle * h,ushort_t block)163*22028508SToomas Soome tftp_sendack(struct tftp_handle *h, ushort_t block)
164*22028508SToomas Soome {
165*22028508SToomas Soome struct {
166*22028508SToomas Soome uchar_t header[HEADER_SIZE];
167*22028508SToomas Soome struct tftphdr t;
168*22028508SToomas Soome } __packed __aligned(4) wbuf;
169*22028508SToomas Soome char *wtail;
170*22028508SToomas Soome
171*22028508SToomas Soome wbuf.t.th_opcode = htons((ushort_t)ACK);
172*22028508SToomas Soome wtail = (char *)&wbuf.t.th_block;
173*22028508SToomas Soome wbuf.t.th_block = htons(block);
174*22028508SToomas Soome wtail += 2;
175*22028508SToomas Soome
176*22028508SToomas Soome sendudp(h->iodesc, &wbuf.t, wtail - (char *)&wbuf.t);
177*22028508SToomas Soome }
178*22028508SToomas Soome
179*22028508SToomas Soome static ssize_t
recvtftp(struct iodesc * d,void ** pkt,void ** payload,time_t tleft,void * recv_extra)180*22028508SToomas Soome recvtftp(struct iodesc *d, void **pkt, void **payload, time_t tleft,
181*22028508SToomas Soome void *recv_extra)
182*22028508SToomas Soome {
183*22028508SToomas Soome struct tftprecv_extra *extra;
184*22028508SToomas Soome struct tftp_handle *h;
185*22028508SToomas Soome struct tftphdr *t;
186*22028508SToomas Soome void *ptr = NULL;
187*22028508SToomas Soome ssize_t len;
188*22028508SToomas Soome
189*22028508SToomas Soome errno = 0;
190*22028508SToomas Soome extra = recv_extra;
191*22028508SToomas Soome h = extra->tftp_handle;
192*22028508SToomas Soome
193*22028508SToomas Soome len = readudp(d, &ptr, (void **)&t, tleft);
194*22028508SToomas Soome
195*22028508SToomas Soome if (len < 4) {
196*22028508SToomas Soome free(ptr);
197*22028508SToomas Soome return (-1);
198*22028508SToomas Soome }
199*22028508SToomas Soome
200*22028508SToomas Soome extra->rtype = ntohs(t->th_opcode);
201*22028508SToomas Soome switch (ntohs(t->th_opcode)) {
202*22028508SToomas Soome case DATA: {
203*22028508SToomas Soome int got;
204*22028508SToomas Soome
205*22028508SToomas Soome if (htons(t->th_block) < (ushort_t)d->xid) {
206*22028508SToomas Soome /*
207*22028508SToomas Soome * Apparently our ACK was missed, re-send.
208*22028508SToomas Soome */
209*22028508SToomas Soome tftp_sendack(h, htons(t->th_block));
210*22028508SToomas Soome free(ptr);
211*22028508SToomas Soome return (-1);
212*22028508SToomas Soome }
213*22028508SToomas Soome if (htons(t->th_block) != (ushort_t)d->xid) {
214*22028508SToomas Soome /*
215*22028508SToomas Soome * Packet from the future, drop this.
216*22028508SToomas Soome */
217*22028508SToomas Soome free(ptr);
218*22028508SToomas Soome return (-1);
219*22028508SToomas Soome }
220*22028508SToomas Soome if (d->xid == 1) {
221*22028508SToomas Soome /*
222*22028508SToomas Soome * First data packet from new port.
223*22028508SToomas Soome */
224*22028508SToomas Soome struct udphdr *uh;
225*22028508SToomas Soome uh = (struct udphdr *)t - 1;
226*22028508SToomas Soome d->destport = uh->uh_sport;
227*22028508SToomas Soome }
228*22028508SToomas Soome got = len - (t->th_data - (char *)t);
229*22028508SToomas Soome *pkt = ptr;
230*22028508SToomas Soome *payload = t;
231*22028508SToomas Soome return (got);
232*22028508SToomas Soome }
233*22028508SToomas Soome case ERROR:
234*22028508SToomas Soome if ((unsigned)ntohs(t->th_code) > TFTP_MAX_ERRCODE) {
235*22028508SToomas Soome printf("illegal tftp error %d\n", ntohs(t->th_code));
236*22028508SToomas Soome errno = EIO;
237*22028508SToomas Soome } else {
238*22028508SToomas Soome #ifdef TFTP_DEBUG
239*22028508SToomas Soome printf("tftp-error %d\n", ntohs(t->th_code));
240*22028508SToomas Soome #endif
241*22028508SToomas Soome errno = tftperrors[ntohs(t->th_code)];
242*22028508SToomas Soome }
243*22028508SToomas Soome free(ptr);
244*22028508SToomas Soome return (-1);
245*22028508SToomas Soome case OACK: {
246*22028508SToomas Soome struct udphdr *uh;
247*22028508SToomas Soome int tftp_oack_len;
248*22028508SToomas Soome
249*22028508SToomas Soome /*
250*22028508SToomas Soome * Unexpected OACK. TFTP transfer already in progress.
251*22028508SToomas Soome * Drop the pkt.
252*22028508SToomas Soome */
253*22028508SToomas Soome if (d->xid != 1) {
254*22028508SToomas Soome free(ptr);
255*22028508SToomas Soome return (-1);
256*22028508SToomas Soome }
257*22028508SToomas Soome
258*22028508SToomas Soome /*
259*22028508SToomas Soome * Remember which port this OACK came from, because we need
260*22028508SToomas Soome * to send the ACK or errors back to it.
261*22028508SToomas Soome */
262*22028508SToomas Soome uh = (struct udphdr *)t - 1;
263*22028508SToomas Soome d->destport = uh->uh_sport;
264*22028508SToomas Soome
265*22028508SToomas Soome /* Parse options ACK-ed by the server. */
266*22028508SToomas Soome tftp_oack_len = len - sizeof (t->th_opcode);
267*22028508SToomas Soome if (tftp_parse_oack(h, t->th_u.tu_stuff, tftp_oack_len) != 0) {
268*22028508SToomas Soome tftp_senderr(h, EOPTNEG, "Malformed OACK");
269*22028508SToomas Soome errno = EIO;
270*22028508SToomas Soome free(ptr);
271*22028508SToomas Soome return (-1);
272*22028508SToomas Soome }
273*22028508SToomas Soome *pkt = ptr;
274*22028508SToomas Soome *payload = t;
275*22028508SToomas Soome return (0);
276*22028508SToomas Soome }
277*22028508SToomas Soome default:
278*22028508SToomas Soome #ifdef TFTP_DEBUG
279*22028508SToomas Soome printf("tftp type %d not handled\n", ntohs(t->th_opcode));
280*22028508SToomas Soome #endif
281*22028508SToomas Soome free(ptr);
282*22028508SToomas Soome return (-1);
283*22028508SToomas Soome }
284*22028508SToomas Soome }
285*22028508SToomas Soome
286*22028508SToomas Soome /* send request, expect first block (or error) */
287*22028508SToomas Soome static int
tftp_makereq(struct tftp_handle * h)288*22028508SToomas Soome tftp_makereq(struct tftp_handle *h)
289*22028508SToomas Soome {
290*22028508SToomas Soome struct {
291*22028508SToomas Soome uchar_t header[HEADER_SIZE];
292*22028508SToomas Soome struct tftphdr t;
293*22028508SToomas Soome uchar_t space[FNAME_SIZE + 6];
294*22028508SToomas Soome } __packed __aligned(4) wbuf;
295*22028508SToomas Soome struct tftprecv_extra recv_extra;
296*22028508SToomas Soome char *wtail;
297*22028508SToomas Soome int l;
298*22028508SToomas Soome ssize_t res;
299*22028508SToomas Soome void *pkt;
300*22028508SToomas Soome struct tftphdr *t;
301*22028508SToomas Soome char *tftp_blksize = NULL;
302*22028508SToomas Soome int blksize_l;
303*22028508SToomas Soome
304*22028508SToomas Soome /*
305*22028508SToomas Soome * Allow overriding default TFTP block size by setting
306*22028508SToomas Soome * a tftp.blksize environment variable.
307*22028508SToomas Soome */
308*22028508SToomas Soome if ((tftp_blksize = getenv("tftp.blksize")) != NULL) {
309*22028508SToomas Soome tftp_set_blksize(h, tftp_blksize);
310*22028508SToomas Soome }
311*22028508SToomas Soome
312*22028508SToomas Soome wbuf.t.th_opcode = htons((ushort_t)RRQ);
313*22028508SToomas Soome wtail = wbuf.t.th_stuff;
314*22028508SToomas Soome l = strlen(h->path);
315*22028508SToomas Soome #ifdef TFTP_PREPEND_PATH
316*22028508SToomas Soome if (l > FNAME_SIZE - (sizeof (TFTP_PREPEND_PATH) - 1))
317*22028508SToomas Soome return (ENAMETOOLONG);
318*22028508SToomas Soome bcopy(TFTP_PREPEND_PATH, wtail, sizeof (TFTP_PREPEND_PATH) - 1);
319*22028508SToomas Soome wtail += sizeof (TFTP_PREPEND_PATH) - 1;
320*22028508SToomas Soome #else
321*22028508SToomas Soome if (l > FNAME_SIZE)
322*22028508SToomas Soome return (ENAMETOOLONG);
323*22028508SToomas Soome #endif
324*22028508SToomas Soome bcopy(h->path, wtail, l + 1);
325*22028508SToomas Soome wtail += l + 1;
326*22028508SToomas Soome bcopy("octet", wtail, 6);
327*22028508SToomas Soome wtail += 6;
328*22028508SToomas Soome bcopy("blksize", wtail, 8);
329*22028508SToomas Soome wtail += 8;
330*22028508SToomas Soome blksize_l = sprintf(wtail, "%d", h->tftp_blksize);
331*22028508SToomas Soome wtail += blksize_l + 1;
332*22028508SToomas Soome bcopy("tsize", wtail, 6);
333*22028508SToomas Soome wtail += 6;
334*22028508SToomas Soome bcopy("0", wtail, 2);
335*22028508SToomas Soome wtail += 2;
336*22028508SToomas Soome
337*22028508SToomas Soome h->iodesc->myport = htons(tftpport + (getsecs() & 0x3ff));
338*22028508SToomas Soome h->iodesc->destport = htons(IPPORT_TFTP);
339*22028508SToomas Soome h->iodesc->xid = 1; /* expected block */
340*22028508SToomas Soome
341*22028508SToomas Soome h->currblock = 0;
342*22028508SToomas Soome h->islastblock = 0;
343*22028508SToomas Soome h->validsize = 0;
344*22028508SToomas Soome
345*22028508SToomas Soome pkt = NULL;
346*22028508SToomas Soome recv_extra.tftp_handle = h;
347*22028508SToomas Soome res = sendrecv(h->iodesc, &sendudp, &wbuf.t, wtail - (char *)&wbuf.t,
348*22028508SToomas Soome &recvtftp, &pkt, (void **)&t, &recv_extra);
349*22028508SToomas Soome if (res == -1) {
350*22028508SToomas Soome free(pkt);
351*22028508SToomas Soome return (errno);
352*22028508SToomas Soome }
353*22028508SToomas Soome
354*22028508SToomas Soome free(h->pkt);
355*22028508SToomas Soome h->pkt = pkt;
356*22028508SToomas Soome h->tftp_hdr = t;
357*22028508SToomas Soome
358*22028508SToomas Soome if (recv_extra.rtype == OACK)
359*22028508SToomas Soome return (tftp_getnextblock(h));
360*22028508SToomas Soome
361*22028508SToomas Soome /* Server ignored our blksize request, revert to TFTP default. */
362*22028508SToomas Soome h->tftp_blksize = SEGSIZE;
363*22028508SToomas Soome
364*22028508SToomas Soome switch (recv_extra.rtype) {
365*22028508SToomas Soome case DATA: {
366*22028508SToomas Soome h->currblock = 1;
367*22028508SToomas Soome h->validsize = res;
368*22028508SToomas Soome h->islastblock = 0;
369*22028508SToomas Soome if (res < h->tftp_blksize) {
370*22028508SToomas Soome h->islastblock = 1; /* very short file */
371*22028508SToomas Soome tftp_sendack(h, h->currblock);
372*22028508SToomas Soome }
373*22028508SToomas Soome return (0);
374*22028508SToomas Soome }
375*22028508SToomas Soome case ERROR:
376*22028508SToomas Soome default:
377*22028508SToomas Soome return (errno);
378*22028508SToomas Soome }
379*22028508SToomas Soome
380*22028508SToomas Soome }
381*22028508SToomas Soome
382*22028508SToomas Soome /* ack block, expect next */
383*22028508SToomas Soome static int
tftp_getnextblock(struct tftp_handle * h)384*22028508SToomas Soome tftp_getnextblock(struct tftp_handle *h)
385*22028508SToomas Soome {
386*22028508SToomas Soome struct {
387*22028508SToomas Soome uchar_t header[HEADER_SIZE];
388*22028508SToomas Soome struct tftphdr t;
389*22028508SToomas Soome } __packed __aligned(4) wbuf;
390*22028508SToomas Soome struct tftprecv_extra recv_extra;
391*22028508SToomas Soome char *wtail;
392*22028508SToomas Soome int res;
393*22028508SToomas Soome void *pkt;
394*22028508SToomas Soome struct tftphdr *t;
395*22028508SToomas Soome
396*22028508SToomas Soome wbuf.t.th_opcode = htons((ushort_t)ACK);
397*22028508SToomas Soome wtail = (char *)&wbuf.t.th_block;
398*22028508SToomas Soome wbuf.t.th_block = htons((ushort_t)h->currblock);
399*22028508SToomas Soome wtail += 2;
400*22028508SToomas Soome
401*22028508SToomas Soome h->iodesc->xid = h->currblock + 1; /* expected block */
402*22028508SToomas Soome
403*22028508SToomas Soome pkt = NULL;
404*22028508SToomas Soome recv_extra.tftp_handle = h;
405*22028508SToomas Soome res = sendrecv(h->iodesc, &sendudp, &wbuf.t, wtail - (char *)&wbuf.t,
406*22028508SToomas Soome &recvtftp, &pkt, (void **)&t, &recv_extra);
407*22028508SToomas Soome
408*22028508SToomas Soome if (res == -1) { /* 0 is OK! */
409*22028508SToomas Soome free(pkt);
410*22028508SToomas Soome return (errno);
411*22028508SToomas Soome }
412*22028508SToomas Soome
413*22028508SToomas Soome free(h->pkt);
414*22028508SToomas Soome h->pkt = pkt;
415*22028508SToomas Soome h->tftp_hdr = t;
416*22028508SToomas Soome h->currblock++;
417*22028508SToomas Soome h->validsize = res;
418*22028508SToomas Soome if (res < h->tftp_blksize)
419*22028508SToomas Soome h->islastblock = 1; /* EOF */
420*22028508SToomas Soome
421*22028508SToomas Soome if (h->islastblock == 1) {
422*22028508SToomas Soome /* Send an ACK for the last block */
423*22028508SToomas Soome wbuf.t.th_block = htons((ushort_t)h->currblock);
424*22028508SToomas Soome sendudp(h->iodesc, &wbuf.t, wtail - (char *)&wbuf.t);
425*22028508SToomas Soome }
426*22028508SToomas Soome
427*22028508SToomas Soome return (0);
428*22028508SToomas Soome }
429*22028508SToomas Soome
430*22028508SToomas Soome static int
tftp_open(const char * path,struct open_file * f)431*22028508SToomas Soome tftp_open(const char *path, struct open_file *f)
432*22028508SToomas Soome {
433*22028508SToomas Soome struct devdesc *dev;
434*22028508SToomas Soome struct tftp_handle *tftpfile;
435*22028508SToomas Soome struct iodesc *io;
436*22028508SToomas Soome int res;
437*22028508SToomas Soome size_t pathsize;
438*22028508SToomas Soome const char *extraslash;
439*22028508SToomas Soome
440*22028508SToomas Soome if (netproto != NET_TFTP)
441*22028508SToomas Soome return (EINVAL);
442*22028508SToomas Soome
443*22028508SToomas Soome if (f->f_dev->dv_type != DEVT_NET)
444*22028508SToomas Soome return (EINVAL);
445*22028508SToomas Soome
446*22028508SToomas Soome if (is_open)
447*22028508SToomas Soome return (EBUSY);
448*22028508SToomas Soome
449*22028508SToomas Soome tftpfile = calloc(1, sizeof (*tftpfile));
450*22028508SToomas Soome if (!tftpfile)
451*22028508SToomas Soome return (ENOMEM);
452*22028508SToomas Soome
453*22028508SToomas Soome tftpfile->tftp_blksize = TFTP_REQUESTED_BLKSIZE;
454*22028508SToomas Soome dev = f->f_devdata;
455*22028508SToomas Soome tftpfile->iodesc = io = socktodesc(*(int *)(dev->d_opendata));
456*22028508SToomas Soome if (io == NULL) {
457*22028508SToomas Soome free(tftpfile);
458*22028508SToomas Soome return (EINVAL);
459*22028508SToomas Soome }
460*22028508SToomas Soome
461*22028508SToomas Soome io->destip = rootip;
462*22028508SToomas Soome tftpfile->off = 0;
463*22028508SToomas Soome pathsize = (strlen(rootpath) + 1 + strlen(path) + 1) * sizeof (char);
464*22028508SToomas Soome tftpfile->path = malloc(pathsize);
465*22028508SToomas Soome if (tftpfile->path == NULL) {
466*22028508SToomas Soome free(tftpfile);
467*22028508SToomas Soome return (ENOMEM);
468*22028508SToomas Soome }
469*22028508SToomas Soome if (rootpath[strlen(rootpath) - 1] == '/' || path[0] == '/')
470*22028508SToomas Soome extraslash = "";
471*22028508SToomas Soome else
472*22028508SToomas Soome extraslash = "/";
473*22028508SToomas Soome res = snprintf(tftpfile->path, pathsize, "%s%s%s",
474*22028508SToomas Soome rootpath, extraslash, path);
475*22028508SToomas Soome if (res < 0 || res > pathsize) {
476*22028508SToomas Soome free(tftpfile->path);
477*22028508SToomas Soome free(tftpfile);
478*22028508SToomas Soome return (ENOMEM);
479*22028508SToomas Soome }
480*22028508SToomas Soome
481*22028508SToomas Soome res = tftp_makereq(tftpfile);
482*22028508SToomas Soome
483*22028508SToomas Soome if (res) {
484*22028508SToomas Soome free(tftpfile->path);
485*22028508SToomas Soome free(tftpfile->pkt);
486*22028508SToomas Soome free(tftpfile);
487*22028508SToomas Soome return (res);
488*22028508SToomas Soome }
489*22028508SToomas Soome f->f_fsdata = tftpfile;
490*22028508SToomas Soome is_open = 1;
491*22028508SToomas Soome return (0);
492*22028508SToomas Soome }
493*22028508SToomas Soome
494*22028508SToomas Soome static int
tftp_read(struct open_file * f,void * addr,size_t size,size_t * resid)495*22028508SToomas Soome tftp_read(struct open_file *f, void *addr, size_t size,
496*22028508SToomas Soome size_t *resid /* out */)
497*22028508SToomas Soome {
498*22028508SToomas Soome struct tftp_handle *tftpfile;
499*22028508SToomas Soome size_t res;
500*22028508SToomas Soome int rc;
501*22028508SToomas Soome
502*22028508SToomas Soome rc = 0;
503*22028508SToomas Soome res = size;
504*22028508SToomas Soome tftpfile = f->f_fsdata;
505*22028508SToomas Soome
506*22028508SToomas Soome /* Make sure we will not read past file end */
507*22028508SToomas Soome if (tftpfile->tftp_tsize > 0 &&
508*22028508SToomas Soome tftpfile->off + size > tftpfile->tftp_tsize) {
509*22028508SToomas Soome size = tftpfile->tftp_tsize - tftpfile->off;
510*22028508SToomas Soome }
511*22028508SToomas Soome
512*22028508SToomas Soome while (size > 0) {
513*22028508SToomas Soome int needblock, count;
514*22028508SToomas Soome
515*22028508SToomas Soome twiddle(32);
516*22028508SToomas Soome
517*22028508SToomas Soome needblock = tftpfile->off / tftpfile->tftp_blksize + 1;
518*22028508SToomas Soome
519*22028508SToomas Soome if (tftpfile->currblock > needblock) { /* seek backwards */
520*22028508SToomas Soome tftp_senderr(tftpfile, 0, "No error: read aborted");
521*22028508SToomas Soome rc = tftp_makereq(tftpfile);
522*22028508SToomas Soome if (rc != 0)
523*22028508SToomas Soome break;
524*22028508SToomas Soome }
525*22028508SToomas Soome
526*22028508SToomas Soome while (tftpfile->currblock < needblock) {
527*22028508SToomas Soome
528*22028508SToomas Soome rc = tftp_getnextblock(tftpfile);
529*22028508SToomas Soome if (rc) { /* no answer */
530*22028508SToomas Soome #ifdef TFTP_DEBUG
531*22028508SToomas Soome printf("tftp: read error\n");
532*22028508SToomas Soome #endif
533*22028508SToomas Soome return (rc);
534*22028508SToomas Soome }
535*22028508SToomas Soome if (tftpfile->islastblock)
536*22028508SToomas Soome break;
537*22028508SToomas Soome }
538*22028508SToomas Soome
539*22028508SToomas Soome if (tftpfile->currblock == needblock) {
540*22028508SToomas Soome int offinblock, inbuffer;
541*22028508SToomas Soome
542*22028508SToomas Soome offinblock = tftpfile->off % tftpfile->tftp_blksize;
543*22028508SToomas Soome
544*22028508SToomas Soome inbuffer = tftpfile->validsize - offinblock;
545*22028508SToomas Soome if (inbuffer < 0) {
546*22028508SToomas Soome #ifdef TFTP_DEBUG
547*22028508SToomas Soome printf("tftp: invalid offset %d\n",
548*22028508SToomas Soome tftpfile->off);
549*22028508SToomas Soome #endif
550*22028508SToomas Soome return (EINVAL);
551*22028508SToomas Soome }
552*22028508SToomas Soome count = (size < inbuffer ? size : inbuffer);
553*22028508SToomas Soome bcopy(tftpfile->tftp_hdr->th_data + offinblock,
554*22028508SToomas Soome addr, count);
555*22028508SToomas Soome
556*22028508SToomas Soome addr = (char *)addr + count;
557*22028508SToomas Soome tftpfile->off += count;
558*22028508SToomas Soome size -= count;
559*22028508SToomas Soome res -= count;
560*22028508SToomas Soome
561*22028508SToomas Soome if ((tftpfile->islastblock) && (count == inbuffer))
562*22028508SToomas Soome break; /* EOF */
563*22028508SToomas Soome } else {
564*22028508SToomas Soome #ifdef TFTP_DEBUG
565*22028508SToomas Soome printf("tftp: block %d not found\n", needblock);
566*22028508SToomas Soome #endif
567*22028508SToomas Soome return (EINVAL);
568*22028508SToomas Soome }
569*22028508SToomas Soome
570*22028508SToomas Soome }
571*22028508SToomas Soome
572*22028508SToomas Soome if (resid != NULL)
573*22028508SToomas Soome *resid = res;
574*22028508SToomas Soome return (rc);
575*22028508SToomas Soome }
576*22028508SToomas Soome
577*22028508SToomas Soome static int
tftp_close(struct open_file * f)578*22028508SToomas Soome tftp_close(struct open_file *f)
579*22028508SToomas Soome {
580*22028508SToomas Soome struct tftp_handle *tftpfile;
581*22028508SToomas Soome tftpfile = f->f_fsdata;
582*22028508SToomas Soome
583*22028508SToomas Soome /* let it time out ... */
584*22028508SToomas Soome
585*22028508SToomas Soome if (tftpfile) {
586*22028508SToomas Soome free(tftpfile->path);
587*22028508SToomas Soome free(tftpfile->pkt);
588*22028508SToomas Soome free(tftpfile);
589*22028508SToomas Soome }
590*22028508SToomas Soome is_open = 0;
591*22028508SToomas Soome return (0);
592*22028508SToomas Soome }
593*22028508SToomas Soome
594*22028508SToomas Soome static int
tftp_stat(struct open_file * f,struct stat * sb)595*22028508SToomas Soome tftp_stat(struct open_file *f, struct stat *sb)
596*22028508SToomas Soome {
597*22028508SToomas Soome struct tftp_handle *tftpfile;
598*22028508SToomas Soome tftpfile = f->f_fsdata;
599*22028508SToomas Soome
600*22028508SToomas Soome sb->st_mode = 0444 | S_IFREG;
601*22028508SToomas Soome sb->st_nlink = 1;
602*22028508SToomas Soome sb->st_uid = 0;
603*22028508SToomas Soome sb->st_gid = 0;
604*22028508SToomas Soome sb->st_size = tftpfile->tftp_tsize;
605*22028508SToomas Soome return (0);
606*22028508SToomas Soome }
607*22028508SToomas Soome
608*22028508SToomas Soome static off_t
tftp_seek(struct open_file * f,off_t offset,int where)609*22028508SToomas Soome tftp_seek(struct open_file *f, off_t offset, int where)
610*22028508SToomas Soome {
611*22028508SToomas Soome struct tftp_handle *tftpfile;
612*22028508SToomas Soome tftpfile = f->f_fsdata;
613*22028508SToomas Soome
614*22028508SToomas Soome switch (where) {
615*22028508SToomas Soome case SEEK_SET:
616*22028508SToomas Soome tftpfile->off = offset;
617*22028508SToomas Soome break;
618*22028508SToomas Soome case SEEK_CUR:
619*22028508SToomas Soome tftpfile->off += offset;
620*22028508SToomas Soome break;
621*22028508SToomas Soome default:
622*22028508SToomas Soome errno = EOFFSET;
623*22028508SToomas Soome return (-1);
624*22028508SToomas Soome }
625*22028508SToomas Soome return (tftpfile->off);
626*22028508SToomas Soome }
627*22028508SToomas Soome
628*22028508SToomas Soome static int
tftp_set_blksize(struct tftp_handle * h,const char * str)629*22028508SToomas Soome tftp_set_blksize(struct tftp_handle *h, const char *str)
630*22028508SToomas Soome {
631*22028508SToomas Soome char *endptr;
632*22028508SToomas Soome int new_blksize;
633*22028508SToomas Soome int ret = 0;
634*22028508SToomas Soome
635*22028508SToomas Soome if (h == NULL || str == NULL)
636*22028508SToomas Soome return (ret);
637*22028508SToomas Soome
638*22028508SToomas Soome new_blksize =
639*22028508SToomas Soome (unsigned int)strtol(str, &endptr, 0);
640*22028508SToomas Soome
641*22028508SToomas Soome /*
642*22028508SToomas Soome * Only accept blksize value if it is numeric.
643*22028508SToomas Soome * RFC2348 specifies that acceptable values are 8-65464.
644*22028508SToomas Soome * Let's choose a limit less than MAXRSPACE.
645*22028508SToomas Soome */
646*22028508SToomas Soome if (*endptr == '\0' && new_blksize >= 8 &&
647*22028508SToomas Soome new_blksize <= TFTP_MAX_BLKSIZE) {
648*22028508SToomas Soome h->tftp_blksize = new_blksize;
649*22028508SToomas Soome ret = 1;
650*22028508SToomas Soome }
651*22028508SToomas Soome
652*22028508SToomas Soome return (ret);
653*22028508SToomas Soome }
654*22028508SToomas Soome
655*22028508SToomas Soome /*
656*22028508SToomas Soome * In RFC2347, the TFTP Option Acknowledgement package (OACK)
657*22028508SToomas Soome * is used to acknowledge a client's option negotiation request.
658*22028508SToomas Soome * The format of an OACK packet is:
659*22028508SToomas Soome * +-------+---~~---+---+---~~---+---+---~~---+---+---~~---+---+
660*22028508SToomas Soome * | opc | opt1 | 0 | value1 | 0 | optN | 0 | valueN | 0 |
661*22028508SToomas Soome * +-------+---~~---+---+---~~---+---+---~~---+---+---~~---+---+
662*22028508SToomas Soome *
663*22028508SToomas Soome * opc
664*22028508SToomas Soome * The opcode field contains a 6, for Option Acknowledgment.
665*22028508SToomas Soome *
666*22028508SToomas Soome * opt1
667*22028508SToomas Soome * The first option acknowledgment, copied from the original
668*22028508SToomas Soome * request.
669*22028508SToomas Soome *
670*22028508SToomas Soome * value1
671*22028508SToomas Soome * The acknowledged value associated with the first option. If
672*22028508SToomas Soome * and how this value may differ from the original request is
673*22028508SToomas Soome * detailed in the specification for the option.
674*22028508SToomas Soome *
675*22028508SToomas Soome * optN, valueN
676*22028508SToomas Soome * The final option/value acknowledgment pair.
677*22028508SToomas Soome */
678*22028508SToomas Soome static int
tftp_parse_oack(struct tftp_handle * h,char * buf,size_t len)679*22028508SToomas Soome tftp_parse_oack(struct tftp_handle *h, char *buf, size_t len)
680*22028508SToomas Soome {
681*22028508SToomas Soome /*
682*22028508SToomas Soome * We parse the OACK strings into an array
683*22028508SToomas Soome * of name-value pairs.
684*22028508SToomas Soome */
685*22028508SToomas Soome char *tftp_options[128] = { 0 };
686*22028508SToomas Soome char *val = buf;
687*22028508SToomas Soome int i = 0;
688*22028508SToomas Soome int option_idx = 0;
689*22028508SToomas Soome int blksize_is_set = 0;
690*22028508SToomas Soome int tsize = 0;
691*22028508SToomas Soome
692*22028508SToomas Soome unsigned int orig_blksize;
693*22028508SToomas Soome
694*22028508SToomas Soome while (option_idx < 128 && i < len) {
695*22028508SToomas Soome if (buf[i] == '\0') {
696*22028508SToomas Soome if (&buf[i] > val) {
697*22028508SToomas Soome tftp_options[option_idx] = val;
698*22028508SToomas Soome val = &buf[i] + 1;
699*22028508SToomas Soome ++option_idx;
700*22028508SToomas Soome }
701*22028508SToomas Soome }
702*22028508SToomas Soome ++i;
703*22028508SToomas Soome }
704*22028508SToomas Soome
705*22028508SToomas Soome /* Save the block size we requested for sanity check later. */
706*22028508SToomas Soome orig_blksize = h->tftp_blksize;
707*22028508SToomas Soome
708*22028508SToomas Soome /*
709*22028508SToomas Soome * Parse individual TFTP options.
710*22028508SToomas Soome * * "blksize" is specified in RFC2348.
711*22028508SToomas Soome * * "tsize" is specified in RFC2349.
712*22028508SToomas Soome */
713*22028508SToomas Soome for (i = 0; i < option_idx; i += 2) {
714*22028508SToomas Soome if (strcasecmp(tftp_options[i], "blksize") == 0) {
715*22028508SToomas Soome if (i + 1 < option_idx)
716*22028508SToomas Soome blksize_is_set =
717*22028508SToomas Soome tftp_set_blksize(h, tftp_options[i + 1]);
718*22028508SToomas Soome } else if (strcasecmp(tftp_options[i], "tsize") == 0) {
719*22028508SToomas Soome if (i + 1 < option_idx)
720*22028508SToomas Soome tsize = strtol(tftp_options[i + 1], NULL, 10);
721*22028508SToomas Soome if (tsize != 0)
722*22028508SToomas Soome h->tftp_tsize = tsize;
723*22028508SToomas Soome } else {
724*22028508SToomas Soome /*
725*22028508SToomas Soome * Do not allow any options we did not expect to be
726*22028508SToomas Soome * ACKed.
727*22028508SToomas Soome */
728*22028508SToomas Soome printf("unexpected tftp option '%s'\n",
729*22028508SToomas Soome tftp_options[i]);
730*22028508SToomas Soome return (-1);
731*22028508SToomas Soome }
732*22028508SToomas Soome }
733*22028508SToomas Soome
734*22028508SToomas Soome if (!blksize_is_set) {
735*22028508SToomas Soome /*
736*22028508SToomas Soome * If TFTP blksize was not set, try defaulting
737*22028508SToomas Soome * to the legacy TFTP blksize of SEGSIZE(512)
738*22028508SToomas Soome */
739*22028508SToomas Soome h->tftp_blksize = SEGSIZE;
740*22028508SToomas Soome } else if (h->tftp_blksize > orig_blksize) {
741*22028508SToomas Soome /*
742*22028508SToomas Soome * Server should not be proposing block sizes that
743*22028508SToomas Soome * exceed what we said we can handle.
744*22028508SToomas Soome */
745*22028508SToomas Soome printf("unexpected blksize %u\n", h->tftp_blksize);
746*22028508SToomas Soome return (-1);
747*22028508SToomas Soome }
748*22028508SToomas Soome
749*22028508SToomas Soome #ifdef TFTP_DEBUG
750*22028508SToomas Soome printf("tftp_blksize: %u\n", h->tftp_blksize);
751*22028508SToomas Soome printf("tftp_tsize: %lu\n", h->tftp_tsize);
752*22028508SToomas Soome #endif
753*22028508SToomas Soome return (0);
754*22028508SToomas Soome }
755