task2_1

LOCALS

model tiny
.code
org 100h

start:

jmp main					; перепрыгиваем все процедуры
					dw	2144h	; сказали в памяти, что это мы
	oldVec2f		dd	0


my_handler proc 			; наш обработчик
	cmp ax, 82f2h			; если пришло то, что надо
							; с нужным магическим числом и функцией
	jne @@end

	cmp dx, 22ffh
	jne @@end

	mov ax, error_code_ok  	; выставляем код, что резидент наш
	xchg dh, dl
@@end:
	iret
my_handler endp


get_old_2f proc
	push ax
	call our35h
	push dx
	pop es
	pop ax
ret
get_old_2f endp


set_int proc  					; ставим свой обработчик my_handler на вектор 2fh
	call get_old_2f				; узнали старый вектор 2fh
	mov	word ptr [oldVec2f], bx
	mov	word ptr [oldVec2f+2], es	; записали вектор в переменную
	mov dx, offset my_handler
	call our25h_to_old_handler
ret
set_int endp

del_int proc
	call our35h
	push dx
	pop es
	cmp word ptr es:[bx-6], 2144h
	jne @@end

	push ds
	mov dx, word ptr es:[bx-4]
	mov ax, word ptr es:[bx-2]

	mov ds, ax
	;mov ax, 252fh
	;int 21h
	call our25h_to_old_handler
	pop ds

	push es
	mov ax, es:[2ch]
	mov es, ax
	mov ah, 49h
	int 21h
	pop es

	mov ah, 49h
	int 21h
@@end:
ret
del_int endp


our35h proc
	push   ES             ; сохранение ES
    xor    BX,BX          ; 0=> BX
    mov    ES,BX          ; 0=> ES, сегмент начала ОП
    mov    BX,ES:[4*2fh]  ; смещение старого обработчика
    mov    DX,ES:[4*2fh+2]; cегмент старого обработчика
    pop    ES

ret
our35h endp

our25h_to_my_handler proc
	cli								; нельзя внешние прерывания
	push es

	xor ax,ax
	push ax
	pop es
	mov ax, offset my_handler		; нашли смещение нового обработчика
;	mov bx, offset my_handler + 2	; нашли сегмент нового обработчика
	mov bx, ds
	mov es:[4*2fh], ax
	mov es:[4:2fh  + 2], bx
	pop es
	sti								; можно внешние прерывания
ret
our25h_to_my_handler endp


our25h_to_old_handler proc
	cli								; нельзя внешние прерывания
	push es
	xor ax,ax
	push ax
	pop es
	mov es:[4*2fh], dx
	mov es:[4:2fh  + 2], ds
	pop es
	sti								; можно внешние прерывания
ret
our25h_to_old_handler endp

main:  						; здесь все начинается

jmp new_start

get_nums proc
	push -1
	mov bx, 10h
	@@loop1:
		xor dx, dx
		div bx
		push dx
		cmp ax, 0
		jne @@loop1

	xor cx, cx
	mov cx, 4
	xor dx, dx
	@@loop:
		pop dx
		cmp dx, -1
		je @@end

		cmp dl, 0ah
		jge @@more

		add dl, 30h
		jmp @@write
	@@more:
		add dl, 37h

	@@write:
		mov ah, 02h
		int 21h
	jmp @@loop
@@end:
ret
get_nums endp

next_line proc
	mov ah, 09h
	lea dx, nl
	int 21h
ret
next_line endp

colon proc
	mov ah, 02h
	mov dl, 3ah
	int 21h
ret
colon endp

write_current_addr proc
	mov ax, 352fh
	int 21h

	mov ax, es
	;mov ax, offset my_handler
	call get_nums

	call colon

	mov ax, bx
	;mov ax, ds
	call get_nums
ret
write_current_addr endp

finish proc
	push ds
	mov dx, word ptr[oldVec2f]
	mov ds, word ptr [oldVec2f + 2]

	mov ax, ds
	call get_nums

	call colon

	mov ax, dx
	call get_nums
	pop ds

	call next_line

	call write_current_addr

ret
finish endp


new_start:


get_args:  					; надо прочитать ключи
	xor cx, cx
	mov cx, cs:[80h]
	sub cx, 2000h
	dec cx

	xor bx, bx
	mov si, 82h  			; адрес первого символа в аргументе
	mov bx, 2fh  			; код слеша
	cmp [si], bl
	jne bad  				; первый символ не наш - все плохо
	dec cx
	jmp ok

bad:
	jmp afail
ok:
	inc si

	xor bx, bx
	mov bx, 31h  			; код единицы
	cmp[si], bl
	je is_end_r2  			; прочитали строку /1 - первый ключ

	xor bx, bx
	mov bx, 64h  			; код единицы
	cmp[si], bl
	je is_end_del

	jmp afail 				; плохой аргумент

is_end_r2:
	dec cx
	cmp cx, 0
	je r2
	jmp afail

is_end_del:
	dec cx
	cmp cx, 0
	je del
	jmp afail

r2:
	mov ah, magic_num
	mov al, magic_func
	mov dx, magic_num_dx
	int 2fh
	cmp ax, error_code_ok
	jne set

	cmp dx, 0ff22h
	je installed

set:
	call set_int

	call finish
	mov ah, 31h 			; ставим резидента с помощью 31h функции
	mov al, 00h 			; код возврата
	mov bx, offset main
	sub bx, offset start
	add bx, 0fh
	sar bx, 4
	add bx, 10h

	mov dx, bx ;(offset r2 - offset start + 15)/16 + 1	; сколько параграфов поделить сдвигом вправо
	int 21h

del:
	call del_int
	mov ah, 9h
	lea dx, msg_del
	int 21h
	ret


afail: 						; если неправильный аргумент
	mov ah, 9h
	lea dx, argmess
	int 21h
	ret



installed:					; если резидент уже загружен
	mov ah, 9h
	lea dx, msg_n_inst
	int 21h
	ret


	magic_num		db	82h
	magic_func		db	0f2h
	magic_num_dx	dw	22ffh
	error_code_ok	dw	0f2h
	argmess			db	"Error: Incorrect args",	24h
	msg_n_inst		db	"Allready installed", 		0dh, 0ah, '$'
	msg_inst 		db 	"Resident installed", 		0dh, 0ah, '$'
	msg_del 		db 	"Deleted", 		0dh, 0ah, '$'
	nl				db								0dh, 0ah, '$'
end start


; почему сначала смещение, а потом сегмент
; почему везде умножаем на 4, когда делаем сами 25h