[BACK]Return to subr_tftproot.c CVS log [TXT][DIR] Up to [cvs.NetBSD.org] / src / sys / kern

Annotation of src/sys/kern/subr_tftproot.c, Revision 1.13

1.13    ! manu        1: /*     $NetBSD: subr_tftproot.c,v 1.12 2012/12/01 11:41:50 mbalmer Exp $ */
1.1       manu        2:
                      3: /*-
                      4:  * Copyright (c) 2007 Emmanuel Dreyfus, all rights reserved.
                      5:  *
                      6:  * Redistribution and use in source and binary forms, with or without
                      7:  * modification, are permitted provided that the following conditions
                      8:  * are met:
                      9:  * 1. Redistributions of source code must retain the above copyright
                     10:  *    notice, this list of conditions and the following disclaimer.
                     11:  * 2. Redistributions in binary form must reproduce the above copyright
                     12:  *    notice, this list of conditions and the following disclaimer in the
                     13:  *    documentation and/or other materials provided with the distribution.
                     14:  * 3. All advertising materials mentioning features or use of this software
                     15:  *    must display the following acknowledgement:
                     16:  *     This product includes software developed by Emmanuel Dreyfus
                     17:  * 4. The name of the author may not be used to endorse or promote
                     18:  *    products derived from this software without specific prior written
                     19:  *    permission.
                     20:  *
                     21:  * THIS SOFTWARE IS PROVIDED BY THE THE AUTHOR AND CONTRIBUTORS ``AS IS''
                     22:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
                     23:  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
                     24:  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
                     25:  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
                     26:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
                     27:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
                     28:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
                     29:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
                     30:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
                     31:  * POSSIBILITY OF SUCH DAMAGE.
                     32:  */
                     33:
                     34: /*
                     35:  * Download the root RAMdisk through TFTP at root mount time
                     36:  */
                     37:
                     38: #include "opt_tftproot.h"
                     39: #include "opt_md.h"
                     40:
                     41: #include <sys/cdefs.h>
1.13    ! manu       42: __KERNEL_RCSID(0, "$NetBSD: subr_tftproot.c,v 1.12 2012/12/01 11:41:50 mbalmer Exp $");
1.1       manu       43:
                     44: #include <sys/param.h>
                     45: #include <sys/types.h>
1.13    ! manu       46: #include <sys/vnode.h>
1.1       manu       47: #include <sys/mount.h>
                     48: #include <sys/lwp.h>
1.2       yamt       49: #include <sys/kmem.h>
1.1       manu       50: #include <sys/mbuf.h>
                     51: #include <sys/timevar.h>
                     52: #include <sys/socketvar.h>
                     53:
                     54: #include <net/if.h>
                     55:
                     56: #include <dev/md.h>
                     57:
                     58: #include <netinet/in.h>
                     59:
                     60: #include <nfs/rpcv2.h>
                     61:
                     62: #include <nfs/nfsproto.h>
                     63: #include <nfs/nfs.h>
                     64: #include <nfs/nfsmount.h>
                     65: #include <nfs/nfsdiskless.h>
                     66: #include <nfs/nfs_var.h>
                     67:
                     68: extern void       mdattach(int);
                     69:
                     70: /*
                     71:  * Copied from <lib/libsa/tftp.h>
                     72:  */
                     73:
                     74: #define SEGSIZE         512             /* data segment size */
                     75:
                     76: /*
                     77:  * Packet types.
                     78:  */
                     79: #define RRQ     01                      /* read request */
                     80: #define WRQ     02                      /* write request */
                     81: #define DATA    03                      /* data packet */
                     82: #define ACK     04                      /* acknowledgement */
                     83: #define ERROR   05                      /* error code */
                     84:
                     85: struct  tftphdr {
                     86:         short   th_opcode;              /* packet type */
                     87:         union {
                     88:                 unsigned short tu_block; /* block # */
                     89:                 short   tu_code;        /* error code */
                     90:                 char    tu_stuff[1];    /* request packet stuff */
                     91:         } th_u;
                     92:         char    th_data[1];             /* data or error string */
                     93: } __packed;
                     94:
                     95: #define th_block        th_u.tu_block
                     96: #define th_code         th_u.tu_code
                     97: #define th_stuff        th_u.tu_stuff
                     98: #define th_msg          th_data
                     99:
                    100: #define IPPORT_TFTP 69
                    101:
                    102: #ifdef TFTPROOT_DEBUG
                    103: #define DPRINTF(x) printf x
                    104: #else
                    105: #define DPRINTF(x)
                    106: #endif
                    107:
                    108: struct tftproot_handle {
                    109:        struct nfs_diskless *trh_nd;
                    110:        void *trh_base;
                    111:        size_t trh_len;
                    112:        unsigned short trh_block;
                    113:        int trh_flags;
                    114: };
                    115:
                    116: #define TRH_FINISHED   1
                    117:
