Резидент с PSP

text segment 'code' 
assume cs:text,ds:text,es:text
       org 256 
main proc
    jmp init  
;Поля данных резидентной секции
old_2fh dd 0    ; Ячейка для сохранения системного вектора 2Fh
old_09h dd 0    ; Ячейка для сохранения системного вектора 09h
new_09h proc
; Сохраним используемые регистры
    push es
    push ax
    push bx
    push cx
    push dx
    push ds
    push cs ; Настроим DS на наш сегмент для простоты программирования
    pop ds
    xor ax,ax                 ; Чтение флагов состояния
    MOV ES,AX                 ; клавиатуры в регистр AL и
    MOV AL,ES:[417h]    ; определение нажаты ли клавиши
    AND AL,00001100b    ; Ctrl-Alt, если нет, то
    CMP AL,00001100b    ; возврат управления
    JNZ exit
    mov ax, 0013h
    int 10h
    mov ax,0900h    ; Показываем сообщение фио
    mov dx,offset msg_fio
    int 21h
; Восстановим использовавшиеся регистры 
exit: 
    pop ds 
    pop dx 
    pop cx 
    pop bx 
    pop ax 
    pop es
    jmp cs:old_09h                ; Передадим управление системному обработчику "int09h"
    new_09h endp                  ; Конец процедуры
 
; Обработчик мультиплексорного прерывания 
new_2fh proc
    cmp ah,0f1h ; Проверим номер функции мультиплексорного прерывания
    jne out_2fh ; Не наша - на выход
    cmp al,00h  ; Подфункция проверки на повторную установку?
    je inst               ; Да, сообщим о невозможности повторной установки
    cmp al,01h  ;Подфункция выгрузки? 
    je off                     ; Да - на выгрузку
    jmp short out_2fh   ; Неизвестная подфункция, на выход
inst: 
    mov al,0ffh            ; Программа уже установлена
    iret                 ; Выход из прерывания
out_2fh:
    jmp cs:old_2fh ; Переход в следующий по цепочке обработчик прерывания 2Fh
; Выгрузим программу из памяти, предварительно восстановив все перехваченные ею векторы
; Сохраним используемые регистры 
off: 
    push ds 
    push es 
    push dx 
; Восстановим вектор 09h  ; 
    mov ax,2509h          ;Функция установки вектора
    lds dx,cs:old_09h    ; Заполним DS:DX
    int 21h 
; Восстановим вектор 2fh 
    mov ax,252fh          ;Функция установки вектора
    lds dx,cs:old_2fh     ;Заполним DS:DX
    int 21h
; Получим из PSP адрес собственного окружения и выгрузим его 
    mov es,cs:2ch   ; ES окружение
    mov ah,49h  ; Функция освобождения блока памяти
    int 21h
; Выгрузим теперь саму программу
    push cs              ; Загрузим в ES содержимое CS, т.е. сегментный адрес PSP
    pop es
    mov ah,49h  ; Функция освобождения блока памяти
    int 21h
; Восстановим использовавшиеся регистры 
    pop dx 
    pop es 
    pop ds
    iret                  ;Возврат в вызвавшую программу
new_2fh endp        ; Конец процедуры обработки прерывания 2Fh
end_res=$   ; Смещение конца резидентной части программы
main endp 
tail db 'off'              ; Ожидаемый хвост команды
flag db 0                ;Флаг требования выгрузки
 
; Процедура инициализации 
init proc
    mov cl,es:80h         ;Получим длину хвоста PSP
    cmp cl,0                ;Длина хвоста = 0 ?
    je live                     ;Да программа запущена без параметров
    xor ch,ch               ; Теперь СХ=СL=длина хвоста
    mov di,81h             ; DS:SI хвост в PSP
    lea si,tail               ; DS:SI поле tail
    mov al,' '                ;Уберём пробелы из начала хвоста
repe  scasb           ;Сканируем хвост, пока пробелы
    dec di                   ;DI первый символ после пробелов
    mov cx,3               ;Ожидаемая длина параметра
repe  cmpsb          ;Сравниваем введённый хвост с ожидаемым 
    jne live                  ;Введена неизвестная команда
    inc flag                  ;Введено "off", установим флаг запроса на выгрузку
 
; Проверим, не установлена ли уже данная программа 
live:
    mov ah,0f1h ; Установим нашу функцию
    mov al,0                  ; и подфункцию на наличие нашей программы в оперативной памяти
    int 2fh 
    cmp al,0ffh            ; Программа установлена?
    je installed           ; Да, при наличии запроса на выгрузку её можно выгрузить
 
; Сохраним вектор 2fh
    mov ax,352fh                      ; Функция получения вектора 2fh
    int 21h
    mov word ptr cs:old_2fh,bx        ; Сохраним смещение системного обработчика 
    mov word ptr cs:old_2fh+2,es    ; Сохраним сегмент системного обработчика
; Заполним вектор 2fh
    mov ax,252fh                     ; Функция установления вектора прерывания 2fh
    mov dx,offset new_2fh              ; Смещение нашего обработчика 
    int 21h 
 
; Сохраним вектор 09h
    mov ax,3509h                       ; Функция получения вектора 09h
    int 21h
    mov word ptr cs:old_09h,bx        ; Сохраним смещение системного обработчика 
    mov word ptr cs:old_09h+2,es    ; Сохраним сегмент системного обработчика ; Заполним вектор 09h
    mov ax,2509h                       ; Функция установления вектора прерывания 09h
    mov dx,offset new_09h              ; Смещение нашего обработчика 
    int 21h
 
; Выведем на экран информационное сообщение 
by: 
    mov ah,09h  ; Функция вывода на экран
    lea dx,mes  ; DS:DX адрес строки
    int 21h
    mov ax,3100h        ; Функция "завершиться и остаться резидентным"
    mov dx,(end_res-main+10fh)/8   ;Размер в параграфах
installed:
    cmp flag, 1 ; Запрос на выгрузку установлен?
    je unins                  ; Да, на выгрузку
; Выведем на экран информационное сообщение
    mov ah, 09h ; Функция вывода на экран
    lea dx,mes1 ; DS:DX адрес строки
    int 21h
    mov ax,4c01h    ; Функция завершения с кодом возврата
    int 21h
unins:
; Перешлём в первую (резидентную) копию программы запрос на выгрузку 
    mov ax,0f101h   ; Наша функция с подфункцией выгрузки
    int 2fh     ; Мультиплексное прерывание
; Выведем на экран информационное сообщение
    mov ah,09h  ; Функция вывода на экран
    lea dx,mes2 ; DS:DX адрес строки
    int 21h
    mov ax,4c00h    ; Функция завершения программы
    int 21h
 
mes db 'Program installed'
mes1 db 'Program already installed'
mes2 db 'Program is uninstalled'
msg_fio db 'H=E=L=L=O'
init endp
text ends
end main