WebSVN – Kolibri OS – Diff – /drivers/disk/tmpdisk.asm

 ERROR_INVALID_ID        equ 2 ; .DiskId must be from 0 to 9
 ERROR_SIZE_TOO_LARGE    equ 3 ; .DiskSize is too large
 ERROR_SIZE_TOO_SMALL    equ 4 ; .DiskSize is too small
 ERROR_NO_MEMORY         equ 5 ; memory allocation failed
 include '../struct.inc'
 API_VERSION             equ 1
-; Input structures:
-struc add_disk_struc
+; Input structures:
-{
+struct add_disk_struc
-.DiskSize       dd      ? ; disk size in sectors, 1 sector = 512 bytes
+DiskSize        dd      ? ; disk size in sectors, 1 sector = 512 bytes
-; Note: .DiskSize is the full size, including FAT service data.
+; Note: DiskSize is the full size, including FAT service data.
-; Size for useful data is slightly less than this number.
-.DiskId         db      ? ; from 0 to 9
-.sizeof:
-}
-virtual at 0
+; Size for useful data is slightly less than this number.
-add_disk_struc  add_disk_struc
+DiskId          db      ? ; from 0 to 9
-end virtual
-struc del_disk_struc
+ends
-{
-.DiskId         db      ? ; from 0 to 9
-.sizeof:
-}
-virtual at 0
+struct del_disk_struc
-del_disk_struc del_disk_struc
+DiskId          db      ? ; from 0 to 9
-end virtual
+ends
 max_num_disks   equ     10
-; standard driver stuff
-format MS COFF
+; standard driver stuff; version of driver model = 5
+format PE DLL native 0.05
-DEBUG equ 0
-include 'proc32.inc'
-include 'imports.inc'
-public START
-public version
+DEBUG equ 0
-struc IOCTL
-{
-        .handle         dd ?
-        .io_code        dd ?
-        .input          dd ?
-        .inp_size       dd ?
-        .output         dd ?
-        .out_size       dd ?
-}
+section '.flat' code readable writable executable
 data fixups
-virtual at 0
+end data
-IOCTL IOCTL
+entry START
-end virtual
+include '../proc32.inc'
 include '../peimport.inc'
-section '.flat' code readable align 16
+include '../macros.inc'
 ; the start procedure (see the description above)
 proc START
 ; This procedure is called in two situations:
 ; when the driver is loading and when the system is shutting down.
 ; 1. Check that the driver is loading; do nothing unless so.
         xor     eax, eax ; set return value in case we will do nothing
         cmp     dword [esp+4], 1
         jne     .nothing
 ; 2. Register the driver in the system.
-        stdcall RegService, my_service, service_proc
+        invoke  RegService, my_service, service_proc
 ; 3. Return the value returned by RegService back to the system.
         jmp     .return
 .no.srv_getversion:
         dec     ecx     ; check for DEV_ADD_DISK
         jnz     .no.dev_add_disk
 ; 5. This is DEV_ADD_DISK request, input is add_disk_struc, output is 1 byte
-; 5a. Input size must be exactly add_disk_struc.sizeof bytes.
+; 5a. Input size must be exactly sizeof.add_disk_struc bytes.
-        cmp     [edx+IOCTL.inp_size], add_disk_struc.sizeof
+        cmp     [edx+IOCTL.inp_size], sizeof.add_disk_struc
         jnz     .return
 ; 5b. Load input parameters and call the worker procedure.
         mov     eax, [edx+IOCTL.input]
         movzx   ebx, [eax+add_disk_struc.DiskId]
         mov     esi, [eax+add_disk_struc.DiskSize]
         jmp     .return
 .no.dev_add_disk:
         dec     ecx     ; check for DEV_DEL_DISK
         jnz     .return
 ; 6. This is DEV_DEL_DISK request, input is del_disk_struc
-; 6a. Input size must be exactly del_disk_struc.sizeof bytes.
+; 6a. Input size must be exactly sizeof.del_disk_struc bytes.
-        cmp     [edx+IOCTL.inp_size], del_disk_struc.sizeof
+        cmp     [edx+IOCTL.inp_size], sizeof.del_disk_struc
         jnz     .return
 ; 6b. Load input parameters and call the worker procedure.
         mov     eax, [edx+IOCTL.input]
         movzx   ebx, [eax+del_disk_struc.DiskId]
         call    del_disk
         jb      .return
 ; 3. Allocate memory for the disk, store the pointer in edi.
 ; If failed, return the corresponding error code.
         mov     eax, esi
         shl     eax, 9
-        stdcall KernelAlloc, eax
+        invoke  KernelAlloc, eax
         mov     edi, eax
         test    eax, eax
         mov     al, ERROR_NO_MEMORY
         jz      .return
 ; 4. Store the pointer and the size in the global variables.
 ; Play extra safe and store new value only if old value is zero.
         xor     eax, eax
         lock cmpxchg [disk_pointers+ebx*4], edi
         jz      @f
 ; Otherwise, free the allocated memory and return the corresponding error code.
-        stdcall KernelFree, edi
+        invoke  KernelFree, edi
         mov     al, ERROR_INVALID_ID
         jmp     .return
 @@:
         mov     [disk_sizes+ebx*4], esi
 ; 5. Call the worker procedure for formatting this disk.
         push    'tmp'
         mov     eax, esp ; eax points to 'tmp' + zero byte + zero dword
         lea     ecx, [ebx+'0'] ; ecx = digit
         mov     [eax+3], cl ; eax points to 'tmp#' + zero dword
 ; 6b. Call the kernel API. Use disk id as 'userdata' parameter for callbacks.
-        stdcall DiskAdd, disk_functions, eax, ebx, 0
+        invoke  DiskAdd, disk_functions, eax, ebx, 0
 ; 6c. Restore the stack after 6a.
         pop     ecx ecx
 ; 6c. Check the result. If DiskAdd has failed, cleanup and return
 ; ERROR_NO_MEMORY, this is the most probable or even the only reason to fail.
         test    eax, eax
         jnz     @f
         mov     [disk_sizes+ebx*4], 0
         mov     [disk_pointers+ebx*4], 0
-        stdcall KernelFree, edi
+        invoke  KernelFree, edi
         mov     al, ERROR_NO_MEMORY
         jmp     .return
 @@:
         push    eax
 ; 6d. Notify the kernel that media is inserted.
-        stdcall DiskMediaChanged, eax, 1
+        invoke  DiskMediaChanged, eax, 1
 ; 6e. Disk is fully configured; store its handle in the global variable
 ; and return success.
         pop     [disk_handles+ebx*4]
         xor     eax, eax
 ; 7. Return.
 ; 3. Check that the handle is non-zero.
 ; Otherwise, return the corresponding error code.
         test    edx, edx
         jz      .return
 ; 4. Delete the disk from the system.
-        stdcall DiskDel, edx
+        invoke  DiskDel, edx
 ; 5. Return success.
 ; Note that we can't free memory yet; it will be done in tmpdisk_close.
         xor     eax, eax
 .return:
         retn
 times max_num_disks dd 0
 ; disk_sizes = array of disk sizes
 label disk_sizes dword
 times max_num_disks dd 0
 version         dd      0x00060006
-my_service      db      'tmpdisk',0
-; uninitialized data
-; actually, not used here

Subversion Repositories Kolibri OS

(root)/drivers/disk/tmpdisk.asm – Rev 2644 → 5044