?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Release 960717
[wine/multimedia.git] / memory / local.c
blob9f5c762f76437855aae41a0af9f5c86a77d5420e
1 /*
2 * Local heap functions
3 *
4 * Copyright 1995 Alexandre Julliard
5 * Copyright 1996 Huw Davies
6 */
7
8 /*
9 * Note:
10 * All local heap functions need the current DS as first parameter
11 * when called from the emulation library, so they take one more
12 * parameter than usual.
13 */
14
15 #include <stdlib.h>
16 #include <string.h>
17 #include "windows.h"
18 #include "ldt.h"
19 #include "global.h"
20 #include "heap.h"
21 #include "instance.h"
22 #include "local.h"
23 #include "module.h"
24 #include "stackframe.h"
25 #include "toolhelp.h"
26 #include "stddebug.h"
27 #include "debug.h"
28
29 typedef struct
30 {
31 /* Arena header */
32 WORD prev; /* Previous arena | arena type */
33 WORD next; /* Next arena */
34 /* Start of the memory block or free-list info */
35 WORD size; /* Size of the free block */
36 WORD free_prev; /* Previous free block */
37 WORD free_next; /* Next free block */
38 } LOCALARENA;
39
40 #define ARENA_HEADER_SIZE 4
41 #define ARENA_HEADER( handle) ( ((handle) & ~3) - ARENA_HEADER_SIZE)
42
43 /* Arena types (stored in 'prev' field of the arena) */
44 #define LOCAL_ARENA_FREE 0
45 #define LOCAL_ARENA_FIXED 1
46
47 /* Layout of a handle entry table
48 *
49 * WORD count of entries
50 * LOCALHANDLEENTRY[count] entries
51 * WORD near ptr to next table
52 */
53 typedef struct
54 {
55 WORD addr; /* Address of the MOVEABLE block */
56 BYTE flags; /* Flags for this block */
57 BYTE lock; /* Lock count */
58 } LOCALHANDLEENTRY;
59
60 #pragma pack(1)
61
62 typedef struct
63 {
64 WORD check; /* 00 Heap checking flag */
65 WORD freeze; /* 02 Heap frozen flag */
66 WORD items; /* 04 Count of items on the heap */
67 WORD first; /* 06 First item of the heap */
68 WORD pad1; /* 08 Always 0 */
69 WORD last; /* 0a Last item of the heap */
70 WORD pad2; /* 0c Always 0 */
71 BYTE ncompact; /* 0e Compactions counter */
72 BYTE dislevel; /* 0f Discard level */
73 DWORD distotal; /* 10 Total bytes discarded */
74 WORD htable; /* 14 Pointer to handle table */
75 WORD hfree; /* 16 Pointer to free handle table */
76 WORD hdelta; /* 18 Delta to expand the handle table */
77 WORD expand; /* 1a Pointer to expand function (unused) */
78 WORD pstat; /* 1c Pointer to status structure (unused) */
79 FARPROC16 notify WINE_PACKED; /* 1e Pointer to LocalNotify() function */
80 WORD lock; /* 22 Lock count for the heap */
81 WORD extra; /* 24 Extra bytes to allocate when expanding */
82 WORD minsize; /* 26 Minimum size of the heap */
83 WORD magic; /* 28 Magic number */
84 } LOCALHEAPINFO;
85
86 #pragma pack(4)
87
88 #define LOCAL_HEAP_MAGIC 0x484c /* 'LH' */
89
90
91 /* All local heap allocations are aligned on 4-byte boundaries */
92 #define LALIGN(word) (((word) + 3) & ~3)
93
94 #define ARENA_PTR(ptr,arena) ((LOCALARENA *)((char*)(ptr)+(arena)))
95 #define ARENA_PREV(ptr,arena) (ARENA_PTR((ptr),(arena))->prev & ~3)
96 #define ARENA_NEXT(ptr,arena) (ARENA_PTR((ptr),(arena))->next)
97 #define ARENA_FLAGS(ptr,arena) (ARENA_PTR((ptr),(arena))->prev & 3)
98
99 /* determine whether the handle belongs to a fixed or a moveable block */
100 #define HANDLE_FIXED(handle) (((handle) & 3) == 0)
101 #define HANDLE_MOVEABLE(handle) (((handle) & 3) == 2)
102
103 /***********************************************************************
104 * LOCAL_GetHeap
105 *
106 * Return a pointer to the local heap, making sure it exists.
107 */
108 static LOCALHEAPINFO *LOCAL_GetHeap( HANDLE16 ds )
109 {
110 LOCALHEAPINFO *pInfo;
111 INSTANCEDATA *ptr = (INSTANCEDATA *)PTR_SEG_OFF_TO_LIN( ds, 0 );
112 dprintf_local( stddeb, "Heap at %p, %04x\n", ptr, ptr->heap );
113 if (!ptr || !ptr->heap) return NULL;
114 if (IsBadReadPtr((SEGPTR)MAKELONG( ptr->heap, ds ), sizeof(LOCALHEAPINFO)))
115 return NULL;
116 pInfo = (LOCALHEAPINFO*)((char*)ptr + ptr->heap);
117 if (pInfo->magic != LOCAL_HEAP_MAGIC) return NULL;
118 return pInfo;
119 }
120
121
122 /***********************************************************************
123 * LOCAL_MakeBlockFree
124 *
125 * Make a block free, inserting it in the free-list.
126 * 'block' is the handle of the block arena; 'baseptr' points to
127 * the beginning of the data segment containing the heap.
128 */
129 static void LOCAL_MakeBlockFree( char *baseptr, WORD block )
130 {
131 LOCALARENA *pArena, *pNext;
132 WORD next;
133
134 /* Mark the block as free */
135
136 pArena = ARENA_PTR( baseptr, block );
137 pArena->prev = (pArena->prev & ~3) | LOCAL_ARENA_FREE;
138 pArena->size = pArena->next - block;
139
140 /* Find the next free block (last block is always free) */
141
142 next = pArena->next;
143 for (;;)
144 {
145 pNext = ARENA_PTR( baseptr, next );
146 if ((pNext->prev & 3) == LOCAL_ARENA_FREE) break;
147 next = pNext->next;
148 }
149
150 dprintf_local( stddeb, "Local_AddFreeBlock %04x, next %04x\n", block, next );
151 /* Insert the free block in the free-list */
152
153 pArena->free_prev = pNext->free_prev;
154 pArena->free_next = next;
155 ARENA_PTR(baseptr,pNext->free_prev)->free_next = block;
156 pNext->free_prev = block;
157 }
158
159
160 /***********************************************************************
161 * LOCAL_RemoveFreeBlock
162 *
163 * Remove a block from the free-list.
164 * 'block' is the handle of the block arena; 'baseptr' points to
165 * the beginning of the data segment containing the heap.
166 */
167 static void LOCAL_RemoveFreeBlock( char *baseptr, WORD block )
168 {
169 /* Mark the block as fixed */
170
171 LOCALARENA *pArena = ARENA_PTR( baseptr, block );
172 pArena->prev = (pArena->prev & ~3) | LOCAL_ARENA_FIXED;
173
174 /* Remove it from the list */
175
176 ARENA_PTR(baseptr,pArena->free_prev)->free_next = pArena->free_next;
177 ARENA_PTR(baseptr,pArena->free_next)->free_prev = pArena->free_prev;
178 }
179
180
181 /***********************************************************************
182 * LOCAL_AddBlock
183 *
184 * Insert a new block in the heap.
185 * 'new' is the handle of the new block arena; 'baseptr' points to
186 * the beginning of the data segment containing the heap; 'prev' is
187 * the block before the new one.
188 */
189 static void LOCAL_AddBlock( char *baseptr, WORD prev, WORD new )
190 {
191 LOCALARENA *pPrev = ARENA_PTR( baseptr, prev );
192 LOCALARENA *pNew = ARENA_PTR( baseptr, new );
193
194 pNew->prev = (prev & ~3) | LOCAL_ARENA_FIXED;
195 pNew->next = pPrev->next;
196 ARENA_PTR(baseptr,pPrev->next)->prev &= 3;
197 ARENA_PTR(baseptr,pPrev->next)->prev |= new;
198 pPrev->next = new;
199 }
200
201
202 /***********************************************************************
203 * LOCAL_RemoveBlock
204 *
205 * Remove a block from the heap.
206 * 'block' is the handle of the block arena; 'baseptr' points to
207 * the beginning of the data segment containing the heap.
208 */
209 static void LOCAL_RemoveBlock( char *baseptr, WORD block )
210 {
211 LOCALARENA *pArena, *pTmp;
212
213 /* Remove the block from the free-list */
214
215 dprintf_local( stddeb, "Local_RemoveBlock\n");
216 pArena = ARENA_PTR( baseptr, block );
217 if ((pArena->prev & 3) == LOCAL_ARENA_FREE)
218 LOCAL_RemoveFreeBlock( baseptr, block );
219
220 /* If the previous block is free, expand its size */
221
222 pTmp = ARENA_PTR( baseptr, pArena->prev & ~3 );
223 if ((pTmp->prev & 3) == LOCAL_ARENA_FREE)
224 pTmp->size += pArena->next - block;
225
226 /* Remove the block from the linked list */
227
228 pTmp->next = pArena->next;
229 pTmp = ARENA_PTR( baseptr, pArena->next );
230 pTmp->prev = (pTmp->prev & 3) | (pArena->prev & ~3);
231 }
232
233
234 /***********************************************************************
235 * LOCAL_PrintHeap
236 */
237 static void LOCAL_PrintHeap( HANDLE16 ds )
238 {
239 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
240 LOCALHEAPINFO *pInfo = LOCAL_GetHeap( ds );
241 WORD arena;
242
243 if (!debugging_local) return;
244 if (!pInfo)
245 {
246 printf( "Local Heap corrupted! ds=%04x\n", ds );
247 return;
248 }
249 printf( "Local Heap ds=%04x first=%04x last=%04x items=%d\n",
250 ds, pInfo->first, pInfo->last, pInfo->items );
251
252 arena = pInfo->first;
253 for (;;)
254 {
255 LOCALARENA *pArena = ARENA_PTR(ptr,arena);
256 printf( " %04x: prev=%04x next=%04x type=%d\n", arena,
257 pArena->prev & ~3, pArena->next, pArena->prev & 3 );
258 if (arena == pInfo->first)
259 {
260 printf( " size=%d free_prev=%04x free_next=%04x\n",
261 pArena->size, pArena->free_prev, pArena->free_next );
262 }
263 if ((pArena->prev & 3) == LOCAL_ARENA_FREE)
264 {
265 printf( " size=%d free_prev=%04x free_next=%04x\n",
266 pArena->size, pArena->free_prev, pArena->free_next );
267 if (pArena->next == arena) break; /* last one */
268 if (ARENA_PTR(ptr,pArena->free_next)->free_prev != arena)
269 {
270 printf( "*** arena->free_next->free_prev != arena\n" );
271 break;
272 }
273 }
274 if (pArena->next == arena)
275 {
276 printf( "*** last block is not marked free\n" );
277 break;
278 }
279 if ((ARENA_PTR(ptr,pArena->next)->prev & ~3) != arena)
280 {
281 printf( "*** arena->next->prev != arena (%04x, %04x)\n",
282 pArena->next, ARENA_PTR(ptr,pArena->next)->prev);
283 break;
284 }
285 arena = pArena->next;
286 }
287 }
288
289
290 /***********************************************************************
291 * LocalInit (KERNEL.4)
292 */
293 BOOL16 LocalInit( HANDLE16 selector, WORD start, WORD end )
294 {
295 char *ptr;
296 WORD heapInfoArena, freeArena, lastArena;
297 LOCALHEAPINFO *pHeapInfo;
298 LOCALARENA *pArena, *pFirstArena, *pLastArena;
299 NE_MODULE *pModule;
300
301 /* The initial layout of the heap is: */
302 /* - first arena (FIXED) */
303 /* - heap info structure (FIXED) */
304 /* - large free block (FREE) */
305 /* - last arena (FREE) */
306
307 dprintf_local(stddeb, "LocalInit: %04x %04x-%04x\n", selector, start, end);
308 if (!selector) selector = CURRENT_DS;
309
310 if (debugging_heap)
311 {
312 /* If debugging_heap is set, the global heap blocks are cleared */
313 /* before use, so we can test for double initialization. */
314 if (LOCAL_GetHeap(selector))
315 {
316 fprintf( stderr, "LocalInit: Heap %04x initialized twice.\n", selector);
317 if (debugging_local) LOCAL_PrintHeap(selector);
318 }
319 }
320
321 if (start == 0) {
322 /* Check if the segment is the DGROUP of a module */
323
324 if ((pModule = MODULE_GetPtr( GetExePtr( selector ) )))
325 {
326 SEGTABLEENTRY *pSeg = NE_SEG_TABLE( pModule ) + pModule->dgroup - 1;
327 if (pModule->dgroup && (pSeg->selector == selector)) {
328 /* We can't just use the simple method of using the value
329 * of minsize + stacksize, since there are programs that
330 * resize the data segment before calling InitTask(). So,
331 * we must put it at the end of the segment */
332 start = GlobalSize16( GlobalHandle16( selector ) );
333 start -= end;
334 end += start;
335 dprintf_local( stddeb," new start %04x, minstart: %04x\n", start, pSeg->minsize + pModule->stack_size);
336 }
337 }
338 }
339 ptr = PTR_SEG_OFF_TO_LIN( selector, 0 );
340
341 start = LALIGN( MAX( start, sizeof(INSTANCEDATA) ) );
342 heapInfoArena = LALIGN(start + sizeof(LOCALARENA) );
343 freeArena = LALIGN( heapInfoArena + ARENA_HEADER_SIZE
344 + sizeof(LOCALHEAPINFO) );
345 lastArena = (end - sizeof(LOCALARENA)) & ~3;
346
347 /* Make sure there's enough space. */
348
349 if (freeArena + sizeof(LOCALARENA) >= lastArena) return FALSE;
350
351 /* Initialise the first arena */
352
353 pFirstArena = ARENA_PTR( ptr, start );
354 pFirstArena->prev = start | LOCAL_ARENA_FIXED;
355 pFirstArena->next = heapInfoArena;
356 pFirstArena->size = LALIGN(sizeof(LOCALARENA));
357 pFirstArena->free_prev = start; /* this one */
358 pFirstArena->free_next = freeArena;
359
360 /* Initialise the arena of the heap info structure */
361
362 pArena = ARENA_PTR( ptr, heapInfoArena );
363 pArena->prev = start | LOCAL_ARENA_FIXED;
364 pArena->next = freeArena;
365
366 /* Initialise the heap info structure */
367
368 pHeapInfo = (LOCALHEAPINFO *) (ptr + heapInfoArena + ARENA_HEADER_SIZE );
369 memset( pHeapInfo, 0, sizeof(LOCALHEAPINFO) );
370 pHeapInfo->items = 4;
371 pHeapInfo->first = start;
372 pHeapInfo->last = lastArena;
373 pHeapInfo->htable = 0;
374 pHeapInfo->hdelta = 0x20;
375 pHeapInfo->extra = 0x200;
376 pHeapInfo->minsize = lastArena - freeArena;
377 pHeapInfo->magic = LOCAL_HEAP_MAGIC;
378
379 /* Initialise the large free block */
380
381 pArena = ARENA_PTR( ptr, freeArena );
382 pArena->prev = heapInfoArena | LOCAL_ARENA_FREE;
383 pArena->next = lastArena;
384 pArena->size = lastArena - freeArena;
385 pArena->free_prev = start;
386 pArena->free_next = lastArena;
387
388 /* Initialise the last block */
389
390 pLastArena = ARENA_PTR( ptr, lastArena );
391 pLastArena->prev = freeArena | LOCAL_ARENA_FREE;
392 pLastArena->next = lastArena; /* this one */
393 pLastArena->size = LALIGN(sizeof(LOCALARENA));
394 pLastArena->free_prev = freeArena;
395 pLastArena->free_next = lastArena; /* this one */
396
397 /* Store the local heap address in the instance data */
398
399 ((INSTANCEDATA *)ptr)->heap = heapInfoArena + ARENA_HEADER_SIZE;
400 LOCAL_PrintHeap( selector );
401 return TRUE;
402 }
403
404 /***********************************************************************
405 * LOCAL_GrowHeap
406 */
407 static void LOCAL_GrowHeap( HANDLE16 ds )
408 {
409 HANDLE16 hseg = GlobalHandle16( ds );
410 LONG oldsize = GlobalSize16( hseg );
411 LONG end;
412 LOCALHEAPINFO *pHeapInfo;
413 WORD freeArena, lastArena;
414 LOCALARENA *pArena, *pLastArena;
415 char *ptr;
416
417 /* if nothing can be gained, return */
418 if (oldsize > 0xfff0) return;
419 hseg = GlobalReAlloc16( hseg, 0x10000, GMEM_FIXED );
420 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
421 pHeapInfo = LOCAL_GetHeap( ds );
422 if (pHeapInfo == NULL) {
423 fprintf( stderr, "Local_GrowHeap: heap not found\n" );
424 return;
425 }
426 end = GlobalSize16( hseg );
427 lastArena = (end - sizeof(LOCALARENA)) & ~3;
428
429 /* Update the HeapInfo */
430 pHeapInfo->items++;
431 freeArena = pHeapInfo->last;
432 pHeapInfo->last = lastArena;
433 pHeapInfo->minsize += end - oldsize;
434
435 /* grow the old last block */
436 pArena = ARENA_PTR( ptr, freeArena );
437 pArena->size = lastArena - freeArena;
438 pArena->next = lastArena;
439 pArena->free_next = lastArena;
440
441 /* Initialise the new last block */
442
443 pLastArena = ARENA_PTR( ptr, lastArena );
444 pLastArena->prev = freeArena | LOCAL_ARENA_FREE;
445 pLastArena->next = lastArena; /* this one */
446 pLastArena->size = LALIGN(sizeof(LOCALARENA));
447 pLastArena->free_prev = freeArena;
448 pLastArena->free_next = lastArena; /* this one */
449
450 /* If block before freeArena is also free then merge them */
451 if((ARENA_PTR(ptr, (pArena->prev & ~3))->prev & 3) == LOCAL_ARENA_FREE)
452 {
453 LOCAL_RemoveBlock(ptr, freeArena);
454 pHeapInfo->items--;
455 }
456
457 dprintf_local( stddeb, "Heap expanded\n" );
458 LOCAL_PrintHeap( ds );
459 }
460
461
462 /***********************************************************************
463 * LOCAL_FreeArena
464 */
465 static HLOCAL16 LOCAL_FreeArena( WORD ds, WORD arena )
466 {
467 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
468 LOCALHEAPINFO *pInfo;
469 LOCALARENA *pArena, *pPrev, *pNext;
470
471 dprintf_local( stddeb, "LocalFreeArena: %04x ds=%04x\n", arena, ds );
472 if (!(pInfo = LOCAL_GetHeap( ds ))) return arena;
473
474 pArena = ARENA_PTR( ptr, arena );
475 if ((pArena->prev & 3) == LOCAL_ARENA_FREE)
476 {
477 /* shouldn't happen */
478 fprintf( stderr, "LocalFreeArena: Trying to free block %04x twice!\n",
479 arena );
480 LOCAL_PrintHeap( ds );
481 return arena;
482 }
483
484 /* Check if we can merge with the previous block */
485
486 pPrev = ARENA_PTR( ptr, pArena->prev & ~3 );
487 pNext = ARENA_PTR( ptr, pArena->next );
488 if ((pPrev->prev & 3) == LOCAL_ARENA_FREE)
489 {
490 arena = pArena->prev & ~3;
491 pArena = pPrev;
492 LOCAL_RemoveBlock( ptr, pPrev->next );
493 pInfo->items--;
494 }
495 else /* Make a new free block */
496 {
497 LOCAL_MakeBlockFree( ptr, arena );
498 }
499
500 /* Check if we can merge with the next block */
501
502 if ((pArena->next == pArena->free_next) &&
503 (pArena->next != pInfo->last))
504 {
505 LOCAL_RemoveBlock( ptr, pArena->next );
506 pInfo->items--;
507 }
508 return 0;
509 }
510
511
512 /***********************************************************************
513 * LOCAL_ShrinkArena
514 *
515 * Shrink an arena by creating a free block at its end if possible.
516 * 'size' includes the arena header, and must be aligned.
517 */
518 static void LOCAL_ShrinkArena( WORD ds, WORD arena, WORD size )
519 {
520 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
521 LOCALARENA *pArena = ARENA_PTR( ptr, arena );
522
523 if (arena + size + LALIGN(sizeof(LOCALARENA)) < pArena->next)
524 {
525 LOCALHEAPINFO *pInfo = LOCAL_GetHeap( ds );
526 if (!pInfo) return;
527 LOCAL_AddBlock( ptr, arena, arena + size );
528 pInfo->items++;
529 LOCAL_FreeArena( ds, arena + size );
530 }
531 }
532
533
534 /***********************************************************************
535 * LOCAL_GrowArenaDownward
536 *
537 * Grow an arena downward by using the previous arena (must be free).
538 */
539 static void LOCAL_GrowArenaDownward( WORD ds, WORD arena, WORD newsize )
540 {
541 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
542 LOCALHEAPINFO *pInfo;
543 LOCALARENA *pArena = ARENA_PTR( ptr, arena );
544 WORD prevArena = pArena->prev & ~3;
545 LOCALARENA *pPrevArena = ARENA_PTR( ptr, prevArena );
546 WORD offset, size;
547 char *p;
548
549 if (!(pInfo = LOCAL_GetHeap( ds ))) return;
550 offset = pPrevArena->size;
551 size = pArena->next - arena - ARENA_HEADER_SIZE;
552 LOCAL_RemoveFreeBlock( ptr, prevArena );
553 LOCAL_RemoveBlock( ptr, arena );
554 pInfo->items--;
555 p = (char *)pPrevArena + ARENA_HEADER_SIZE;
556 while (offset < size)
557 {
558 memcpy( p, p + offset, offset );
559 p += offset;
560 size -= offset;
561 }
562 if (size) memcpy( p, p + offset, size );
563 LOCAL_ShrinkArena( ds, prevArena, newsize );
564 }
565
566
567 /***********************************************************************
568 * LOCAL_GetFreeSpace
569 */
570 static WORD LOCAL_GetFreeSpace(WORD ds, WORD countdiscard)
571 {
572 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
573 LOCALHEAPINFO *pInfo;
574 LOCALARENA *pArena;
575 WORD arena;
576 WORD freespace = 0;
577
578 if (!(pInfo = LOCAL_GetHeap( ds )))
579 {
580 fprintf( stderr, "LOCAL_GetFreeSpace: Local heap not found\n" );
581 LOCAL_PrintHeap(ds);
582 return 0;
583 }
584 arena = pInfo->first;
585 pArena = ARENA_PTR( ptr, arena );
586 while (arena != pArena->free_next)
587 {
588 arena = pArena->free_next;
589 pArena = ARENA_PTR( ptr, arena );
590 if (pArena->size >= freespace) freespace = pArena->size;
591 }
592 /* FIXME doesn't yet calculate space that would become free if everything
593 were discarded when countdiscard == 1 */
594 if (freespace < ARENA_HEADER_SIZE) freespace = 0;
595 else freespace -= ARENA_HEADER_SIZE;
596 return freespace;
597 }
598
599
600 /***********************************************************************
601 * LOCAL_Compact
602 */
603 static WORD LOCAL_Compact( HANDLE16 ds, WORD minfree, WORD flags )
604 {
605 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
606 LOCALHEAPINFO *pInfo;
607 LOCALARENA *pArena, *pMoveArena, *pFinalArena;
608 WORD arena, movearena, finalarena, table;
609 WORD count, movesize, size;
610 WORD freespace;
611 LOCALHANDLEENTRY *pEntry;
612
613 if (!(pInfo = LOCAL_GetHeap( ds )))
614 {
615 fprintf( stderr, "Local_Compact: Local heap not found\n" );
616 LOCAL_PrintHeap(ds);
617 return 0;
618 }
619 dprintf_local(stddeb,
620 "LOCAL_Compact: ds = %04x, minfree = %04x, flags = %04x\n",
621 ds, minfree, flags);
622 freespace = LOCAL_GetFreeSpace(ds, minfree ? 0 : 1);
623 if(freespace >= minfree || (flags & LMEM_NOCOMPACT))
624 {
625 dprintf_local(stddeb, "Returning %04x.\n", freespace);
626 return freespace;
627 }
628 dprintf_local(stddeb, "Local_Compact: Compacting heap %04x.\n", ds);
629 table = pInfo->htable;
630 while(table)
631 {
632 pEntry = (LOCALHANDLEENTRY *)(ptr + table + sizeof(WORD));
633 for(count = *(WORD *)(ptr + table); count > 0; count--, pEntry++)
634 {
635 if((pEntry->lock == 0) && !(pEntry->flags &
636 (LMEM_DISCARDED >> 8)))
637 {
638 /* OK we can move this one if we want */
639 dprintf_local(stddeb,
640 "handle %04x (block %04x) can be moved.\n",
641 (WORD)((char *)pEntry - ptr), pEntry->addr);
642 movearena = ARENA_HEADER(pEntry->addr);
643 pMoveArena = ARENA_PTR(ptr, movearena);
644 movesize = pMoveArena->next - movearena;
645 arena = pInfo->first;
646 pArena = ARENA_PTR(ptr, arena);
647 size = 0xffff;
648 finalarena = 0;
649 /* Try to find the smallest arena that will do, */
650 /* which is below us in memory */
651 for(;;)
652 {
653 arena = pArena->free_next;
654 pArena = ARENA_PTR(ptr, arena);
655 if(arena >= movearena)
656 break;
657 if(arena == pArena->free_next)
658 break;
659 if((pArena->size >= movesize) && (pArena->size < size))
660 {
661 size = pArena->size;
662 finalarena = arena;
663 }
664 }
665 if (finalarena) /* Actually got somewhere to move */
666 {
667 dprintf_local(stddeb, "Moving it to %04x.\n", finalarena);
668 pFinalArena = ARENA_PTR(ptr, finalarena);
669 size = pFinalArena->size;
670 LOCAL_RemoveFreeBlock(ptr, finalarena);
671 LOCAL_ShrinkArena( ds, finalarena, movesize );
672 /* Copy the arena to it's new location */
673 memcpy((char *)pFinalArena + ARENA_HEADER_SIZE,
674 (char *)pMoveArena + ARENA_HEADER_SIZE,
675 movesize - ARENA_HEADER_SIZE );
676 /* Free the old location */
677 LOCAL_FreeArena(ds, movearena);
678 /* Update handle table entry */
679 pEntry->addr = finalarena + ARENA_HEADER_SIZE;
680 }
681 else if((ARENA_PTR(ptr, pMoveArena->prev & ~3)->prev & 3)
682 == LOCAL_ARENA_FREE)
683 {
684 /* Previous arena is free (but < movesize) */
685 /* so we can 'slide' movearena down into it */
686 finalarena = pMoveArena->prev & ~3;
687 LOCAL_GrowArenaDownward( ds, movearena, movesize );
688 /* Update handle table entry */
689 pEntry->addr = finalarena + ARENA_HEADER_SIZE;
690 }
691 }
692 }
693 table = *(WORD *)pEntry;
694 }
695 freespace = LOCAL_GetFreeSpace(ds, minfree ? 0 : 1);
696 if(freespace >= minfree || (flags & LMEM_NODISCARD))
697 {
698 dprintf_local(stddeb, "Returning %04x.\n", freespace);
699 return freespace;
700 }
701
702 table = pInfo->htable;
703 while(table)
704 {
705 pEntry = (LOCALHANDLEENTRY *)(ptr + table + sizeof(WORD));
706 for(count = *(WORD *)(ptr + table); count > 0; count--, pEntry++)
707 {
708 if((pEntry->flags & (LMEM_DISCARDABLE >> 8)) &&
709 pEntry->lock == 0 && !(pEntry->flags & (LMEM_DISCARDED >> 8)))
710 {
711 dprintf_local(stddeb, "Discarding handle %04x (block %04x).\n",
712 (char *)pEntry - ptr, pEntry->addr);
713 LOCAL_FreeArena(ds, pEntry->addr);
714 pEntry->addr = 0;
715 pEntry->flags |= (LMEM_DISCARDED >> 8);
716 /* Call localnotify proc */
717 }
718 }
719 table = *(WORD *)pEntry;
720 }
721 return LOCAL_Compact(ds, 0xffff, LMEM_NODISCARD);
722 }
723
724
725 /***********************************************************************
726 * LOCAL_FindFreeBlock
727 */
728 static HLOCAL16 LOCAL_FindFreeBlock( HANDLE16 ds, WORD size )
729 {
730 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
731 LOCALHEAPINFO *pInfo;
732 LOCALARENA *pArena;
733 WORD arena;
734
735 if (!(pInfo = LOCAL_GetHeap( ds )))
736 {
737 fprintf( stderr, "Local_FindFreeBlock: Local heap not found\n" );
738 LOCAL_PrintHeap(ds);
739 return 0;
740 }
741
742 arena = pInfo->first;
743 pArena = ARENA_PTR( ptr, arena );
744 for (;;) {
745 arena = pArena->free_next;
746 pArena = ARENA_PTR( ptr, arena );
747 if (arena == pArena->free_next) break;
748 if (pArena->size >= size) return arena;
749 }
750 dprintf_local( stddeb, "Local_FindFreeBlock: not enough space\n" );
751 if (debugging_local) LOCAL_PrintHeap(ds);
752 return 0;
753 }
754
755 /***********************************************************************
756 * LOCAL_GetBlock
757 * The segment may get moved around in this function, so all callers
758 * should reset their pointer variables.
759 */
760 static HLOCAL16 LOCAL_GetBlock( HANDLE16 ds, WORD size, WORD flags )
761 {
762 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
763 LOCALHEAPINFO *pInfo;
764 LOCALARENA *pArena;
765 WORD arena;
766
767 if (!(pInfo = LOCAL_GetHeap( ds )))
768 {
769 fprintf( stderr, "Local_GetBlock: Local heap not found\n");
770 LOCAL_PrintHeap(ds);
771 return 0;
772 }
773
774 size += ARENA_HEADER_SIZE;
775 size = LALIGN( MAX( size, sizeof(LOCALARENA) ) );
776
777 /* Find a suitable free block */
778 arena = LOCAL_FindFreeBlock( ds, size );
779 if (arena == 0) {
780 /* no space: try to make some */
781 LOCAL_Compact( ds, size, flags );
782 arena = LOCAL_FindFreeBlock( ds, size );
783 }
784 if (arena == 0) {
785 /* still no space: try to grow the segment */
786 LOCAL_GrowHeap( ds );
787 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
788 pInfo = LOCAL_GetHeap( ds );
789 arena = LOCAL_FindFreeBlock( ds, size );
790 }
791 if (arena == 0) {
792 fprintf( stderr, "Local_GetBlock: not enough space in heap %04x for %d bytes\n",
793 ds, size );
794 return 0;
795 }
796
797 /* Make a block out of the free arena */
798 pArena = ARENA_PTR( ptr, arena );
799 dprintf_local(stddeb, "LOCAL_GetBlock size = %04x, arena %04x size %04x\n",
800 size, arena, pArena->size );
801 LOCAL_RemoveFreeBlock( ptr, arena );
802 LOCAL_ShrinkArena( ds, arena, size );
803
804 if (flags & LMEM_ZEROINIT) memset( (char *)pArena + ARENA_HEADER_SIZE, 0,
805 size - ARENA_HEADER_SIZE );
806 return arena + ARENA_HEADER_SIZE;
807 }
808
809
810 /***********************************************************************
811 * LOCAL_NewHTable
812 */
813 static BOOL16 LOCAL_NewHTable( HANDLE16 ds )
814 {
815 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
816 LOCALHEAPINFO *pInfo;
817 LOCALHANDLEENTRY *pEntry;
818 HLOCAL16 handle;
819 int i;
820
821 dprintf_local( stddeb, "Local_NewHTable\n" );
822 if (!(pInfo = LOCAL_GetHeap( ds )))
823 {
824 fprintf( stderr, "Local heap not found\n");
825 LOCAL_PrintHeap(ds);
826 return FALSE;
827 }
828
829 if (!(handle = LOCAL_GetBlock( ds, pInfo->hdelta * sizeof(LOCALHANDLEENTRY)
830 + 2 * sizeof(WORD), LMEM_FIXED )))
831 return FALSE;
832 if (!(ptr = PTR_SEG_OFF_TO_LIN( ds, 0 )))
833 fprintf(stderr, "LOCAL_NewHTable: ptr == NULL after GetBlock.\n");
834 if (!(pInfo = LOCAL_GetHeap( ds )))
835 fprintf(stderr,"LOCAL_NewHTable: pInfo == NULL after GetBlock.\n");
836
837 /* Fill the entry table */
838
839 *(WORD *)(ptr + handle) = pInfo->hdelta;
840 pEntry = (LOCALHANDLEENTRY *)(ptr + handle + sizeof(WORD));
841 for (i = pInfo->hdelta; i > 0; i--) (pEntry++)->lock = 0xff;
842 *(WORD *)pEntry = pInfo->htable;
843 pInfo->htable = handle;
844 return TRUE;
845 }
846
847
848 /***********************************************************************
849 * LOCAL_GetNewHandleEntry
850 */
851 static HLOCAL16 LOCAL_GetNewHandleEntry( HANDLE16 ds )
852 {
853 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
854 LOCALHEAPINFO *pInfo;
855 LOCALHANDLEENTRY *pEntry = NULL;
856 WORD table;
857
858 if (!(pInfo = LOCAL_GetHeap( ds )))
859 {
860 fprintf( stderr, "LOCAL_GetNewHandleEntry: Local heap not found\n");
861 LOCAL_PrintHeap(ds);
862 return 0;
863 }
864
865 /* Find a free slot in existing tables */
866
867 table = pInfo->htable;
868 while (table)
869 {
870 WORD count = *(WORD *)(ptr + table);
871 pEntry = (LOCALHANDLEENTRY *)(ptr + table + sizeof(WORD));
872 for (; count > 0; count--, pEntry++)
873 if (pEntry->lock == 0xff) break;
874 if (count) break;
875 table = *(WORD *)pEntry;
876 }
877
878 if (!table) /* We need to create a new table */
879 {
880 if (!LOCAL_NewHTable( ds )) return 0;
881 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
882 pInfo = LOCAL_GetHeap( ds );
883 pEntry = (LOCALHANDLEENTRY *)(ptr + pInfo->htable + sizeof(WORD));
884 }
885
886 /* Now allocate this entry */
887
888 pEntry->lock = 0;
889 dprintf_local( stddeb, "LOCAL_GetNewHandleEntry(%04x): %04x\n",
890 ds, ((char *)pEntry - ptr) );
891 return (HLOCAL16)((char *)pEntry - ptr);
892 }
893
894
895 /***********************************************************************
896 * LOCAL_FreeHandleEntry
897 *
898 * Free a handle table entry.
899 */
900 static void LOCAL_FreeHandleEntry( HANDLE16 ds, HLOCAL16 handle )
901 {
902 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
903 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
904 LOCALHEAPINFO *pInfo;
905 WORD *pTable;
906 WORD table, count, i;
907
908 if (!(pInfo = LOCAL_GetHeap( ds ))) return;
909
910 /* Find the table where this handle comes from */
911
912 pTable = &pInfo->htable;
913 while (*pTable)
914 {
915 WORD size = (*(WORD *)(ptr + *pTable)) * sizeof(LOCALHANDLEENTRY);
916 if ((handle >= *pTable + sizeof(WORD)) &&
917 (handle < *pTable + sizeof(WORD) + size)) break; /* Found it */
918 pTable = (WORD *)(ptr + *pTable + sizeof(WORD) + size);
919 }
920 if (!*pTable)
921 {
922 fprintf(stderr, "LOCAL_FreeHandleEntry: invalid entry %04x\n", handle);
923 LOCAL_PrintHeap( ds );
924 return;
925 }
926
927 /* Make the entry free */
928
929 pEntry->addr = 0; /* just in case */
930 pEntry->lock = 0xff;
931
932 /* Now check if all entries in this table are free */
933
934 table = *pTable;
935 pEntry = (LOCALHANDLEENTRY *)(ptr + table + sizeof(WORD));
936 count = *(WORD *)(ptr + table);
937 for (i = count; i > 0; i--, pEntry++) if (pEntry->lock != 0xff) return;
938
939 /* Remove the table from the linked list and free it */
940
941 dprintf_local( stddeb, "LOCAL_FreeHandleEntry(%04x): freeing table %04x\n",
942 ds, table);
943 *pTable = *(WORD *)pEntry;
944 LOCAL_FreeArena( ds, ARENA_HEADER( table ) );
945 }
946
947
948 /***********************************************************************
949 * LOCAL_Free
950 *
951 * Implementation of LocalFree().
952 */
953 HLOCAL16 LOCAL_Free( HANDLE16 ds, HLOCAL16 handle )
954 {
955 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
956
957 dprintf_local( stddeb, "LocalFree: %04x ds=%04x\n", handle, ds );
958
959 if (!handle) { fprintf( stderr, "LOCAL_Free: handle is 0.\n" ); return 0; }
960 if (HANDLE_FIXED( handle ))
961 {
962 if (!LOCAL_FreeArena( ds, ARENA_HEADER( handle ) )) return 0; /* OK */
963 else return handle; /* couldn't free it */
964 }
965 else
966 {
967 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
968 if (!(pEntry->flags & (LMEM_DISCARDED >> 8)))
969 {
970 dprintf_local( stddeb, "LocalFree: real block at %04x\n",
971 pEntry->addr );
972 if (LOCAL_FreeArena( ds, ARENA_HEADER(pEntry->addr) ))
973 return handle; /* couldn't free it */
974 }
975 LOCAL_FreeHandleEntry( ds, handle );
976 return 0; /* OK */
977 }
978 }
979
980
981 /***********************************************************************
982 * LOCAL_Alloc
983 *
984 * Implementation of LocalAlloc().
985 */
986 HLOCAL16 LOCAL_Alloc( HANDLE16 ds, WORD flags, WORD size )
987 {
988 char *ptr;
989 HLOCAL16 handle;
990
991 dprintf_local( stddeb, "LocalAlloc: %04x %d ds=%04x\n", flags, size, ds );
992
993 if (flags & LMEM_MOVEABLE)
994 {
995 LOCALHANDLEENTRY *plhe;
996 HLOCAL16 hmem;
997
998 if (!(hmem = LOCAL_GetBlock( ds, size, flags ))) return 0;
999 if (!(handle = LOCAL_GetNewHandleEntry( ds )))
1000 {
1001 fprintf( stderr, "LocalAlloc: couldn't get handle\n");
1002 LOCAL_FreeArena( ds, ARENA_HEADER(hmem) );
1003 return 0;
1004 }
1005 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1006 plhe = (LOCALHANDLEENTRY *)(ptr + handle);
1007 plhe->addr = hmem;
1008 plhe->flags = (BYTE)(flags >> 8);
1009 plhe->lock = 0;
1010 }
1011 else handle = LOCAL_GetBlock( ds, size, flags );
1012
1013 return handle;
1014 }
1015
1016
1017 /***********************************************************************
1018 * LOCAL_ReAlloc
1019 *
1020 * Implementation of LocalReAlloc().
1021 */
1022 HLOCAL16 LOCAL_ReAlloc( HANDLE16 ds, HLOCAL16 handle, WORD size, WORD flags )
1023 {
1024 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1025 LOCALHEAPINFO *pInfo;
1026 LOCALARENA *pArena, *pNext;
1027 LOCALHANDLEENTRY *pEntry;
1028 WORD arena, newhandle, blockhandle, oldsize;
1029 LONG nextarena;
1030
1031 if (!handle) return LOCAL_Alloc( ds, size, flags );
1032
1033 dprintf_local( stddeb, "LocalReAlloc: %04x %d %04x ds=%04x\n",
1034 handle, size, flags, ds );
1035 if (!(pInfo = LOCAL_GetHeap( ds ))) return 0;
1036
1037 if (HANDLE_FIXED( handle )) blockhandle = handle;
1038 else
1039 {
1040 pEntry = (LOCALHANDLEENTRY *) (ptr + handle);
1041 if(pEntry->flags & (LMEM_DISCARDED >> 8))
1042 {
1043 dprintf_local(stddeb, "ReAllocating discarded block\n");
1044 if (!(pEntry->addr = LOCAL_GetBlock( ds, size, flags))) return 0;
1045 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 ); /* Reload ptr */
1046 pEntry = (LOCALHANDLEENTRY *) (ptr + handle);
1047 pEntry->flags = (BYTE) (flags >> 8);
1048 pEntry->lock = 0;
1049 return handle;
1050 }
1051 if (!(blockhandle = pEntry->addr))
1052 {
1053 fprintf( stderr, "Local_ReAlloc(%04x,%04x): invalid handle\n",
1054 ds, handle );
1055 return 0;
1056 }
1057 }
1058
1059 arena = ARENA_HEADER( blockhandle );
1060 dprintf_local( stddeb, "LocalReAlloc: arena is %04x\n", arena );
1061 pArena = ARENA_PTR( ptr, arena );
1062
1063 if ((flags & LMEM_MODIFY) && (flags & LMEM_DISCARDABLE))
1064 {
1065 if (HANDLE_FIXED(handle))
1066 {
1067 fprintf(stderr,"LocalReAlloc: LMEM_MODIFY & LMEM_DISCARDABLE on a fixed handle.\n");
1068 return handle;
1069 }
1070 dprintf_local( stddeb, "Making block discardable.\n" );
1071 pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
1072 pEntry->flags |= (LMEM_DISCARDABLE >> 8);
1073 return handle;
1074 }
1075
1076 if ((flags & LMEM_MODIFY) || (flags & LMEM_DISCARDABLE))
1077 {
1078 fprintf(stderr,"LocalReAlloc: flags %04x. MODIFY & DISCARDABLE should both be set\n", flags);
1079 return handle;
1080 }
1081
1082 if (!size)
1083 {
1084 if (HANDLE_FIXED(handle))
1085 {
1086 if (flags & LMEM_MOVEABLE)
1087 {
1088 dprintf_local(stddeb, "Freeing fixed block.\n");
1089 return LOCAL_Free( ds, handle );
1090 }
1091 else size = 1;
1092 }
1093 else
1094 {
1095 pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
1096 if (pEntry->lock == 0)
1097 {
1098 /* discards moveable blocks is this right? */
1099 dprintf_local(stddeb,"Discarding block\n");
1100 LOCAL_FreeArena(ds, ARENA_HEADER(pEntry->addr));
1101 pEntry->addr = 0;
1102 pEntry->flags = (LMEM_DISCARDED >> 8);
1103 return 0;
1104 }
1105 return handle;
1106 }
1107 }
1108
1109 size = LALIGN( size );
1110 oldsize = pArena->next - arena - ARENA_HEADER_SIZE;
1111 nextarena = LALIGN(blockhandle + size);
1112
1113 /* Check for size reduction */
1114
1115 if (nextarena <= pArena->next)
1116 {
1117 if (nextarena < pArena->next - LALIGN(sizeof(LOCALARENA)))
1118 {
1119 dprintf_local( stddeb, "size reduction, making new free block\n");
1120 /* It is worth making a new free block */
1121 LOCAL_AddBlock( ptr, arena, nextarena );
1122 pInfo->items++;
1123 LOCAL_FreeArena( ds, nextarena );
1124 }
1125 dprintf_local( stddeb, "LocalReAlloc: returning %04x\n", handle );
1126 return handle;
1127 }
1128
1129 /* Check if the next block is free */
1130
1131 pNext = ARENA_PTR( ptr, pArena->next );
1132 if (((pNext->prev & 3) == LOCAL_ARENA_FREE) &&
1133 (nextarena <= pNext->next))
1134 {
1135 LOCAL_RemoveBlock( ptr, pArena->next );
1136 if (nextarena < pArena->next - LALIGN(sizeof(LOCALARENA)))
1137 {
1138 dprintf_local( stddeb, "size increase, making new free block\n");
1139 /* It is worth making a new free block */
1140 LOCAL_AddBlock( ptr, arena, nextarena );
1141 pInfo->items++;
1142 LOCAL_FreeArena( ds, nextarena );
1143 }
1144 dprintf_local( stddeb, "LocalReAlloc: returning %04x\n", handle );
1145 return handle;
1146 }
1147
1148 /* Now we have to allocate a new block, but not if fixed block and no
1149 LMEM_MOVEABLE */
1150
1151 if (HANDLE_FIXED(handle) && !(flags & LMEM_MOVEABLE))
1152 {
1153 dprintf_local(stddeb, "Needed to move fixed block, but LMEM_MOVEABLE not specified.\n");
1154 return 0; /* FIXME: should we free it here? */
1155 }
1156
1157 newhandle = LOCAL_GetBlock( ds, size, flags );
1158 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 ); /* Reload ptr */
1159 if (!newhandle)
1160 {
1161 /* Remove the block from the heap and try again */
1162 LPSTR buffer = HeapAlloc( SystemHeap, 0, oldsize );
1163 if (!buffer) return 0;
1164 memcpy( buffer, ptr + (arena + ARENA_HEADER_SIZE), oldsize );
1165 LOCAL_FreeArena( ds, arena );
1166 if (!(newhandle = LOCAL_GetBlock( ds, size, flags )))
1167 {
1168 if (!(newhandle = LOCAL_GetBlock( ds, oldsize, flags )))
1169 {
1170 fprintf( stderr, "LocalRealloc: can't restore saved block\n" );
1171 HeapFree( SystemHeap, 0, buffer );
1172 return 0;
1173 }
1174 size = oldsize;
1175 }
1176 ptr = PTR_SEG_OFF_TO_LIN( ds, 0 ); /* Reload ptr */
1177 memcpy( ptr + newhandle, buffer, oldsize );
1178 HeapFree( SystemHeap, 0, buffer );
1179 }
1180 else
1181 {
1182 memcpy( ptr + newhandle, ptr + (arena + ARENA_HEADER_SIZE), oldsize );
1183 LOCAL_FreeArena( ds, arena );
1184 }
1185 if (HANDLE_MOVEABLE( handle ))
1186 {
1187 dprintf_local( stddeb, "LocalReAlloc: fixing handle\n");
1188 pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
1189 pEntry->addr = newhandle;
1190 pEntry->lock = 0;
1191 newhandle = handle;
1192 }
1193 if (size == oldsize) newhandle = 0; /* Realloc failed */
1194 dprintf_local( stddeb, "LocalReAlloc: returning %04x\n", newhandle );
1195 return newhandle;
1196 }
1197
1198
1199 /***********************************************************************
1200 * LOCAL_InternalLock
1201 */
1202 static HLOCAL16 LOCAL_InternalLock( LPSTR heap, HLOCAL16 handle )
1203 {
1204 dprintf_local( stddeb, "LocalLock: %04x ", handle );
1205 if (HANDLE_MOVEABLE(handle))
1206 {
1207 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY *)(heap + handle);
1208 if (pEntry->lock < 0xfe) pEntry->lock++;
1209 handle = pEntry->addr;
1210 }
1211 dprintf_local( stddeb, "returning %04x\n", handle );
1212 return handle;
1213 }
1214
1215
1216 /***********************************************************************
1217 * LOCAL_Lock
1218 */
1219 LPSTR LOCAL_Lock( HANDLE16 ds, HLOCAL16 handle )
1220 {
1221 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1222 return handle ? ptr + LOCAL_InternalLock( ptr, handle ) : NULL;
1223 }
1224
1225
1226 /***********************************************************************
1227 * LOCAL_Unlock
1228 */
1229 BOOL16 LOCAL_Unlock( HANDLE16 ds, HLOCAL16 handle )
1230 {
1231 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1232
1233 dprintf_local( stddeb, "LocalUnlock: %04x\n", handle );
1234 if (HANDLE_MOVEABLE(handle))
1235 {
1236 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
1237 if (!pEntry->lock || (pEntry->lock == 0xff)) return FALSE;
1238 /* For moveable block, return the new lock count */
1239 /* (see _Windows_Internals_ p. 197) */
1240 return --pEntry->lock;
1241 }
1242 else return FALSE;
1243 }
1244
1245
1246 /***********************************************************************
1247 * LOCAL_Size
1248 *
1249 * Implementation of LocalSize().
1250 */
1251 WORD LOCAL_Size( HANDLE16 ds, HLOCAL16 handle )
1252 {
1253 char *ptr = PTR_SEG_OFF_TO_LIN( CURRENT_DS, 0 );
1254 LOCALARENA *pArena;
1255
1256 dprintf_local( stddeb, "LocalSize: %04x ds=%04x\n", handle, ds );
1257
1258 if (HANDLE_MOVEABLE( handle )) handle = *(WORD *)(ptr + handle);
1259 pArena = ARENA_PTR( ptr, ARENA_HEADER(handle) );
1260 return pArena->next - handle;
1261 }
1262
1263
1264 /***********************************************************************
1265 * LOCAL_Flags
1266 *
1267 * Implementation of LocalFlags().
1268 */
1269 WORD LOCAL_Flags( HANDLE16 ds, HLOCAL16 handle )
1270 {
1271 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1272
1273 if (HANDLE_MOVEABLE(handle))
1274 {
1275 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY *)(ptr + handle);
1276 dprintf_local( stddeb, "LOCAL_Flags(%04x,%04x): returning %04x\n",
1277 ds, handle, pEntry->lock | (pEntry->flags << 8) );
1278 return pEntry->lock | (pEntry->flags << 8);
1279 }
1280 else
1281 {
1282 dprintf_local( stddeb, "LOCAL_Flags(%04x,%04x): returning 0\n",
1283 ds, handle );
1284 return 0;
1285 }
1286 }
1287
1288
1289 /***********************************************************************
1290 * LOCAL_HeapSize
1291 *
1292 * Implementation of LocalHeapSize().
1293 */
1294 WORD LOCAL_HeapSize( HANDLE16 ds )
1295 {
1296 LOCALHEAPINFO *pInfo = LOCAL_GetHeap( ds );
1297 if (!pInfo) return 0;
1298 return pInfo->last - pInfo->first;
1299 }
1300
1301
1302 /***********************************************************************
1303 * LOCAL_CountFree
1304 *
1305 * Implementation of LocalCountFree().
1306 */
1307 WORD LOCAL_CountFree( HANDLE16 ds )
1308 {
1309 WORD arena, total;
1310 LOCALARENA *pArena;
1311 LOCALHEAPINFO *pInfo;
1312 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1313
1314 if (!(pInfo = LOCAL_GetHeap( ds )))
1315 {
1316 fprintf( stderr, "LOCAL_Handle(%04x): Local heap not found\n", ds );
1317 LOCAL_PrintHeap( ds );
1318 return 0;
1319 }
1320
1321 total = 0;
1322 arena = pInfo->first;
1323 pArena = ARENA_PTR( ptr, arena );
1324 for (;;)
1325 {
1326 arena = pArena->free_next;
1327 pArena = ARENA_PTR( ptr, arena );
1328 if (arena == pArena->free_next) break;
1329 total += pArena->size;
1330 }
1331 dprintf_local( stddeb, "LOCAL_CountFree(%04x): returning %d\n", ds, total);
1332 return total;
1333 }
1334
1335
1336 /***********************************************************************
1337 * LOCAL_Handle
1338 *
1339 * Implementation of LocalHandle().
1340 */
1341 HLOCAL16 LOCAL_Handle( HANDLE16 ds, WORD addr )
1342 {
1343 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1344 LOCALHEAPINFO *pInfo;
1345 WORD table;
1346
1347 if (!(pInfo = LOCAL_GetHeap( ds )))
1348 {
1349 fprintf( stderr, "LOCAL_Handle(%04x): Local heap not found\n", ds );
1350 LOCAL_PrintHeap( ds );
1351 return 0;
1352 }
1353
1354 /* Find the address in the entry tables */
1355
1356 table = pInfo->htable;
1357 while (table)
1358 {
1359 WORD count = *(WORD *)(ptr + table);
1360 LOCALHANDLEENTRY *pEntry = (LOCALHANDLEENTRY*)(ptr+table+sizeof(WORD));
1361 for (; count > 0; count--, pEntry++)
1362 if (pEntry->addr == addr) return (HLOCAL16)((char *)pEntry - ptr);
1363 table = *(WORD *)pEntry;
1364 }
1365
1366 return (HLOCAL16)addr; /* Fixed block handle is addr */
1367 }
1368
1369
1370 /***********************************************************************
1371 * LocalAlloc16 (KERNEL.5)
1372 */
1373 HLOCAL16 LocalAlloc16( UINT16 flags, WORD size )
1374 {
1375 return LOCAL_Alloc( CURRENT_DS, flags, size );
1376 }
1377
1378
1379 /***********************************************************************
1380 * LocalReAlloc16 (KERNEL.6)
1381 */
1382 HLOCAL16 LocalReAlloc16( HLOCAL16 handle, WORD size, UINT16 flags )
1383 {
1384 return LOCAL_ReAlloc( CURRENT_DS, handle, size, flags );
1385 }
1386
1387
1388 /***********************************************************************
1389 * LocalFree16 (KERNEL.7)
1390 */
1391 HLOCAL16 LocalFree16( HLOCAL16 handle )
1392 {
1393 return LOCAL_Free( CURRENT_DS, handle );
1394 }
1395
1396
1397 /***********************************************************************
1398 * LocalLock16 (KERNEL.8)
1399 *
1400 * Note: only the offset part of the pointer is returned by the relay code.
1401 */
1402 SEGPTR LocalLock16( HLOCAL16 handle )
1403 {
1404 HANDLE16 ds = CURRENT_DS;
1405 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1406 return PTR_SEG_OFF_TO_SEGPTR( ds, LOCAL_InternalLock( ptr, handle ) );
1407 }
1408
1409
1410 /***********************************************************************
1411 * LocalUnlock16 (KERNEL.9)
1412 */
1413 BOOL16 LocalUnlock16( HLOCAL16 handle )
1414 {
1415 return LOCAL_Unlock( CURRENT_DS, handle );
1416 }
1417
1418
1419 /***********************************************************************
1420 * LocalSize16 (KERNEL.10)
1421 */
1422 UINT16 LocalSize16( HLOCAL16 handle )
1423 {
1424 return LOCAL_Size( CURRENT_DS, handle );
1425 }
1426
1427
1428 /***********************************************************************
1429 * LocalHandle16 (KERNEL.11)
1430 */
1431 HLOCAL16 LocalHandle16( WORD addr )
1432 {
1433 return LOCAL_Handle( CURRENT_DS, addr );
1434 }
1435
1436
1437 /***********************************************************************
1438 * LocalFlags16 (KERNEL.12)
1439 */
1440 UINT16 LocalFlags16( HLOCAL16 handle )
1441 {
1442 return LOCAL_Flags( CURRENT_DS, handle );
1443 }
1444
1445
1446 /***********************************************************************
1447 * LocalCompact16 (KERNEL.13)
1448 */
1449 UINT16 LocalCompact16( UINT16 minfree )
1450 {
1451 dprintf_local( stddeb, "LocalCompact: %04x\n", minfree );
1452 return LOCAL_Compact( CURRENT_DS, minfree, 0 );
1453 }
1454
1455
1456 /***********************************************************************
1457 * LocalNotify (KERNEL.14)
1458 */
1459 FARPROC16 LocalNotify( FARPROC16 func )
1460 {
1461 LOCALHEAPINFO *pInfo;
1462 FARPROC16 oldNotify;
1463 HANDLE16 ds = CURRENT_DS;
1464
1465 if (!(pInfo = LOCAL_GetHeap( ds )))
1466 {
1467 fprintf( stderr, "LOCAL_Notify(%04x): Local heap not found\n", ds );
1468 LOCAL_PrintHeap( ds );
1469 return 0;
1470 }
1471 dprintf_local( stddeb, "LocalNotify(%04x): %08lx\n", ds, (DWORD)func );
1472 oldNotify = pInfo->notify;
1473 pInfo->notify = func;
1474 return oldNotify;
1475 }
1476
1477
1478 /***********************************************************************
1479 * LocalShrink16 (KERNEL.121)
1480 */
1481 UINT16 LocalShrink16( HGLOBAL16 handle, UINT16 newsize )
1482 {
1483 dprintf_local( stddeb, "LocalShrink: %04x %04x\n", handle, newsize );
1484 return 0;
1485 }
1486
1487
1488 /***********************************************************************
1489 * GetHeapSpaces (KERNEL.138)
1490 */
1491 DWORD GetHeapSpaces( HMODULE16 module )
1492 {
1493 NE_MODULE *pModule;
1494 WORD ds;
1495
1496 module = GetExePtr( module );
1497 if (!(pModule = MODULE_GetPtr( module ))) return 0;
1498 ds = (NE_SEG_TABLE( pModule ) + pModule->dgroup - 1)->selector;
1499 return MAKELONG( LOCAL_CountFree( ds ), LOCAL_HeapSize( ds ) );
1500 }
1501
1502
1503 /***********************************************************************
1504 * LocalCountFree (KERNEL.161)
1505 */
1506 WORD LocalCountFree(void)
1507 {
1508 return LOCAL_CountFree( CURRENT_DS );
1509 }
1510
1511
1512 /***********************************************************************
1513 * LocalHeapSize (KERNEL.162)
1514 */
1515 WORD LocalHeapSize()
1516 {
1517 dprintf_local( stddeb, "LocalHeapSize:\n" );
1518 return LOCAL_HeapSize( CURRENT_DS );
1519 }
1520
1521
1522 /***********************************************************************
1523 * LocalHandleDelta (KERNEL.310)
1524 */
1525 WORD LocalHandleDelta( WORD delta )
1526 {
1527 LOCALHEAPINFO *pInfo;
1528
1529 if (!(pInfo = LOCAL_GetHeap( CURRENT_DS )))
1530 {
1531 fprintf( stderr, "LocalHandleDelta: Local heap not found\n");
1532 LOCAL_PrintHeap( CURRENT_DS );
1533 return 0;
1534 }
1535 if (delta) pInfo->hdelta = delta;
1536 dprintf_local(stddeb, "LocalHandleDelta: returning %04x\n", pInfo->hdelta);
1537 return pInfo->hdelta;
1538 }
1539
1540
1541 /***********************************************************************
1542 * LocalInfo (TOOLHELP.56)
1543 */
1544 BOOL16 LocalInfo( LOCALINFO *pLocalInfo, HGLOBAL16 handle )
1545 {
1546 LOCALHEAPINFO *pInfo = LOCAL_GetHeap(SELECTOROF(WIN16_GlobalLock16(handle)));
1547 if (!pInfo) return FALSE;
1548 pLocalInfo->wcItems = pInfo->items;
1549 return TRUE;
1550 }
1551
1552
1553 /***********************************************************************
1554 * LocalFirst (TOOLHELP.57)
1555 */
1556 BOOL16 LocalFirst( LOCALENTRY *pLocalEntry, HGLOBAL16 handle )
1557 {
1558 WORD ds = GlobalHandleToSel( handle );
1559 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1560 LOCALHEAPINFO *pInfo = LOCAL_GetHeap( ds );
1561 if (!pInfo) return FALSE;
1562
1563 pLocalEntry->hHandle = pInfo->first + ARENA_HEADER_SIZE;
1564 pLocalEntry->wAddress = pLocalEntry->hHandle;
1565 pLocalEntry->wFlags = LF_FIXED;
1566 pLocalEntry->wcLock = 0;
1567 pLocalEntry->wType = LT_NORMAL;
1568 pLocalEntry->hHeap = handle;
1569 pLocalEntry->wHeapType = NORMAL_HEAP;
1570 pLocalEntry->wNext = ARENA_PTR(ptr,pInfo->first)->next;
1571 pLocalEntry->wSize = pLocalEntry->wNext - pLocalEntry->hHandle;
1572 return TRUE;
1573 }
1574
1575
1576 /***********************************************************************
1577 * LocalNext (TOOLHELP.58)
1578 */
1579 BOOL16 LocalNext( LOCALENTRY *pLocalEntry )
1580 {
1581 WORD ds = GlobalHandleToSel( pLocalEntry->hHeap );
1582 char *ptr = PTR_SEG_OFF_TO_LIN( ds, 0 );
1583 LOCALARENA *pArena;
1584
1585 if (!LOCAL_GetHeap( ds )) return FALSE;
1586 if (!pLocalEntry->wNext) return FALSE;
1587 pArena = ARENA_PTR( ptr, pLocalEntry->wNext );
1588
1589 pLocalEntry->hHandle = pLocalEntry->wNext + ARENA_HEADER_SIZE;
1590 pLocalEntry->wAddress = pLocalEntry->hHandle;
1591 pLocalEntry->wFlags = (pArena->prev & 3) + 1;
1592 pLocalEntry->wcLock = 0;
1593 pLocalEntry->wType = LT_NORMAL;
1594 if (pArena->next != pLocalEntry->wNext) /* last one? */
1595 pLocalEntry->wNext = pArena->next;
1596 else
1597 pLocalEntry->wNext = 0;
1598 pLocalEntry->wSize = pLocalEntry->wNext - pLocalEntry->hHandle;
1599 return TRUE;
1600 }
1601
1602
1603 /***********************************************************************
1604 * LocalAlloc32 (KERNEL32.371)
1605 */
1606 HLOCAL32 LocalAlloc32( UINT32 flags, DWORD size )
1607 {
1608 return (HLOCAL32)GlobalAlloc32( flags, size );
1609 }
1610
1611
1612 /***********************************************************************
1613 * LocalCompact32 (KERNEL32.372)
1614 */
1615 UINT32 LocalCompact32( UINT32 minfree )
1616 {
1617 return 0; /* LocalCompact does nothing in Win32 */
1618 }
1619
1620
1621 /***********************************************************************
1622 * LocalFlags32 (KERNEL32.374)
1623 */
1624 UINT32 LocalFlags32( HLOCAL32 handle )
1625 {
1626 return GlobalFlags32( (HGLOBAL32)handle );
1627 }
1628
1629
1630 /***********************************************************************
1631 * LocalFree32 (KERNEL32.375)
1632 */
1633 HLOCAL32 LocalFree32( HLOCAL32 handle )
1634 {
1635 return (HLOCAL32)GlobalFree32( (HGLOBAL32)handle );
1636 }
1637
1638
1639 /***********************************************************************
1640 * LocalHandle32 (KERNEL32.376)
1641 */
1642 HLOCAL32 LocalHandle32( LPCVOID ptr )
1643 {
1644 return (HLOCAL32)GlobalHandle32( ptr );
1645 }
1646
1647
1648 /***********************************************************************
1649 * LocalLock32 (KERNEL32.377)
1650 */
1651 LPVOID LocalLock32( HLOCAL32 handle )
1652 {
1653 return GlobalLock32( (HGLOBAL32)handle );
1654 }
1655
1656
1657 /***********************************************************************
1658 * LocalReAlloc32 (KERNEL32.378)
1659 */
1660 HLOCAL32 LocalReAlloc32( HLOCAL32 handle, DWORD size, UINT32 flags )
1661 {
1662 return (HLOCAL32)GlobalReAlloc32( (HGLOBAL32)handle, size, flags );
1663 }
1664
1665
1666 /***********************************************************************
1667 * LocalShrink32 (KERNEL32.379)
1668 */
1669 UINT32 LocalShrink32( HGLOBAL32 handle, UINT32 newsize )
1670 {
1671 return 0; /* LocalShrink does nothing in Win32 */
1672 }
1673
1674
1675 /***********************************************************************
1676 * LocalSize32 (KERNEL32.380)
1677 */
1678 UINT32 LocalSize32( HLOCAL32 handle )
1679 {
1680 return GlobalSize32( (HGLOBAL32)handle );
1681 }
1682
1683
1684 /***********************************************************************
1685 * LocalUnlock32 (KERNEL32.381)
1686 */
1687 BOOL32 LocalUnlock32( HLOCAL32 handle )
1688 {
1689 return GlobalUnlock32( (HGLOBAL32)handle );
1690 }
Maarten's multimedia branch