Home | History | Annotate | Line # | Download | only in sboot
etherfun.c revision 1.9
      1 /*	$NetBSD: etherfun.c,v 1.9 2008/01/12 09:54:33 tsutsui Exp $	*/
      2 
      3 /*
      4  *
      5  * Copyright (c) 1995 Charles D. Cranor and Seth Widoff
      6  * All rights reserved.
      7  *
      8  * Redistribution and use in source and binary forms, with or without
      9  * modification, are permitted provided that the following conditions
     10  * are met:
     11  * 1. Redistributions of source code must retain the above copyright
     12  *    notice, this list of conditions and the following disclaimer.
     13  * 2. Redistributions in binary form must reproduce the above copyright
     14  *    notice, this list of conditions and the following disclaimer in the
     15  *    documentation and/or other materials provided with the distribution.
     16  * 3. All advertising materials mentioning features or use of this software
     17  *    must display the following acknowledgement:
     18  *      This product includes software developed by Charles D. Cranor
     19  *	and Seth Widoff.
     20  * 4. The name of the author may not be used to endorse or promote products
     21  *    derived from this software without specific prior written permission.
     22  *
     23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     24  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     25  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     26  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     27  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     28  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     29  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     30  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     31  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     32  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     33  */
     34 /* etherfun.c */
     35 
     36 #include "sboot.h"
     37 #include "etherfun.h"
     38 
     39 /* Construct and send a rev arp packet */
     40 void
     41 do_rev_arp(void)
     42 {
     43 	int i;
     44 
     45 	for (i = 0; i < 6; i++) {
     46 		eh->ether_dhost[i] = 0xff;
     47 	}
     48 	memcpy(eh->ether_shost, myea, 6);
     49 	eh->ether_type = ETYPE_RARP;
     50 
     51 	rarp->ar_hrd = 1;	/* hardware type is 1 */
     52 	rarp->ar_pro = PTYPE_IP;
     53 	rarp->ar_hln = 6;	/* length of hardware address is 6 bytes */
     54 	rarp->ar_pln = 4;	/* length of ip address is 4 byte */
     55 	rarp->ar_op = OPCODE_RARP;
     56 	memcpy(rarp->arp_sha, myea, sizeof(myea));
     57 	memcpy(rarp->arp_tha, myea, sizeof(myea));
     58 	for (i = 0; i < 4; i++) {
     59 		rarp->arp_spa[i] = rarp->arp_tpa[i] = 0x00;
     60 	}
     61 
     62 	le_put(buf, 76);
     63 }
     64 
     65 /* Receive and disassemble the rev_arp reply */
     66 
     67 int
     68 get_rev_arp(void)
     69 {
     70 
     71 	le_get(buf, sizeof(buf), 6);
     72 	if (eh->ether_type == ETYPE_RARP && rarp->ar_op == OPCODE_REPLY) {
     73 		memcpy(myip, rarp->arp_tpa, sizeof(rarp->arp_tpa));
     74 		memcpy(servip, rarp->arp_spa, sizeof(rarp->arp_spa));
     75 		memcpy(servea, rarp->arp_sha, sizeof(rarp->arp_sha));
     76 		return 1;
     77 	}
     78 	return 0;
     79 }
     80 
     81 /* Try to get a reply to a rev arp request */
     82 
     83 int
     84 rev_arp(void)
     85 {
     86 	int tries = 0;
     87 
     88 	while (tries < 5) {
     89 		do_rev_arp();
     90 		if (get_rev_arp()) {
     91 			return 1;
     92 		}
     93 		tries++;
     94 	}
     95 	return 0;
     96 }
     97 
     98 /* Send a tftp read request or acknowledgement
     99    mesgtype 0 is a read request, 1 is an aknowledgement */
    100 
    101 void
    102 do_send_tftp(int mesgtype)
    103 {
    104 	u_long res, iptmp, lcv;
    105 	u_char *tot;
    106 
    107 	if (mesgtype == 0) {
    108 		tot = tftp_r + (sizeof(MSG) - 1);
    109 		myport = (u_short)time();
    110 		if (myport < 1000)
    111 			myport += 1000;
    112 		servport = FTP_PORT; /* to start */
    113 	} else {
    114 		tot = (u_char *)tftp_a + 4;
    115 	}
    116 
    117 	memcpy (eh->ether_dhost, servea, sizeof(servea));
    118 	memcpy (eh->ether_shost, myea, sizeof(myea));
    119 	eh->ether_type = ETYPE_IP;
    120 
    121 	iph->ip_v = IP_VERSION;
    122 	iph->ip_hl = IP_HLEN;
    123 	iph->ip_tos = 0;	/* type of service is 0 */
    124 	iph->ip_id = 0;		/* id field is 0 */
    125 	iph->ip_off = IP_DF;
    126 	iph->ip_ttl = 3;	/* time to live is 3 seconds/hops */
    127 	iph->ip_p = IPP_UDP;
    128 	memcpy(iph->ip_src, myip, sizeof(myip));
    129 	memcpy(iph->ip_dst, servip, sizeof(servip));
    130 	iph->ip_sum = 0;
    131 	iph->ip_len = tot - (u_char *)iph;
    132 	res = oc_cksum(iph, sizeof(struct ip), 0);
    133 	iph->ip_sum = 0xffff & ~res;
    134 	udph->uh_sport = myport;
    135 	udph->uh_dport = servport;
    136 	udph->uh_sum = 0;
    137 
    138 	if (mesgtype)  {
    139 		tftp_a->op_code = FTPOP_ACKN;
    140 		tftp_a->block = (u_short)(mesgtype);
    141 	} else {
    142 		memcpy (&iptmp, myip, sizeof(iptmp));
    143 		memcpy(tftp_r, MSG, (sizeof(MSG)-1));
    144 		for (lcv = 9; lcv >= 2; lcv--) {
    145 			tftp_r[lcv] = HEXDIGITS[iptmp & 0xF];
    146 
    147 			iptmp = iptmp >> 4;
    148 		}
    149 	}
    150 
    151 	udph->uh_ulen = tot - (u_char *)udph;
    152 
    153 	le_put(buf, tot - buf);
    154 }
    155 
    156 /* Attempt to tftp a file and read it into memory */
    157 
    158 int
    159 do_get_file(void)
    160 {
    161 	int fail = 0, oldlen;
    162 	char *loadat = (char *)LOAD_ADDR;
    163 	last_ack = 0;
    164 
    165 	do_send_tftp(READ);
    166 	for (;;) {
    167 		if (le_get(buf, sizeof(buf), 5) == 0) { /* timeout occurred */
    168 			if (last_ack) {
    169 				do_send_tftp(last_ack);
    170 			} else {
    171 				do_send_tftp(READ);
    172 			}
    173 			fail++;
    174 			if (fail > 5) {
    175 			        printf("\n");
    176 				return 1;
    177 			}
    178 		} else {
    179 			printf("%x \r", tftp->info.block*512);
    180 			if ((eh->ether_type != ETYPE_IP) ||
    181 			    (iph->ip_p != IPP_UDP)) {
    182 				fail++;
    183 				continue;
    184 			}
    185 			if (servport == FTP_PORT) servport = udph->uh_sport;
    186 			if (tftp->info.op_code == FTPOP_ERR) {
    187 				printf("TFTP: Download error %d: %s\n",
    188 				    tftp->info.block, tftp->data);
    189 				return 1;
    190 			}
    191 			if (tftp->info.block != last_ack + 1) {
    192 				/* we received the wrong block */
    193 				if (tftp->info.block < last_ack +1) {
    194 					/* ackn whatever we received */
    195 					do_send_tftp(tftp->info.block);
    196 				} else {
    197 					/* ackn the last confirmed block */
    198 					do_send_tftp(last_ack);
    199 				}
    200 				fail++;
    201 			} else {		/* we got the right block */
    202 				fail = 0;
    203 				last_ack++;
    204 				oldlen = udph->uh_ulen;
    205 				do_send_tftp( last_ack );
    206 #if 0
    207 				printf("memcpy %x %x %d\n", loadat,
    208 				    &tftp->data, oldlen - 12);
    209 #endif
    210 				memcpy(loadat, &tftp->data, oldlen - 12);
    211 				loadat += oldlen - 12;
    212 				if (oldlen < (8 + 4 + 512)) {
    213 					printf("\n");
    214 					return 0;
    215 				}
    216 			}
    217 		}
    218 	}
    219 	printf("\n");
    220 	return 0;
    221 }
    222