271 lines
4.5 KiB
NASM
271 lines
4.5 KiB
NASM
format MS COFF
|
|
|
|
DEBUG equ 0
|
|
|
|
include 'proc32.inc'
|
|
include 'imports.inc'
|
|
|
|
struc IOCTL
|
|
{ .handle dd ?
|
|
.io_code dd ?
|
|
.input dd ?
|
|
.inp_size dd ?
|
|
.output dd ?
|
|
.out_size dd ?
|
|
}
|
|
|
|
virtual at 0
|
|
IOCTL IOCTL
|
|
end virtual
|
|
|
|
public START
|
|
public version
|
|
|
|
DRV_ENTRY equ 1
|
|
DRV_EXIT equ -1
|
|
|
|
MT_3B equ 0
|
|
MT_3BScroll equ 1
|
|
MT_5BScroll equ 2
|
|
|
|
PS2_DRV_VER equ 1
|
|
|
|
section '.flat' code readable align 16
|
|
|
|
|
|
proc START stdcall, state:dword
|
|
|
|
cmp [state], DRV_ENTRY
|
|
jne .fin
|
|
.init:
|
|
|
|
call detect_mouse
|
|
test eax,eax
|
|
jnz .exit
|
|
|
|
mov [MouseType],MT_3B
|
|
|
|
call try_mode_ID3
|
|
test eax,eax
|
|
jnz .stop_try
|
|
mov [MouseType],MT_3BScroll
|
|
|
|
call try_mode_ID4
|
|
test eax,eax
|
|
jnz .stop_try
|
|
mov [MouseType],MT_5BScroll
|
|
|
|
.stop_try:
|
|
|
|
mov bl, 0x20 ; read command byte
|
|
call kbd_cmd
|
|
cmp ah,1
|
|
je .exit
|
|
|
|
call kbd_read
|
|
cmp ah,1
|
|
je .exit
|
|
|
|
or al, 10b
|
|
push eax
|
|
mov bl, 0x60 ; write command byte
|
|
call kbd_cmd
|
|
cmp ah,1
|
|
je .exit
|
|
|
|
pop eax
|
|
call kbd_write
|
|
cmp ah,1
|
|
je .exit
|
|
|
|
mov al, 0xF4 ; enable data reporting
|
|
call mouse_cmd
|
|
|
|
mov bl, 0xAE ; enable keyboard interface
|
|
call kbd_cmd
|
|
|
|
stdcall AttachIntHandler, 12, irq_handler, dword 0
|
|
stdcall RegService, my_service, service_proc
|
|
ret
|
|
|
|
.fin:
|
|
;stdcall DetachIntHandler, 12, irq_handler
|
|
mov bl, 0xA7 ; disable mouse interface
|
|
call kbd_cmd
|
|
xor eax, eax
|
|
ret
|
|
|
|
.exit:
|
|
mov bl, 0xA7 ; disable mouse interface
|
|
call kbd_cmd
|
|
mov bl, 0xAE ; enable keyboard interface
|
|
call kbd_cmd
|
|
xor eax, eax
|
|
ret
|
|
endp
|
|
|
|
proc service_proc stdcall, ioctl:dword
|
|
mov edi, [ioctl]
|
|
mov eax, [edi+IOCTL.io_code]
|
|
test eax, eax
|
|
jz .getversion
|
|
cmp eax,1
|
|
jz .gettype
|
|
|
|
.err:
|
|
or eax, -1
|
|
ret
|
|
|
|
.ok:
|
|
xor eax, eax
|
|
ret
|
|
|
|
.getversion:
|
|
cmp [edi+IOCTL.out_size], 4
|
|
jb .err
|
|
mov edi, [edi+IOCTL.output]
|
|
mov dword [edi], PS2_DRV_VER ; version of driver
|
|
jmp .ok
|
|
.gettype:
|
|
cmp [edi+IOCTL.out_size], 4
|
|
jb .err
|
|
mov edi, [edi+IOCTL.output]
|
|
mov eax,[MouseType]
|
|
mov dword [edi], eax ; mouse type
|
|
jmp .ok
|
|
endp
|
|
|
|
detect_mouse:
|
|
|
|
mov bl, 0xAD ; disable keyboard interface
|
|
call kbd_cmd
|
|
cmp ah,1
|
|
je .fail
|
|
|
|
mov bl, 0xA8 ; enable mouse interface
|
|
call kbd_cmd
|
|
cmp ah,1
|
|
je .fail
|
|
|
|
mov al, 0xFF ; reset
|
|
call mouse_cmd
|
|
jc .fail
|
|
|
|
call mouse_read
|
|
jc .fail
|
|
cmp al, 0xAA
|
|
jne .fail ; dead mouse
|
|
|
|
; get device ID
|
|
call mouse_read
|
|
jc .fail
|
|
cmp al, 0x00
|
|
jne .fail ; unknown device
|
|
xor eax,eax
|
|
ret
|
|
|
|
.fail:
|
|
or eax,-1
|
|
ret
|
|
|
|
try_mode_ID3:
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xC8 ;200d
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0x64 ;100d
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0x50 ;80d
|
|
call mouse_cmd
|
|
jc .fail
|
|
|
|
mov al, 0xF2 ;Get device id
|
|
call mouse_cmd
|
|
jc .fail
|
|
|
|
call mouse_read
|
|
jc .fail
|
|
cmp al, 0x03
|
|
jne .fail
|
|
|
|
xor eax,eax
|
|
ret
|
|
.fail:
|
|
or eax,-1
|
|
ret
|
|
|
|
try_mode_ID4:
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xC8 ;200d
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xC8 ;100d
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0xF3 ;Set Sample Rate
|
|
call mouse_cmd
|
|
jc .fail
|
|
mov al, 0x50 ;80d
|
|
call mouse_cmd
|
|
jc .fail
|
|
|
|
mov al, 0xF2 ;Get device id
|
|
call mouse_cmd
|
|
jc .fail
|
|
|
|
call mouse_read
|
|
jc .fail
|
|
cmp al, 0x04
|
|
jne .fail
|
|
|
|
xor eax,eax
|
|
ret
|
|
|
|
.fail:
|
|
or eax,-1
|
|
ret
|
|
|
|
include 'ps2m_iofuncs.inc'
|
|
include 'ps2m_irqh.inc'
|
|
|
|
section '.data' data readable writable align 16
|
|
|
|
version dd 0x00050005
|
|
my_service db 'ps2mouse',0
|
|
|
|
;iofuncs data
|
|
mouse_cmd_byte db 0
|
|
mouse_nr_tries db 0
|
|
mouse_nr_resends db 0
|
|
|
|
;hid data
|
|
mouse_byte dd 0
|
|
|
|
first_byte db 0
|
|
second_byte db 0
|
|
third_byte db 0
|
|
fourth_byte db 0
|
|
|
|
;main data
|
|
MouseType dd 0
|
|
|
|
XMoving dd 0
|
|
YMoving dd 0
|
|
ZMoving dd 0
|
|
ButtonState dd 0
|
|
;timerTicks dd 0
|