[pxe] Select the correct network device on multiport cards
[people/mcb30/gpxe.git] / src / arch / i386 / image / pxe_image.c
1 /*
2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 /**
20  * @file
21  *
22  * PXE image format
23  *
24  */
25
26 #include <pxe.h>
27 #include <pxe_call.h>
28 #include <gpxe/uaccess.h>
29 #include <gpxe/image.h>
30 #include <gpxe/segment.h>
31 #include <gpxe/netdevice.h>
32 #include <gpxe/features.h>
33
34 FEATURE ( FEATURE_IMAGE, "PXE", DHCP_EB_FEATURE_PXE, 1 );
35
36 struct image_type pxe_image_type __image_type ( PROBE_PXE );
37
38 /**
39  * Execute PXE image
40  *
41  * @v image             PXE image
42  * @ret rc              Return status code
43  */
44 static int pxe_exec ( struct image *image ) {
45         struct net_device *netdev;
46         int rc;
47
48         /* Ensure that PXE stack is ready to use */
49         pxe_init_structures();
50         pxe_hook_int1a();
51
52         /* Arbitrarily pick the first open network device to use for PXE */
53         for_each_netdev ( netdev ) {
54                 if ( netdev->state & NETDEV_OPEN ) {
55                         pxe_set_netdev ( netdev );
56                         break;
57                 }
58         }
59
60         /* Many things will break if pxe_netdev is NULL */
61         if ( ! pxe_netdev ) {
62                 DBGC ( image, "IMAGE %p could not locate PXE net device\n",
63                        image );
64                 return -ENODEV;
65         }
66
67         /* Start PXE NBP */
68         rc = pxe_start_nbp();
69
70         /* Deactivate PXE */
71         pxe_set_netdev ( NULL );
72         pxe_unhook_int1a();
73
74         return rc;
75 }
76
77 /**
78  * Load PXE image into memory
79  *
80  * @v image             PXE file
81  * @ret rc              Return status code
82  */
83 int pxe_load ( struct image *image ) {
84         userptr_t buffer = real_to_user ( 0, 0x7c00 );
85         size_t filesz = image->len;
86         size_t memsz = image->len;
87         int rc;
88
89         /* Images too large to fit in base memory cannot be PXE
90          * images.  We include this check to help prevent unrecognised
91          * images from being marked as PXE images, since PXE images
92          * have no signature we can check against.
93          */
94         if ( filesz > ( 0xa0000 - 0x7c00 ) )
95                 return -ENOEXEC;
96
97         /* There are no signature checks for PXE; we will accept anything */
98         if ( ! image->type )
99                 image->type = &pxe_image_type;
100
101         /* Verify and prepare segment */
102         if ( ( rc = prep_segment ( buffer, filesz, memsz ) ) != 0 ) {
103                 DBGC ( image, "IMAGE %p could not prepare segment: %s\n",
104                        image, strerror ( rc ) );
105                 return rc;
106         }
107
108         /* Copy image to segment */
109         memcpy_user ( buffer, 0, image->data, 0, filesz );
110
111         return 0;
112 }
113
114 /** PXE image type */
115 struct image_type pxe_image_type __image_type ( PROBE_PXE ) = {
116         .name = "PXE",
117         .load = pxe_load,
118         .exec = pxe_exec,
119 };