?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
kernel32/tests/pipe: Enable compilation with long types.
[wine.git] / dlls / krnl386.exe16 / ne_segment.c
blob3eb8bae2ff3961af858b608a9544eb875df1c530
1 /*
2 * NE segment loading
3 *
4 * Copyright 1993 Robert J. Amstadt
5 * Copyright 1995 Alexandre Julliard
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <sys/types.h>
25 #include <fcntl.h>
26 #include <ctype.h>
27 #include <string.h>
28
29 #include "wine/winbase16.h"
30 #include "wownt32.h"
31 #include "winternl.h"
32 #include "kernel16_private.h"
33 #include "wine/debug.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(fixup);
36 WINE_DECLARE_DEBUG_CHANNEL(dll);
37 WINE_DECLARE_DEBUG_CHANNEL(module);
38
39 /*
40 * Relocation table entry
41 */
42 struct relocation_entry_s
43 {
44 BYTE address_type; /* Relocation address type */
45 BYTE relocation_type; /* Relocation type */
46 WORD offset; /* Offset in segment to fixup */
47 WORD target1; /* Target specification */
48 WORD target2; /* Target specification */
49 };
50
51 /*
52 * Relocation address types
53 */
54 #define NE_RADDR_LOWBYTE 0
55 #define NE_RADDR_SELECTOR 2
56 #define NE_RADDR_POINTER32 3
57 #define NE_RADDR_OFFSET16 5
58 #define NE_RADDR_POINTER48 11
59 #define NE_RADDR_OFFSET32 13
60
61 /*
62 * Relocation types
63 */
64 #define NE_RELTYPE_INTERNAL 0
65 #define NE_RELTYPE_ORDINAL 1
66 #define NE_RELTYPE_NAME 2
67 #define NE_RELTYPE_OSFIXUP 3
68 #define NE_RELFLAG_ADDITIVE 4
69
70 /* Self-loading modules contain this structure in their first segment */
71 typedef struct
72 {
73 WORD version; /* Must be "A0" (0x3041) */
74 WORD reserved;
75 FARPROC16 BootApp; /* startup procedure */
76 FARPROC16 LoadAppSeg; /* procedure to load a segment */
77 FARPROC16 reserved2;
78 FARPROC16 MyAlloc; /* memory allocation procedure,
79 * wine must write this field */
80 FARPROC16 EntryAddrProc;
81 FARPROC16 ExitProc; /* exit procedure */
82 WORD reserved3[4];
83 FARPROC16 SetOwner; /* Set Owner procedure, exported by wine */
84 } SELFLOADHEADER;
85
86 #define SEL(x) ((x)|1)
87
88 static void NE_FixupSegmentPrologs(NE_MODULE *pModule, WORD segnum);
89
90
91 /***********************************************************************
92 * NE_GetRelocAddrName
93 */
94 static const char *NE_GetRelocAddrName( BYTE addr_type, int additive )
95 {
96 switch(addr_type & 0x7f)
97 {
98 case NE_RADDR_LOWBYTE: return additive ? "BYTE add" : "BYTE";
99 case NE_RADDR_OFFSET16: return additive ? "OFFSET16 add" : "OFFSET16";
100 case NE_RADDR_POINTER32: return additive ? "POINTER32 add" : "POINTER32";
101 case NE_RADDR_SELECTOR: return additive ? "SELECTOR add" : "SELECTOR";
102 case NE_RADDR_POINTER48: return additive ? "POINTER48 add" : "POINTER48";
103 case NE_RADDR_OFFSET32: return additive ? "OFFSET32 add" : "OFFSET32";
104 }
105 return "???";
106 }
107
108
109 /***********************************************************************
110 * NE_OpenFile
111 */
112 static HFILE16 NE_OpenFile( NE_MODULE *pModule )
113 {
114 char *name = NE_MODULE_NAME( pModule );
115 HANDLE handle = CreateFileA( name, GENERIC_READ, FILE_SHARE_READ,
116 NULL, OPEN_EXISTING, 0, 0 );
117
118 if (handle == INVALID_HANDLE_VALUE)
119 {
120 ERR( "Can't open file '%s' for module %04x\n", name, pModule->self );
121 return HFILE_ERROR;
122 }
123 return Win32HandleToDosFileHandle( handle );
124 }
125
126
127 /***********************************************************************
128 * apply_relocations
129 *
130 * Apply relocations to a segment. Helper for NE_LoadSegment.
131 */
132 static inline BOOL apply_relocations( NE_MODULE *pModule, const struct relocation_entry_s *rep,
133 int count, int segnum )
134 {
135 BYTE *func_name;
136 char buffer[256];
137 int i, ordinal;
138 WORD offset, *sp;
139 HMODULE16 module;
140 FARPROC16 address = 0;
141 HMODULE16 *pModuleTable = (HMODULE16 *)((char *)pModule + pModule->ne_modtab);
142 SEGTABLEENTRY *pSegTable = NE_SEG_TABLE( pModule );
143 SEGTABLEENTRY *pSeg = pSegTable + segnum - 1;
144
145 /*
146 * Go through the relocation table one entry at a time.
147 */
148 for (i = 0; i < count; i++, rep++)
149 {
150 /*
151 * Get the target address corresponding to this entry.
152 */
153
154 /* If additive, there is no target chain list. Instead, add source
155 and target */
156 int additive = rep->relocation_type & NE_RELFLAG_ADDITIVE;
157 switch (rep->relocation_type & 3)
158 {
159 case NE_RELTYPE_ORDINAL:
160 module = pModuleTable[rep->target1-1];
161 ordinal = rep->target2;
162 address = NE_GetEntryPoint( module, ordinal );
163 if (!address)
164 {
165 NE_MODULE *pTarget = NE_GetPtr( module );
166 if (!pTarget)
167 WARN_(module)("Module not found: %04x, reference %d of module %*.*s\n",
168 module, rep->target1,
169 *((BYTE *)pModule + pModule->ne_restab),
170 *((BYTE *)pModule + pModule->ne_restab),
171 (char *)pModule + pModule->ne_restab + 1 );
172 else
173 {
174 ERR("No implementation for %.*s.%d, setting to 0xdeadbeef\n",
175 *((BYTE *)pTarget + pTarget->ne_restab),
176 (char *)pTarget + pTarget->ne_restab + 1,
177 ordinal );
178 address = (FARPROC16)0xdeadbeef;
179 }
180 }
181 if (TRACE_ON(fixup))
182 {
183 NE_MODULE *pTarget = NE_GetPtr( module );
184 TRACE("%d: %.*s.%d=%04x:%04x %s\n", i + 1,
185 *((BYTE *)pTarget + pTarget->ne_restab),
186 (char *)pTarget + pTarget->ne_restab + 1,
187 ordinal, HIWORD(address), LOWORD(address),
188 NE_GetRelocAddrName( rep->address_type, additive ) );
189 }
190 break;
191
192 case NE_RELTYPE_NAME:
193 module = pModuleTable[rep->target1-1];
194 func_name = (BYTE *)pModule + pModule->ne_imptab + rep->target2;
195 memcpy( buffer, func_name+1, *func_name );
196 buffer[*func_name] = '\0';
197 ordinal = NE_GetOrdinal( module, buffer );
198 address = NE_GetEntryPoint( module, ordinal );
199
200 if (ERR_ON(fixup) && !address)
201 {
202 NE_MODULE *pTarget = NE_GetPtr( module );
203 ERR("No implementation for %.*s.%s, setting to 0xdeadbeef\n",
204 *((BYTE *)pTarget + pTarget->ne_restab),
205 (char *)pTarget + pTarget->ne_restab + 1, buffer );
206 }
207 if (!address) address = (FARPROC16) 0xdeadbeef;
208 if (TRACE_ON(fixup))
209 {
210 NE_MODULE *pTarget = NE_GetPtr( module );
211 TRACE("%d: %.*s.%s=%04x:%04x %s\n", i + 1,
212 *((BYTE *)pTarget + pTarget->ne_restab),
213 (char *)pTarget + pTarget->ne_restab + 1,
214 buffer, HIWORD(address), LOWORD(address),
215 NE_GetRelocAddrName( rep->address_type, additive ) );
216 }
217 break;
218
219 case NE_RELTYPE_INTERNAL:
220 if ((rep->target1 & 0xff) == 0xff)
221 {
222 address = NE_GetEntryPoint( pModule->self, rep->target2 );
223 }
224 else
225 {
226 address = (FARPROC16)MAKESEGPTR( SEL(pSegTable[rep->target1-1].hSeg), rep->target2 );
227 }
228
229 TRACE("%d: %04x:%04x %s\n",
230 i + 1, HIWORD(address), LOWORD(address),
231 NE_GetRelocAddrName( rep->address_type, additive ) );
232 break;
233
234 case NE_RELTYPE_OSFIXUP:
235 /* Relocation type 7:
236 *
237 * These appear to be used as fixups for the Windows
238 * floating point emulator. Let's just ignore them and
239 * try to use the hardware floating point. Linux should
240 * successfully emulate the coprocessor if it doesn't
241 * exist.
242 */
243 TRACE("%d: TYPE %d, OFFSET %04x, TARGET %04x %04x %s\n",
244 i + 1, rep->relocation_type, rep->offset,
245 rep->target1, rep->target2,
246 NE_GetRelocAddrName( rep->address_type, additive ) );
247 continue;
248 }
249
250 offset = rep->offset;
251
252 /* Apparently, high bit of address_type is sometimes set; */
253 /* we ignore it for now */
254 if (rep->address_type > NE_RADDR_OFFSET32)
255 {
256 char module[10];
257 GetModuleName16( pModule->self, module, sizeof(module) );
258 ERR("WARNING: module %s: unknown reloc addr type = 0x%02x. Please report.\n",
259 module, rep->address_type );
260 }
261
262 if (additive)
263 {
264 sp = MapSL( MAKESEGPTR( SEL(pSeg->hSeg), offset ) );
265 TRACE(" %04x:%04x\n", offset, *sp );
266 switch (rep->address_type & 0x7f)
267 {
268 case NE_RADDR_LOWBYTE:
269 *(BYTE *)sp += LOBYTE((int)address);
270 break;
271 case NE_RADDR_OFFSET16:
272 *sp += LOWORD(address);
273 break;
274 case NE_RADDR_POINTER32:
275 *sp += LOWORD(address);
276 *(sp+1) = HIWORD(address);
277 break;
278 case NE_RADDR_SELECTOR:
279 /* Borland creates additive records with offset zero. Strange, but OK */
280 if (*sp)
281 ERR("Additive selector to %04x.Please report\n",*sp);
282 else
283 *sp = HIWORD(address);
284 break;
285 default:
286 goto unknown;
287 }
288 }
289 else /* non-additive fixup */
290 {
291 do
292 {
293 WORD next_offset;
294
295 sp = MapSL( MAKESEGPTR( SEL(pSeg->hSeg), offset ) );
296 next_offset = *sp;
297 TRACE(" %04x:%04x\n", offset, *sp );
298 switch (rep->address_type & 0x7f)
299 {
300 case NE_RADDR_LOWBYTE:
301 *(BYTE *)sp = LOBYTE((int)address);
302 break;
303 case NE_RADDR_OFFSET16:
304 *sp = LOWORD(address);
305 break;
306 case NE_RADDR_POINTER32:
307 *(FARPROC16 *)sp = address;
308 break;
309 case NE_RADDR_SELECTOR:
310 *sp = SELECTOROF(address);
311 break;
312 default:
313 goto unknown;
314 }
315 if (next_offset == offset) break; /* avoid infinite loop */
316 if (next_offset >= GlobalSize16(pSeg->hSeg)) break;
317 offset = next_offset;
318 } while (offset != 0xffff);
319 }
320 }
321 return TRUE;
322
323 unknown:
324 WARN("WARNING: %d: unknown ADDR TYPE %d, "
325 "TYPE %d, OFFSET %04x, TARGET %04x %04x\n",
326 i + 1, rep->address_type, rep->relocation_type,
327 rep->offset, rep->target1, rep->target2);
328 return FALSE;
329 }
330
331
332 /***********************************************************************
333 * NE_LoadSegment
334 */
335 BOOL NE_LoadSegment( NE_MODULE *pModule, WORD segnum )
336 {
337 WORD count;
338 DWORD pos;
339 const struct relocation_entry_s *rep;
340 int size;
341 SEGTABLEENTRY *pSegTable = NE_SEG_TABLE( pModule );
342 SEGTABLEENTRY *pSeg = pSegTable + segnum - 1;
343
344 if (pSeg->flags & NE_SEGFLAGS_LOADED)
345 {
346 /* self-loader ? -> already loaded it */
347 if (pModule->ne_flags & NE_FFLAGS_SELFLOAD)
348 return TRUE;
349
350 /* leave, except for DGROUP, as this may be the second instance */
351 if (segnum != pModule->ne_autodata)
352 return TRUE;
353 }
354
355 if (!pSeg->filepos) return TRUE; /* No file image, just return */
356
357 TRACE_(module)("Loading segment %d, hSeg=%04x, flags=%04x\n",
358 segnum, pSeg->hSeg, pSeg->flags );
359 pos = pSeg->filepos << pModule->ne_align;
360 if (pSeg->size) size = pSeg->size;
361 else size = pSeg->minsize ? pSeg->minsize : 0x10000;
362
363 if (pModule->ne_flags & NE_FFLAGS_SELFLOAD && segnum > 1)
364 {
365 /* Implement self-loading segments */
366 SELFLOADHEADER *selfloadheader;
367 WORD old_ss = CURRENT_SS, old_sp = CURRENT_SP;
368 HFILE16 hFile16;
369 WORD args[3];
370 DWORD ret;
371
372 selfloadheader = MapSL( MAKESEGPTR(SEL(pSegTable->hSeg),0) );
373 CURRENT_SS = pModule->self_loading_sel;
374 CURRENT_SP = 0xff00 - sizeof(STACK16FRAME);
375
376 hFile16 = NE_OpenFile( pModule );
377 TRACE_(dll)("Call LoadAppSegProc(hmodule=0x%04x,hf=%x,segnum=%d)\n",
378 pModule->self,hFile16,segnum );
379 args[2] = pModule->self;
380 args[1] = hFile16;
381 args[0] = segnum;
382 WOWCallback16Ex( (DWORD)selfloadheader->LoadAppSeg, WCB16_PASCAL, sizeof(args), args, &ret );
383 pSeg->hSeg = LOWORD(ret);
384 TRACE_(dll)("Ret LoadAppSegProc: hSeg=0x%04x\n", pSeg->hSeg);
385 _lclose16( hFile16 );
386 CURRENT_SS = old_ss;
387 CURRENT_SP = old_sp;
388
389 pSeg->flags |= NE_SEGFLAGS_LOADED;
390 return TRUE;
391 }
392 else if (!(pSeg->flags & NE_SEGFLAGS_ITERATED))
393 {
394 void *mem = GlobalLock16(pSeg->hSeg);
395 if (!NE_READ_DATA( pModule, mem, pos, size ))
396 return FALSE;
397 pos += size;
398 }
399 else
400 {
401 /*
402 The following bit of code for "iterated segments" was written without
403 any documentation on the format of these segments. It seems to work,
404 but may be missing something.
405 */
406 const char *buff = NE_GET_DATA( pModule, pos, size );
407 const char* curr = buff;
408 char *mem = GlobalLock16(pSeg->hSeg);
409
410 pos += size;
411 if (buff == NULL) return FALSE;
412
413 while(curr < buff + size) {
414 unsigned int rept = ((const short *)curr)[0];
415 unsigned int len = ((const short *)curr)[1];
416
417 curr += 2*sizeof(short);
418 while (rept--)
419 {
420 memcpy( mem, curr, len );
421 mem += len;
422 }
423 curr += len;
424 }
425 }
426
427 pSeg->flags |= NE_SEGFLAGS_LOADED;
428
429 /* Perform exported function prolog fixups */
430 NE_FixupSegmentPrologs( pModule, segnum );
431
432 if (!(pSeg->flags & NE_SEGFLAGS_RELOC_DATA))
433 return TRUE; /* No relocation data, we are done */
434
435 if (!NE_READ_DATA( pModule, &count, pos, sizeof(count) ) || !count) return TRUE;
436 pos += sizeof(count);
437
438 TRACE("Fixups for %.*s, segment %d, hSeg %04x\n",
439 *((BYTE *)pModule + pModule->ne_restab),
440 (char *)pModule + pModule->ne_restab + 1,
441 segnum, pSeg->hSeg );
442
443 if (!(rep = NE_GET_DATA( pModule, pos, count * sizeof(struct relocation_entry_s) )))
444 return FALSE;
445
446 return apply_relocations( pModule, rep, count, segnum );
447 }
448
449
450 /***********************************************************************
451 * NE_LoadAllSegments
452 */
453 BOOL NE_LoadAllSegments( NE_MODULE *pModule )
454 {
455 int i;
456 SEGTABLEENTRY * pSegTable = NE_SEG_TABLE(pModule);
457
458 if (pModule->ne_flags & NE_FFLAGS_SELFLOAD)
459 {
460 HFILE16 hFile16;
461 HGLOBAL16 sel;
462 /* Handle self-loading modules */
463 SELFLOADHEADER *selfloadheader;
464 HMODULE16 mod = GetModuleHandle16("KERNEL");
465 WORD old_ss = CURRENT_SS, old_sp = CURRENT_SP;
466 WORD args[2];
467
468 TRACE_(module)("%.*s is a self-loading module!\n",
469 *((BYTE*)pModule + pModule->ne_restab),
470 (char *)pModule + pModule->ne_restab + 1);
471 if (!NE_LoadSegment( pModule, 1 )) return FALSE;
472 selfloadheader = MapSL( MAKESEGPTR(SEL(pSegTable->hSeg), 0) );
473 selfloadheader->EntryAddrProc = GetProcAddress16(mod,"EntryAddrProc");
474 selfloadheader->MyAlloc = GetProcAddress16(mod,"MyAlloc");
475 selfloadheader->SetOwner = GetProcAddress16(mod,"FarSetOwner");
476 sel = GlobalAlloc16( GMEM_ZEROINIT, 0xFF00 );
477 pModule->self_loading_sel = SEL(sel);
478 FarSetOwner16( sel, pModule->self );
479 CURRENT_SS = pModule->self_loading_sel;
480 CURRENT_SP = 0xff00 - sizeof(STACK16FRAME);
481
482 hFile16 = NE_OpenFile(pModule);
483 TRACE_(dll)("CallBootAppProc(hModule=0x%04x,hf=0x%04x)\n",
484 pModule->self,hFile16);
485 args[1] = pModule->self;
486 args[0] = hFile16;
487 WOWCallback16Ex( (DWORD)selfloadheader->BootApp, WCB16_PASCAL, sizeof(args), args, NULL );
488 TRACE_(dll)("Return from CallBootAppProc\n");
489 _lclose16(hFile16);
490 CURRENT_SS = old_ss;
491 CURRENT_SP = old_sp;
492
493 for (i = 2; i <= pModule->ne_cseg; i++)
494 if (!NE_LoadSegment( pModule, i )) return FALSE;
495 }
496 else
497 {
498 for (i = 1; i <= pModule->ne_cseg; i++)
499 if (!NE_LoadSegment( pModule, i )) return FALSE;
500 }
501 return TRUE;
502 }
503
504
505 /***********************************************************************
506 * NE_FixupSegmentPrologs
507 *
508 * Fixup exported functions prologs of one segment
509 */
510 static void NE_FixupSegmentPrologs(NE_MODULE *pModule, WORD segnum)
511 {
512 SEGTABLEENTRY *pSegTable = NE_SEG_TABLE( pModule );
513 ET_BUNDLE *bundle;
514 ET_ENTRY *entry;
515 WORD dgroup, num_entries, sel = SEL(pSegTable[segnum-1].hSeg);
516 BYTE *pSeg, *pFunc;
517
518 TRACE("(%d);\n", segnum);
519
520 if (pSegTable[segnum-1].flags & NE_SEGFLAGS_DATA)
521 {
522 pSegTable[segnum-1].flags |= NE_SEGFLAGS_LOADED;
523 return;
524 }
525
526 if (!pModule->ne_autodata) return;
527
528 if (!pSegTable[pModule->ne_autodata-1].hSeg) return;
529 dgroup = SEL(pSegTable[pModule->ne_autodata-1].hSeg);
530
531 pSeg = MapSL( MAKESEGPTR(sel, 0) );
532
533 bundle = (ET_BUNDLE *)((BYTE *)pModule+pModule->ne_enttab);
534
535 do {
536 TRACE("num_entries: %d, bundle: %p, next: %04x, pSeg: %p\n", bundle->last - bundle->first, bundle, bundle->next, pSeg);
537 if (!(num_entries = bundle->last - bundle->first))
538 return;
539 entry = (ET_ENTRY *)((BYTE *)bundle+6);
540 while (num_entries--)
541 {
542 /*TRACE("entry: %p, entry->segnum: %d, entry->offs: %04x\n", entry, entry->segnum, entry->offs);*/
543 if (entry->segnum == segnum)
544 {
545 pFunc = pSeg+entry->offs;
546 TRACE("pFunc: %p, *(DWORD *)pFunc: %08lx, num_entries: %d\n", pFunc, *(DWORD *)pFunc, num_entries);
547 if (*(pFunc+2) == 0x90)
548 {
549 if (*(WORD *)pFunc == 0x581e) /* push ds, pop ax */
550 {
551 TRACE("patch %04x:%04x -> mov ax, ds\n", sel, entry->offs);
552 *(WORD *)pFunc = 0xd88c; /* mov ax, ds */
553 }
554
555 if (*(WORD *)pFunc == 0xd88c)
556 {
557 if ((entry->flags & 2)) /* public data ? */
558 {
559 TRACE("patch %04x:%04x -> mov ax, dgroup [%04x]\n", sel, entry->offs, dgroup);
560 *pFunc = 0xb8; /* mov ax, */
561 *(WORD *)(pFunc+1) = dgroup;
562 }
563 else if ((pModule->ne_flags & NE_FFLAGS_MULTIPLEDATA)
564 && (entry->flags & 1)) /* exported ? */
565 {
566 TRACE("patch %04x:%04x -> nop, nop\n", sel, entry->offs);
567 *(WORD *)pFunc = 0x9090; /* nop, nop */
568 }
569 }
570 }
571 }
572 entry++;
573 }
574 } while ( (bundle->next) && (bundle = ((ET_BUNDLE *)((BYTE *)pModule + bundle->next))) );
575 }
576
577
578 /***********************************************************************
579 * PatchCodeHandle (KERNEL.110)
580 *
581 * Needed for self-loading modules.
582 */
583 DWORD WINAPI PatchCodeHandle16(HANDLE16 hSeg)
584 {
585 WORD segnum;
586 WORD sel = SEL(hSeg);
587 NE_MODULE *pModule = NE_GetPtr(FarGetOwner16(sel));
588 SEGTABLEENTRY *pSegTable = NE_SEG_TABLE(pModule);
589
590 TRACE_(module)("(%04x);\n", hSeg);
591
592 /* find the segment number of the module that belongs to hSeg */
593 for (segnum = 1; segnum <= pModule->ne_cseg; segnum++)
594 {
595 if (SEL(pSegTable[segnum-1].hSeg) == sel)
596 {
597 NE_FixupSegmentPrologs(pModule, segnum);
598 break;
599 }
600 }
601
602 return MAKELONG(hSeg, sel);
603 }
604
605
606 /***********************************************************************
607 * NE_GetDLLInitParams
608 */
609 static VOID NE_GetDLLInitParams( NE_MODULE *pModule,
610 WORD *hInst, WORD *ds, WORD *heap )
611 {
612 SEGTABLEENTRY *pSegTable = NE_SEG_TABLE( pModule );
613
614 if (!(pModule->ne_flags & NE_FFLAGS_SINGLEDATA))
615 {
616 if (pModule->ne_flags & NE_FFLAGS_MULTIPLEDATA || pModule->ne_autodata)
617 {
618 /* Not SINGLEDATA */
619 ERR_(dll)("Library is not marked SINGLEDATA\n");
620 }
621 /* DATA NONE DLL */
622 *ds = 0;
623 *heap = 0;
624 }
625 else /* DATA SINGLE DLL */
626 {
627 if (pModule->ne_autodata) {
628 *ds = SEL(pSegTable[pModule->ne_autodata-1].hSeg);
629 *heap = pModule->ne_heap;
630 }
631 else /* hmm, DLL has no dgroup,
632 but why has it NE_FFLAGS_SINGLEDATA set ?
633 Buggy DLL compiler ? */
634 {
635 *ds = 0;
636 *heap = 0;
637 }
638 }
639
640 *hInst = *ds ? GlobalHandle16(*ds) : pModule->self;
641 }
642
643
644 /***********************************************************************
645 * NE_InitDLL
646 *
647 * Call the DLL initialization code
648 */
649 static BOOL NE_InitDLL( NE_MODULE *pModule )
650 {
651 SEGTABLEENTRY *pSegTable;
652 WORD hInst, ds, heap;
653 CONTEXT context;
654
655 pSegTable = NE_SEG_TABLE( pModule );
656
657 if (!(pModule->ne_flags & NE_FFLAGS_LIBMODULE) ||
658 (pModule->ne_flags & NE_FFLAGS_WIN32)) return TRUE; /*not a library*/
659
660 /* Call USER signal handler for Win3.1 compatibility. */
661 NE_CallUserSignalProc( pModule->self, USIG16_DLL_LOAD );
662
663 if (!SELECTOROF(pModule->ne_csip)) return TRUE; /* no initialization code */
664
665
666 /* Registers at initialization must be:
667 * cx heap size
668 * di library instance
669 * ds data segment if any
670 * es:si command line (always 0)
671 */
672
673 memset( &context, 0, sizeof(context) );
674
675 NE_GetDLLInitParams( pModule, &hInst, &ds, &heap );
676
677 context.Ecx = heap;
678 context.Edi = hInst;
679 context.SegDs = ds;
680 context.SegEs = ds; /* who knows ... */
681 context.SegCs = SEL(pSegTable[SELECTOROF(pModule->ne_csip)-1].hSeg);
682 context.Eip = OFFSETOF(pModule->ne_csip);
683 context.Ebp = CURRENT_SP + FIELD_OFFSET(STACK16FRAME,bp);
684
685 pModule->ne_csip = 0; /* Don't initialize it twice */
686 TRACE_(dll)("Calling LibMain for %.*s, cs:ip=%04lx:%04lx ds=%04lx di=%04x cx=%04x\n",
687 *((BYTE*)pModule + pModule->ne_restab),
688 (char *)pModule + pModule->ne_restab + 1,
689 context.SegCs, context.Eip, context.SegDs,
690 LOWORD(context.Edi), LOWORD(context.Ecx) );
691 WOWCallback16Ex( 0, WCB16_REGS, 0, NULL, (DWORD *)&context );
692 return TRUE;
693 }
694
695 /***********************************************************************
696 * NE_InitializeDLLs
697 *
698 * Recursively initialize all DLLs (according to the order in which
699 * they where loaded).
700 */
701 void NE_InitializeDLLs( HMODULE16 hModule )
702 {
703 NE_MODULE *pModule;
704 HMODULE16 *pDLL;
705
706 if (!(pModule = NE_GetPtr( hModule ))) return;
707
708 if (pModule->dlls_to_init)
709 {
710 HGLOBAL16 to_init = pModule->dlls_to_init;
711 pModule->dlls_to_init = 0;
712 for (pDLL = GlobalLock16( to_init ); *pDLL; pDLL++)
713 {
714 NE_InitializeDLLs( *pDLL );
715 }
716 GlobalFree16( to_init );
717 }
718 NE_InitDLL( pModule );
719 }
720
721
722 /**********************************************************************
723 * NE_CallUserSignalProc
724 *
725 * According to "Undocumented Windows", the task signal proc is
726 * bypassed for module load/unload notifications, and the USER signal
727 * proc is called directly instead. This is what this function does.
728 */
729 typedef DWORD (WINAPI *pSignalProc)( HANDLE16 module, UINT16 code, UINT16 exit,
730 HINSTANCE16 inst, HQUEUE16 queue );
731
732 void NE_CallUserSignalProc( HMODULE16 hModule, UINT16 code )
733 {
734 FARPROC16 proc;
735 HMODULE16 user = GetModuleHandle16("user.exe");
736
737 if (!user) return;
738 if ((proc = GetProcAddress16( user, "SignalProc" )))
739 {
740 /* USER is always a builtin dll */
741 pSignalProc sigproc = (pSignalProc)((ENTRYPOINT16 *)MapSL( (SEGPTR)proc ))->target;
742 sigproc( hModule, code, 0, 0, 0 );
743 }
744 }
745
746
747 /***********************************************************************
748 * NE_CallDllEntryPoint
749 *
750 * Call the DllEntryPoint of DLLs with subsystem >= 4.0
751 */
752 typedef DWORD (WINAPI *WinNEEntryProc)(DWORD,WORD,WORD,WORD,DWORD,WORD);
753
754 static void NE_CallDllEntryPoint( NE_MODULE *pModule, DWORD dwReason )
755 {
756 WORD hInst, ds, heap;
757 FARPROC16 entryPoint;
758
759 if (!(pModule->ne_flags & NE_FFLAGS_LIBMODULE)) return;
760 if (!(pModule->ne_flags & NE_FFLAGS_BUILTIN) && pModule->ne_expver < 0x0400) return;
761 if (!(entryPoint = GetProcAddress16( pModule->self, "DllEntryPoint" ))) return;
762
763 NE_GetDLLInitParams( pModule, &hInst, &ds, &heap );
764
765 TRACE_(dll)( "Calling %s DllEntryPoint, cs:ip=%04x:%04x\n",
766 NE_MODULE_NAME( pModule ),
767 SELECTOROF(entryPoint), OFFSETOF(entryPoint) );
768
769 if ( pModule->ne_flags & NE_FFLAGS_BUILTIN )
770 {
771 WinNEEntryProc entryProc = (WinNEEntryProc)((ENTRYPOINT16 *)MapSL( (SEGPTR)entryPoint ))->target;
772
773 entryProc( dwReason, hInst, ds, heap, 0, 0 );
774 }
775 else
776 {
777 CONTEXT context;
778 WORD args[8];
779
780 memset( &context, 0, sizeof(context) );
781 context.SegDs = ds;
782 context.SegEs = ds; /* who knows ... */
783 context.SegCs = HIWORD(entryPoint);
784 context.Eip = LOWORD(entryPoint);
785 context.Ebp = CURRENT_SP + FIELD_OFFSET(STACK16FRAME,bp);
786
787 args[7] = HIWORD(dwReason);
788 args[6] = LOWORD(dwReason);
789 args[5] = hInst;
790 args[4] = ds;
791 args[3] = heap;
792 args[2] = 0; /* HIWORD(dwReserved1) */
793 args[1] = 0; /* LOWORD(dwReserved1) */
794 args[0] = 0; /* wReserved2 */
795 WOWCallback16Ex( 0, WCB16_REGS, sizeof(args), args, (DWORD *)&context );
796 }
797 }
798
799 /***********************************************************************
800 * NE_DllProcessAttach
801 *
802 * Call the DllEntryPoint of all modules this one (recursively)
803 * depends on, according to the order in which they were loaded.
804 *
805 * Note that --as opposed to the PE module case-- there is no notion
806 * of 'module loaded into a process' for NE modules, and hence we
807 * have no place to store the fact that the DllEntryPoint of a
808 * given module was already called on behalf of this process (e.g.
809 * due to some earlier LoadLibrary16 call).
810 *
811 * Thus, we just call the DllEntryPoint twice in that case. Win9x
812 * appears to behave this way as well ...
813 *
814 * This routine must only be called with the Win16Lock held.
815 *
816 * FIXME: We should actually abort loading in case the DllEntryPoint
817 * returns FALSE ...
818 *
819 */
820
821 struct ne_init_list
822 {
823 int count;
824 int size;
825 NE_MODULE **module;
826 };
827
828 static void add_to_init_list( struct ne_init_list *list, NE_MODULE *hModule )
829 {
830 NE_MODULE **newModule = NULL;
831 if ( list->count == list->size )
832 {
833 int newSize = list->size + 128;
834
835 if (list->module)
836 newModule = HeapReAlloc( GetProcessHeap(), 0,
837 list->module, newSize*sizeof(NE_MODULE *) );
838 else
839 newModule = HeapAlloc( GetProcessHeap(), 0,
840 newSize*sizeof(NE_MODULE *) );
841 if ( !newModule )
842 {
843 FIXME_(dll)("Out of memory!\n");
844 return;
845 }
846
847 list->module = newModule;
848 list->size = newSize;
849 }
850
851 list->module[list->count++] = hModule;
852 }
853
854 static void free_init_list( struct ne_init_list *list )
855 {
856 if ( list->module )
857 {
858 HeapFree( GetProcessHeap(), 0, list->module );
859 memset( list, 0, sizeof(*list) );
860 }
861 }
862
863 static void fill_init_list( struct ne_init_list *list, HMODULE16 hModule )
864 {
865 NE_MODULE *pModule;
866 HMODULE16 *pModRef;
867 int i;
868
869 if (!(pModule = NE_GetPtr( hModule ))) return;
870
871 /* Never add a module twice */
872 for ( i = 0; i < list->count; i++ )
873 if ( list->module[i] == pModule )
874 return;
875
876 /* Check for recursive call */
877 if ( pModule->ne_flagsothers & 0x80 ) return;
878
879 TRACE_(dll)("(%s) - START\n", NE_MODULE_NAME(pModule) );
880
881 /* Tag current module to prevent recursive loop */
882 pModule->ne_flagsothers |= 0x80;
883
884 /* Recursively attach all DLLs this one depends on */
885 pModRef = (HMODULE16 *)((char *)pModule + pModule->ne_modtab);
886 for ( i = 0; i < pModule->ne_cmod; i++ )
887 if ( pModRef[i] ) fill_init_list( list, pModRef[i] );
888
889 /* Add current module */
890 add_to_init_list( list, pModule );
891
892 /* Remove recursion flag */
893 pModule->ne_flagsothers &= ~0x80;
894
895 TRACE_(dll)("(%s) - END\n", NE_MODULE_NAME(pModule) );
896 }
897
898 static void call_init_list( struct ne_init_list *list )
899 {
900 int i;
901 for ( i = 0; i < list->count; i++ )
902 NE_CallDllEntryPoint( list->module[i], DLL_PROCESS_ATTACH );
903 }
904
905 void NE_DllProcessAttach( HMODULE16 hModule )
906 {
907 struct ne_init_list list;
908 memset( &list, 0, sizeof(list) );
909
910 fill_init_list( &list, hModule );
911 call_init_list( &list );
912 free_init_list( &list );
913 }
914
915
916 /***********************************************************************
917 * NE_Ne2MemFlags
918 *
919 * This function translates NE segment flags to GlobalAlloc flags
920 */
921 static WORD NE_Ne2MemFlags(WORD flags)
922 {
923 WORD memflags = 0;
924 #if 1
925 if (flags & NE_SEGFLAGS_DISCARDABLE)
926 memflags |= GMEM_DISCARDABLE;
927 if (flags & NE_SEGFLAGS_MOVEABLE ||
928 ( ! (flags & NE_SEGFLAGS_DATA) &&
929 ! (flags & NE_SEGFLAGS_LOADED) &&
930 ! (flags & NE_SEGFLAGS_ALLOCATED)
931 )
932 )
933 memflags |= GMEM_MOVEABLE;
934 memflags |= GMEM_ZEROINIT;
935 #else
936 memflags = GMEM_ZEROINIT | GMEM_FIXED;
937 #endif
938 return memflags;
939 }
940
941 /***********************************************************************
942 * MyAlloc (KERNEL.668) Wine-specific export
943 *
944 * MyAlloc() function for self-loading apps.
945 */
946 DWORD WINAPI MyAlloc16( WORD wFlags, WORD wSize, WORD wElem )
947 {
948 WORD size = wSize << wElem;
949 HANDLE16 hMem = 0;
950
951 if (wSize || (wFlags & NE_SEGFLAGS_MOVEABLE))
952 hMem = GlobalAlloc16( NE_Ne2MemFlags(wFlags), size);
953
954 if ( ((wFlags & 0x7) != 0x1) && /* DATA */
955 ((wFlags & 0x7) != 0x7) ) /* DATA|ALLOCATED|LOADED */
956 {
957 WORD hSel = SEL(hMem);
958 WORD access = SelectorAccessRights16(hSel,0,0);
959
960 access |= 2<<2; /* SEGMENT_CODE */
961 SelectorAccessRights16(hSel,1,access);
962 }
963 if (size)
964 return MAKELONG( hMem, SEL(hMem) );
965 else
966 return MAKELONG( 0, hMem );
967 }
968
969 /***********************************************************************
970 * NE_GetInstance
971 */
972 HINSTANCE16 NE_GetInstance( NE_MODULE *pModule )
973 {
974 if ( !pModule->ne_autodata )
975 return pModule->self;
976 else
977 {
978 SEGTABLEENTRY *pSeg;
979 pSeg = NE_SEG_TABLE( pModule ) + pModule->ne_autodata - 1;
980 return pSeg->hSeg;
981 }
982 }
983
984 /***********************************************************************
985 * NE_CreateSegment
986 */
987 BOOL NE_CreateSegment( NE_MODULE *pModule, int segnum )
988 {
989 SEGTABLEENTRY *pSeg = NE_SEG_TABLE( pModule ) + segnum - 1;
990 int minsize;
991 unsigned char selflags;
992
993 if ( segnum < 1 || segnum > pModule->ne_cseg )
994 return FALSE;
995
996 if ( (pModule->ne_flags & NE_FFLAGS_SELFLOAD) && segnum != 1 )
997 return TRUE; /* selfloader allocates segment itself */
998
999 if ( (pSeg->flags & NE_SEGFLAGS_ALLOCATED) && segnum != pModule->ne_autodata )
1000 return TRUE; /* all but DGROUP only allocated once */
1001
1002 minsize = pSeg->minsize ? pSeg->minsize : 0x10000;
1003 if ( segnum == SELECTOROF(pModule->ne_sssp) ) minsize += pModule->ne_stack;
1004 if ( segnum == pModule->ne_autodata ) minsize += pModule->ne_heap;
1005
1006 selflags = (pSeg->flags & NE_SEGFLAGS_DATA) ? LDT_FLAGS_DATA : LDT_FLAGS_CODE;
1007 if (pSeg->flags & NE_SEGFLAGS_32BIT) selflags |= LDT_FLAGS_32BIT;
1008 pSeg->hSeg = GLOBAL_Alloc( NE_Ne2MemFlags(pSeg->flags), minsize, pModule->self, selflags );
1009 if (!pSeg->hSeg) return FALSE;
1010
1011 pSeg->flags |= NE_SEGFLAGS_ALLOCATED;
1012 return TRUE;
1013 }
1014
1015 /***********************************************************************
1016 * NE_CreateAllSegments
1017 */
1018 BOOL NE_CreateAllSegments( NE_MODULE *pModule )
1019 {
1020 int i;
1021 for ( i = 1; i <= pModule->ne_cseg; i++ )
1022 if ( !NE_CreateSegment( pModule, i ) )
1023 return FALSE;
1024
1025 pModule->dgroup_entry = pModule->ne_autodata ? pModule->ne_segtab +
1026 (pModule->ne_autodata - 1) * sizeof(SEGTABLEENTRY) : 0;
1027 return TRUE;
1028 }
1029
1030
1031 /**********************************************************************
1032 * IsSharedSelector (KERNEL.345)
1033 */
1034 BOOL16 WINAPI IsSharedSelector16( HANDLE16 selector )
1035 {
1036 /* Check whether the selector belongs to a DLL */
1037 NE_MODULE *pModule = NE_GetPtr( selector );
1038 if (!pModule) return FALSE;
1039 return (pModule->ne_flags & NE_FFLAGS_LIBMODULE) != 0;
1040 }
Windows API implementation