27a7e42d278962f626adb313af8cbf8ff1406b1e
[people/sha0/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
33 /** PXE load address segment */
34 #define PXE_LOAD_SEGMENT 0
35
36 /** PXE load address offset */
37 #define PXE_LOAD_OFFSET 0x7c00
38
39 struct image_type pxe_image_type __image_type ( PROBE_PXE );
40
41 /**
42  * Execute PXE image
43  *
44  * @v image             PXE image
45  * @ret rc              Return status code
46  */
47 static int pxe_exec ( struct image *image __unused ) {
48         struct net_device *netdev;
49         int discard_b, discard_c;
50         uint16_t rc;
51
52         /* Ensure that PXE stack is ready to use */
53         pxe_init_structures();
54         pxe_hook_int1a();
55
56         /* Arbitrarily pick the first open network device to use for PXE */
57         for_each_netdev ( netdev ) {
58                 pxe_netdev = netdev;
59                 break;
60         }
61
62         /* Far call to PXE NBP */
63         __asm__ __volatile__ ( REAL_CODE ( "pushw %%cx\n\t"
64                                            "pushw %%ax\n\t"
65                                            "movw %%cx, %%es\n\t"
66                                            "lcall $0, $0x7c00\n\t" )
67                                : "=a" ( rc ), "=b" ( discard_b ),
68                                  "=c" ( discard_c )
69                                :  "a" ( & __from_text16 ( ppxe ) ),
70                                   "b" ( & __from_text16 ( pxenv ) ),
71                                   "c" ( rm_cs )
72                                : "edx", "esi", "edi", "ebp", "memory" );
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         /* There are no signature checks for PXE; we will accept anything */
90         if ( ! image->type )
91                 image->type = &pxe_image_type;
92
93         /* Verify and prepare segment */
94         if ( ( rc = prep_segment ( buffer, filesz, memsz ) != 0 ) ) {
95                 DBG ( "PXE image could not prepare segment: %s\n",
96                       strerror ( rc ) );
97                 return rc;
98         }
99
100         /* Copy image to segment */
101         memcpy_user ( buffer, 0, image->data, 0, filesz );
102
103         return 0;
104 }
105
106 /** PXE image type */
107 struct image_type pxe_image_type __image_type ( PROBE_PXE ) = {
108         .name = "PXE",
109         .load = pxe_load,
110         .exec = pxe_exec,
111 };