Add phys_to_user() and copy_user().
[people/xl0/gpxe.git] / src / arch / i386 / include / librm.h
1 #ifndef LIBRM_H
2 #define LIBRM_H
3
4 /* Drag in protected-mode segment selector values */
5 #include "virtaddr.h"
6 #include "realmode.h"
7
8 #ifndef ASSEMBLY
9
10 #include "stddef.h"
11 #include "string.h"
12
13 /*
14  * Data structures and type definitions
15  *
16  */
17
18 /* Access to variables in .data16 and .text16 */
19 extern char *data16;
20 extern char *text16;
21
22 #define __data16( variable )                                            \
23         __attribute__ (( section ( ".data16" ) ))                       \
24         _data16_ ## variable __asm__ ( #variable )
25
26 #define __data16_array( variable, array )                               \
27         __attribute__ (( section ( ".data16" ) ))                       \
28         _data16_ ## variable array __asm__ ( #variable )
29
30 #define __text16( variable )                                            \
31         __attribute__ (( section ( ".text16.data" ) ))                  \
32         _text16_ ## variable __asm__ ( #variable )
33
34 #define __text16_array( variable, array )                               \
35         __attribute__ (( section ( ".text16.data" ) ))                  \
36         _text16_ ## variable array __asm__ ( #variable )
37
38 #define __use_data16( variable )                                        \
39         ( * ( ( typeof ( _data16_ ## variable ) * )                     \
40               & ( data16 [ ( size_t ) & ( _data16_ ## variable ) ] ) ) )
41
42 #define __use_text16( variable )                                        \
43         ( * ( ( typeof ( _text16_ ## variable ) * )                     \
44               & ( text16 [ ( size_t ) & ( _text16_ ## variable ) ] ) ) )
45
46 #define __from_data16( variable )                                       \
47         ( * ( ( typeof ( variable ) * )                                 \
48               ( ( ( void * ) &(variable) ) - ( ( void * ) data16 ) ) ) )
49
50 #define __from_text16( variable )                                       \
51         ( * ( ( typeof ( variable ) * )                                 \
52               ( ( ( void * ) &(variable) ) - ( ( void * ) text16 ) ) ) )
53
54 /* Variables in librm.S, present in the normal data segment */
55 extern uint16_t rm_sp;
56 extern uint16_t rm_ss;
57 extern uint32_t pm_esp;
58 extern uint16_t __data16 ( rm_cs );
59 #define rm_cs __use_data16 ( rm_cs )
60 extern uint16_t __text16 ( rm_ds );
61 #define rm_ds __use_text16 ( rm_ds )
62
63 /* Functions that librm expects to be able to link to.  Included here
64  * so that the compiler will catch prototype mismatches.
65  */
66 extern void gateA20_set ( void );
67
68 /*
69  * librm_mgmt: functions for manipulating base memory and executing
70  * real-mode code.
71  *
72  * Full API documentation for these functions is in realmode.h.
73  *
74  */
75
76 /* Macro for obtaining a physical address from a segment:offset pair. */
77 #define VIRTUAL(x,y) ( phys_to_virt ( ( ( x ) << 4 ) + ( y ) ) )
78
79 /* Copy to/from base memory */
80 static inline __attribute__ (( always_inline )) void
81 copy_to_real_librm ( unsigned int dest_seg, unsigned int dest_off,
82                      void *src, size_t n ) {
83         memcpy ( VIRTUAL ( dest_seg, dest_off ), src, n );
84 }
85 static inline __attribute__ (( always_inline )) void
86 copy_from_real_librm ( void *dest, unsigned int src_seg,
87                        unsigned int src_off, size_t n ) {
88         memcpy ( dest, VIRTUAL ( src_seg, src_off ), n );
89 }
90 #define put_real_librm( var, dest_seg, dest_off )                             \
91         do {                                                                  \
92                 * ( ( typeof(var) * ) VIRTUAL ( dest_seg, dest_off ) ) = var; \
93         } while ( 0 )
94 #define get_real_librm( var, src_seg, src_off )                               \
95         do {                                                                  \
96                 var = * ( ( typeof(var) * ) VIRTUAL ( src_seg, src_off ) ); \
97         } while ( 0 )
98 #define copy_to_real copy_to_real_librm
99 #define copy_from_real copy_from_real_librm
100 #define put_real put_real_librm
101 #define get_real get_real_librm
102
103 /**
104  * A pointer to a user buffer
105  *
106  * Even though we could just use a void *, we use an intptr_t so that
107  * attempts to use normal pointers show up as compiler warnings.  Such
108  * code is actually valid for librm, but not for libkir (i.e. under
109  * KEEP_IT_REAL), so it's good to have the warnings even under librm.
110  */
111 typedef intptr_t userptr_t;
112
113 /**
114  * Copy data to user buffer
115  *
116  * @v buffer    User buffer
117  * @v offset    Offset within user buffer
118  * @v src       Source
119  * @v len       Length
120  */
121 static inline __attribute__ (( always_inline )) void
122 copy_to_user ( userptr_t buffer, off_t offset, const void *src, size_t len ) {
123         memcpy ( ( ( void * ) buffer + offset ), src, len );
124 }
125
126 /**
127  * Copy data from user buffer
128  *
129  * @v dest      Destination
130  * @v buffer    User buffer
131  * @v offset    Offset within user buffer
132  * @v len       Length
133  */
134 static inline __attribute__ (( always_inline )) void
135 copy_from_user ( void *dest, userptr_t buffer, off_t offset, size_t len ) {
136         memcpy ( dest, ( ( void * ) buffer + offset ), len );
137 }
138
139 /**
140  * Copy data between user buffers
141  *
142  * @v dest      Destination user buffer
143  * @v dest_off  Offset within destination buffer
144  * @v src       Source user buffer
145  * @v src_off   Offset within source buffer
146  * @v len       Length
147  */
148 static inline __attribute__ (( always_inline )) void
149 copy_user ( userptr_t dest, off_t dest_off, userptr_t src, off_t src_off,
150             size_t len ) {
151         memcpy ( ( ( void * ) dest + dest_off ), ( ( void * ) src + src_off ),
152                  len );
153 }
154
155 /**
156  * Convert virtual address to user buffer
157  *
158  * @v virtual   Virtual address
159  * @ret buffer  User buffer
160  *
161  * This constructs a user buffer from an ordinary pointer.  Use it
162  * when you need to pass a pointer to an internal buffer to a function
163  * that expects a @c userptr_t.
164  */
165 static inline __attribute__ (( always_inline )) userptr_t
166 virt_to_user ( void * virtual ) {
167         return ( ( intptr_t ) virtual );
168 }
169
170 /**
171  * Convert segment:offset address to user buffer
172  *
173  * @v segment   Real-mode segment
174  * @v offset    Real-mode offset
175  * @ret buffer  User buffer
176  */
177 static inline __attribute__ (( always_inline )) userptr_t
178 real_to_user ( unsigned int segment, unsigned int offset ) {
179         return virt_to_user ( VIRTUAL ( segment, offset ) );
180 }
181
182 /**
183  * Convert physical address to user buffer
184  *
185  * @v physical  Physical address
186  * @ret buffer  User buffer
187  */
188 static inline __attribute__ (( always_inline )) userptr_t
189 phys_to_user ( physaddr_t physical ) {
190         return virt_to_user ( phys_to_virt ( physical ) );
191 }
192
193 /* Copy to/from real-mode stack */
194 extern uint16_t copy_to_rm_stack ( void *data, size_t size );
195 extern void remove_from_rm_stack ( void *data, size_t size );
196
197 /* Place/remove parameter on real-mode stack in a way that's
198  * compatible with libkir
199  */
200 #define BASEMEM_PARAMETER_INIT_LIBRM( param ) \
201         copy_to_rm_stack ( & ( param ), sizeof ( param ) )
202 #define BASEMEM_PARAMETER_DONE_LIBRM( param ) \
203         remove_from_rm_stack ( & ( param ), sizeof ( param ) )
204 #define BASEMEM_PARAMETER_INIT BASEMEM_PARAMETER_INIT_LIBRM
205 #define BASEMEM_PARAMETER_DONE BASEMEM_PARAMETER_DONE_LIBRM
206
207 /* TEXT16_CODE: declare a fragment of code that resides in .text16 */
208 #define TEXT16_CODE( asm_code_str )                     \
209         ".section \".text16\", \"ax\", @progbits\n\t"   \
210         ".code16\n\t"                                   \
211         ".arch i386\n\t"                                \
212         asm_code_str "\n\t"                             \
213         ".code32\n\t"                                   \
214         ".previous\n\t"
215
216 /* REAL_CODE: declare a fragment of code that executes in real mode */
217 #define REAL_CODE( asm_code_str )                       \
218         "pushl $1f\n\t"                                 \
219         "call real_call\n\t"                            \
220         "addl $4, %%esp\n\t"                            \
221         TEXT16_CODE ( "\n1:\n\t"                        \
222                       asm_code_str                      \
223                       "\n\t"                            \
224                       "ret\n\t" )
225
226 #endif /* ASSEMBLY */
227
228 #endif /* LIBRM_H */