1.11      chs       118: int tftproot_dhcpboot(device_t);
1.1       manu      119:
                    120: static int tftproot_getfile(struct tftproot_handle *, struct lwp *);
1.11      chs       121: static int tftproot_recv(struct mbuf *, void *);
1.1       manu      122:
                    123: int
1.11      chs       124: tftproot_dhcpboot(device_t bootdv)
1.1       manu      125: {
                    126:        struct nfs_diskless *nd = NULL;
                    127:        struct ifnet *ifp = NULL;
                    128:        struct lwp *l;
                    129:        struct tftproot_handle trh;
1.11      chs       130:        device_t dv;
1.1       manu      131:        int error = -1;
                    132:
1.10      manu      133:        if (rootspec != NULL) {
                    134:                IFNET_FOREACH(ifp)
                    135:                        if (strcmp(rootspec, ifp->if_xname) == 0)
                    136:                                break;
                    137:        }
                    138:
                    139:        if ((ifp == NULL) &&
                    140:            (bootdv != NULL && device_class(bootdv) == DV_IFNET)) {
1.1       manu      141:                IFNET_FOREACH(ifp)
1.4       cegger    142:                        if (strcmp(device_xname(bootdv), ifp->if_xname) == 0)
1.1       manu      143:                                break;
                    144:        }
                    145:
                    146:        if (ifp == NULL) {
                    147:                DPRINTF(("%s():%d ifp is NULL\n", __func__, __LINE__));
                    148:                goto out;
                    149:        }
                    150:
1.3       joerg     151:        dv = device_find_by_xname(ifp->if_xname);
1.1       manu      152:
                    153:        if ((dv == NULL) || (device_class(dv) != DV_IFNET)) {
                    154:                DPRINTF(("%s():%d cannot find device for interface %s\n",
                    155:                    __func__, __LINE__, ifp->if_xname));
                    156:                goto out;
                    157:        }
                    158:
                    159:        root_device = dv;
                    160:
                    161:        l = curlwp; /* XXX */
                    162:
1.2       yamt      163:        nd = kmem_zalloc(sizeof(*nd), KM_SLEEP);
1.1       manu      164:        nd->nd_ifp = ifp;
                    165:        nd->nd_nomount = 1;
                    166:
                    167:        if ((error = nfs_boot_init(nd, l)) != 0) {
                    168:                DPRINTF(("%s():%d nfs_boot_init returned %d\n",
                    169:                    __func__, __LINE__, error));
                    170:                goto out;
                    171:        }
                    172:
                    173:        /*
                    174:         * Strip leading "tftp:"
                    175:         */
                    176: #define PREFIX "tftp:"
                    177:        if (strstr(nd->nd_bootfile, PREFIX) == nd->nd_bootfile)
                    178:                (void)memmove(nd->nd_bootfile,
                    179:                              nd->nd_bootfile + (sizeof(PREFIX) - 1),
                    180:                              sizeof(nd->nd_bootfile) - sizeof(PREFIX));
                    181: #undef PREFIX
                    182:
                    183:        printf("tftproot: bootfile=%s\n", nd->nd_bootfile);
                    184:
1.8       cegger    185:        memset(&trh, 0, sizeof(trh));
1.1       manu      186:        trh.trh_nd = nd;
                    187:        trh.trh_block = 1;
                    188:
                    189:        if ((error = tftproot_getfile(&trh, l)) != 0) {
                    190:                DPRINTF(("%s():%d tftproot_getfile returned %d\n",
                    191:                    __func__, __LINE__, error));
                    192:                goto out;
                    193:        }
                    194:
                    195:        error = 0;
                    196:
                    197: out:
                    198:        if (nd)
1.2       yamt      199:                kmem_free(nd, sizeof(*nd));
1.1       manu      200:
                    201:        return error;
                    202: }
                    203:
                    204: static int
1.7       dsl       205: tftproot_getfile(struct tftproot_handle *trh, struct lwp *l)
1.1       manu      206: {
                    207:        struct socket *so = NULL;
                    208:        struct mbuf *m_serv = NULL;
                    209:        struct mbuf *m_outbuf = NULL;
                    210:        struct sockaddr_in *sin;
                    211:        struct tftphdr *tftp;
                    212:        size_t packetlen, namelen;
                    213:        int error = -1;
                    214:        const char octetstr[] = "octet";
                    215:        size_t hdrlen = sizeof(*tftp) - sizeof(tftp->th_data);
                    216:        char *cp;
                    217:
1.5       ad        218:        if ((error = socreate(AF_INET, &so, SOCK_DGRAM, 0, l, NULL)) != 0) {
1.1       manu      219:                DPRINTF(("%s():%d socreate returned %d\n",
                    220:                    __func__, __LINE__, error));
                    221:                goto out;
                    222:        }
                    223:
                    224:        /*
                    225:         * Set timeout
                    226:         */
                    227:        if ((error = nfs_boot_setrecvtimo(so))) {
                    228:                DPRINTF(("%s():%d SO_RCVTIMEO failed %d\n",
                    229:                    __func__, __LINE__, error));
                    230:                goto out;
                    231:        }
                    232:
                    233:        /*
                    234:         * Set server address and port
                    235:         */
                    236:        m_serv = m_get(M_WAIT, MT_SONAME);
                    237:        m_serv->m_len = sizeof(*sin);
                    238:        sin = mtod(m_serv, struct sockaddr_in *);
                    239:        memcpy(sin, &trh->trh_nd->nd_root.ndm_saddr, sizeof(*sin));
                    240:        sin->sin_port = htons(IPPORT_TFTP);
                    241:
                    242:        /*
                    243:         * Set send buffer, prepare the TFTP packet
                    244:         */
                    245:        namelen = strlen(trh->trh_nd->nd_bootfile) + 1;
                    246:        packetlen = sizeof(tftp->th_opcode) + namelen + sizeof(octetstr);
                    247:        if (packetlen > MSIZE) {
                    248:                DPRINTF(("%s():%d boot filename too long (%ld bytes)\n",
                    249:                    __func__, __LINE__, (long)namelen));
                    250:                goto out;
                    251:        }
                    252:
                    253:        m_outbuf = m_gethdr(M_WAIT, MT_DATA);
                    254:        m_clget(m_outbuf, M_WAIT);
                    255:        m_outbuf->m_len = packetlen;
                    256:        m_outbuf->m_pkthdr.len = packetlen;
                    257:        m_outbuf->m_pkthdr.rcvif = NULL;
                    258:
                    259:        tftp = mtod(m_outbuf, struct tftphdr *);
                    260:        memset(tftp, 0, packetlen);
                    261:
                    262:        tftp->th_opcode = htons((short)RRQ);
                    263:        cp = tftp->th_stuff;
                    264:        (void)strncpy(cp,  trh->trh_nd->nd_bootfile, namelen);
                    265:        cp += namelen;
                    266:        (void)strncpy(cp, octetstr, sizeof(octetstr));
                    267:
                    268:        /*
                    269:         * Perform the file transfer
                    270:         */
                    271:        sin = (struct sockaddr_in *)&trh->trh_nd->nd_root.ndm_saddr;
                    272:        printf("tftproot: download %s:%s ",
                    273:            inet_ntoa(sin->sin_addr), trh->trh_nd->nd_bootfile);
                    274:
                    275:        do {
                    276:                /*
                    277:                 * Show progress for every 200 blocks (100kB)
                    278:                 */
                    279: #ifndef TFTPROOT_PROGRESS
                    280: #define TFTPROOT_PROGRESS 200
                    281: #endif
                    282:                if ((trh->trh_block % TFTPROOT_PROGRESS) == 0)
                    283:                        twiddle();
                    284:
                    285:                /*
                    286:                 * Send the packet and receive the answer.
                    287:                 * We get the sender address here, which should be
                    288:                 * the same server with a different port
                    289:                 */
                    290:                if ((error = nfs_boot_sendrecv(so, m_serv, NULL, m_outbuf,
                    291:                    tftproot_recv, NULL, &m_serv, trh, l)) != 0) {
                    292:                        DPRINTF(("%s():%d sendrecv failed %d\n",
                    293:                            __func__, __LINE__, error));
                    294:                        goto out;
                    295:                }
                    296:
                    297:                /*
                    298:                 * Accomodate the packet length for acks.
                    299:                 * This is really needed only on first pass
                    300:                 */
                    301:                m_outbuf->m_len = hdrlen;
                    302:                m_outbuf->m_pkthdr.len = hdrlen;
                    303:                tftp->th_opcode = htons((short)ACK);
                    304:                tftp->th_block = htons(trh->trh_block);
                    305:
                    306:
                    307:                /*
                    308:                 * Check for termination
                    309:                 */
                    310:                if (trh->trh_flags & TRH_FINISHED)
                    311:                        break;
                    312:
                    313:                trh->trh_block++;
                    314:        } while (1/* CONSTCOND */);
                    315:
                    316:        printf("\n");
                    317:
                    318:        /*
                    319:         * Ack the last block. so_send frees m_outbuf, therefore
                    320:         * we do not want to free it ourselves.
                    321:         * Ignore errors, as we already have the whole file.
                    322:         */
                    323:        if ((error = (*so->so_send)(so, m_serv, NULL,
                    324:            m_outbuf, NULL, 0, l)) != 0)
                    325:                DPRINTF(("%s():%d tftproot: sosend returned %d\n",
                    326:                    __func__, __LINE__, error));
                    327:        else
                    328:                m_outbuf = NULL;
                    329:
                    330:        /*
                    331:         * And use it as the root ramdisk.
                    332:         */
                    333:        DPRINTF(("%s():%d RAMdisk loaded: %ld@%p\n",
                    334:            __func__, __LINE__, trh->trh_len, trh->trh_base));
                    335:        md_root_setconf(trh->trh_base, trh->trh_len);
                    336:        mdattach(0);
                    337:
                    338:        error = 0;
                    339: out:
                    340:        if (m_serv)
                    341:                m_freem(m_serv);
                    342:
                    343:        if (m_outbuf)
                    344:                m_freem(m_outbuf);
                    345:
                    346:        if (so)
                    347:                soclose(so);
                    348:
                    349:        return error;
                    350: }
                    351:
                    352: static int
1.7       dsl       353: tftproot_recv(struct mbuf *m, void *ctx)
1.1       manu      354: {
                    355:        struct tftproot_handle *trh = ctx;
                    356:        struct tftphdr *tftp;
                    357:        size_t newlen;
                    358:        size_t hdrlen = sizeof(*tftp) - sizeof(tftp->th_data);
                    359:
                    360:        /*
                    361:         * Check for short packet
                    362:         */
                    363:        if (m->m_pkthdr.len < hdrlen) {
                    364:                DPRINTF(("%s():%d short reply (%d bytes)\n",
                    365:                    __func__, __LINE__, m->m_pkthdr.len));
                    366:                return -1;
                    367:        }
                    368:
                    369:        /*
                    370:         * Check for packet too large for being a TFTP packet
                    371:         */
                    372:        if (m->m_pkthdr.len > hdrlen + SEGSIZE) {
                    373:                DPRINTF(("%s():%d packet too big (%d bytes)\n",
                    374:                    __func__, __LINE__, m->m_pkthdr.len));
                    375:                return -1;
                    376:        }
                    377:
                    378:
                    379:        /*
                    380:         * Examine the TFTP header
                    381:         */
                    382:        if (m->m_len > sizeof(*tftp)) {
                    383:                if ((m = m_pullup(m, sizeof(*tftp))) == NULL) {
                    384:                        DPRINTF(("%s():%d m_pullup failed\n",
                    385:                            __func__, __LINE__));
                    386:                        return -1;
                    387:                }
                    388:        }
                    389:        tftp = mtod(m, struct tftphdr *);
                    390:
                    391:        /*
                    392:         * We handle data or error
                    393:         */
                    394:        switch(ntohs(tftp->th_opcode)) {
                    395:        case DATA:
                    396:                break;
                    397:
                    398:        case ERROR: {
                    399:                char errbuf[SEGSIZE + 1];
                    400:                int i;
                    401:
                    402:                for (i = 0; i < SEGSIZE; i++) {
                    403:                        if ((tftp->th_data[i] < ' ') ||
                    404:                            (tftp->th_data[i] > '~')) {
                    405:                                errbuf[i] = '\0';
                    406:                                break;
                    407:                        }
                    408:                        errbuf[i] = tftp->th_data[i];
                    409:                }
                    410:                errbuf[SEGSIZE] = '\0';
                    411:
                    412:                printf("tftproot: TFTP server returned error %d, %s\n",
                    413:                    ntohs(tftp->th_code), errbuf);
                    414:
                    415:                return -1;
                    416:                break;
                    417:        }
                    418:
                    419:        default:
                    420:                DPRINTF(("%s():%d unexpected tftp reply opcode %d\n",
                    421:                    __func__, __LINE__, ntohs(tftp->th_opcode)));
                    422:                return -1;
                    423:                break;
                    424:        }
                    425:
                    426:        /*
                    427:         * Check for last packet, which does not fill the whole space
                    428:         */
                    429:        if (m->m_pkthdr.len < hdrlen + SEGSIZE) {
                    430:                DPRINTF(("%s():%d last chunk (%d bytes)\n",
                    431:                    __func__, __LINE__, m->m_pkthdr.len));
                    432:                trh->trh_flags |= TRH_FINISHED;
                    433:        }
                    434:
                    435:
                    436:        if (ntohs(tftp->th_block) != trh->trh_block) {
                    437:                DPRINTF(("%s():%d expected block %d, got block %d\n",
                    438:                    __func__, __LINE__, trh->trh_block, ntohs(tftp->th_block)));
                    439:                return -1;
                    440:        }
                    441:
                    442:        /*
1.12      mbalmer   443:         * Grow the receiving buffer to accommodate new data
1.1       manu      444:         */
                    445:        newlen = trh->trh_len + (m->m_pkthdr.len - hdrlen);
                    446:        if ((trh->trh_base = realloc(trh->trh_base,
                    447:            newlen, M_TEMP, M_WAITOK)) == NULL) {
                    448:                DPRINTF(("%s():%d failed to realloc %ld bytes\n",
                    449:                    __func__, __LINE__, (long)newlen));
                    450:                return -1;
                    451:        }
                    452:
                    453:        /*
                    454:         * Copy the data
                    455:         */
                    456:        m_copydata(m, hdrlen, m->m_pkthdr.len - hdrlen,
                    457:                   (char *)trh->trh_base + trh->trh_len);
                    458:        trh->trh_len = newlen;
                    459:
                    460:        return 0;
                    461: }
                    462:

CVSweb <webmaster@jp.NetBSD.org>