second_stage.asm

[BITS 16]

; TODO: - implement APM shutdown http://wiki.osdev.org/APM
;       - encapsulation for functions

MOV AX, CS ; initialize
MOV DS, AX ; segments
MOV ES, AX ; correctly, ALWAYS gen

CALL Clear_screen
CALL Draw_rect
CALL Delay
CALL Clear_screen
CALL Reset_cursor
MOV SI, INTRO
CALL PrintShell
MOV SI, EMPTY_CHAR
CALL PrintShell
CALL Date
CALL Convert_century_os
CALL Convert_year_os
MOV SI, DATE_FIELD_OS
CALL PrintShell
MOV SI, EMPTY_LINE
CALL PrintShell
MOV SI, BUNA_SEARA
CALL Print
MOV SI, EMPTY_LINE
CALL PrintShell
MOV SI, EMPTY_LINE
CALL PrintShell
MOV SI, OPTIONS
CALL PrintShell
MOV SI, EMPTY_LINE
CALL PrintShell
MOV SI, PLEASE
CALL PrintShell
CALL Read


Draw_rect:
	PUSHA
	MOV AH, 0xC
	MOV BL, 0x32
	MOV CX, 0x140
	MOV DX, 0x12C
	MOV AL, 0xE
	Draw1:
		CALL Delay_char
		INT 0x10
		DEC BL
		DEC CX
		DEC DX
		CMP BL, 0
		JE Restore_var1
		JMP Draw1
	Restore_var1:
		MOV BL, 0x32
		MOV CX, 0x140
		MOV DX, 0x12C
		JMP Draw2
	Draw2:
		CALL Delay_char
		INT 0x10
		DEC BL
		INC CX
		DEC DX
		CMP BL, 0
		JE Restore_var2
		JMP Draw2
	Restore_var2:
		MOV BL, 0x32
		MOV CX, 0x10E
		MOV DX, 0xFA
		JMP Draw3
	Draw3:
		CALL Delay_char
		INT 0x10
		DEC BL
		INC CX
		DEC DX
		CMP BL, 0
		JE Restore_var3
		JMP Draw3
	Restore_var3:
		MOV BL, 0x32
		MOV CX, 0x172
		MOV DX, 0xFA
		JMP Draw4
	Draw4:
		CALL Delay_char
		INT 0x10
		DEC BL
		DEC CX
		DEC DX
		CMP BL, 0
		JE Exit_draw
		JMP Draw4
Exit_draw:
	POPA
	RET

FreeDraw:
	CALL Clear_screen
	MOV AH, 0xC
	MOV BH, 0x0
	MOV CX, 0x140
	MOV DX, 0x12C
	MOV AL, 0xB
	Check_keypress:
		INT 0x10
		MOV AH, 0x0
		INT 0x16
		CMP AL, 0x77
		JE draw_up
		CMP AL, 0x73
		JE draw_down
		CMP AL, 0x61
		JE draw_left
		CMP AL, 0x64
		JE draw_right
		CMP AL, 0x63
		JE exit_FreeDraw
	draw_up:
		DEC DX
		MOV AH, 0xC
		MOV BH, 0x0
		MOV AL, 0xB
		JMP Check_keypress
	draw_down:
		INC DX
		MOV AH, 0xC
		MOV BH, 0x0
		MOV AL, 0xB
		JMP Check_keypress
	draw_left:
		DEC CX
		MOV AH, 0xC
		MOV BH, 0x0
		MOV AL, 0xB
		JMP Check_keypress
	draw_right:
		INC CX
		MOV AH, 0xC
		MOV BH, 0x0
		MOV AL, 0xB
		JMP Check_keypress
	exit_FreeDraw:
		CALL Clear_screen
		CALL Reset_cursor
		MOV SI, SHELL_PROMPT
		CALL PrintShell
		XOR CL, CL
		JMP ReadShell

Note_pad_init:
	CALL Clear_screen
	CALL Reset_cursor
	JMP Note_pad
Note_pad:
	MOV AH, 0x0
	INT 0x16
	CALL PrintCharInfoShell
	CMP AL, 0x1B
	JE exit_note_pad
	CMP AL, 0xD
	JE new_line_notepad
	JMP Note_pad
new_line_notepad:
	MOV SI, EMPTY_LINE
	CALL PrintInfoShell
	JMP Note_pad
exit_note_pad:
	CALL Clear_screen
	CALL Reset_cursor
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell

Clear_screen:
	PUSHA
	MOV AH, 0x06
	MOV AL, 0
	MOV BH, 0
	MOV CX, 0
	MOV DH, 0x1E
	MOV DL, 0x50
	INT 0x10
	POPA
	RET

Reset_cursor:
	PUSHA
	MOV AH, 0x02
	MOV BH, 0
	MOV DH, 0
	MOV DL, 0
	INT 0x10
	POPA
	RET

PrintChar:
	CALL Delay_char
	MOV BL, 0xA
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

PrintCharShell:
	MOV BL, 0xA
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

PrintCharWarning:
	CALL Delay_char
	MOV BL, 0xC
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

PrintCharWarningShell:
	MOV BL, 0xC
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

PrintCharInfo:
	CALL Delay_char
	MOV BL, 0xB
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

PrintCharInfoShell:
	MOV BL, 0xB
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	RET

Print:
	PUSHA
	next_character:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_print
		CALL PrintChar
		JMP next_character
	exit_print:
	POPA
	RET

PrintShell:
	PUSHA
	next_characterShell:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_printShell
		CALL PrintCharShell
		JMP next_characterShell
	exit_printShell:
	POPA
	RET

PrintWarning:
	PUSHA
	next_characterWarning:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_printWarning
		CALL PrintCharWarning
		JMP next_characterWarning
	exit_printWarning:
	POPA
	RET

PrintWarningShell:
	PUSHA
	next_characterWarningShell:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_printWarningShell
		CALL PrintCharWarningShell
		JMP next_characterWarningShell
	exit_printWarningShell:
	POPA
	RET

PrintInfo:
	PUSHA
	next_characterInfo:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_printInfo
		CALL PrintCharInfo
		JMP next_characterInfo
	exit_printInfo:
	POPA
	RET

PrintInfoShell:
	PUSHA
	next_characterInfoShell:
		MOV AL, [SI]
		INC SI
		OR AL, AL
		JZ exit_printInfoShell
		CALL PrintCharInfoShell
		JMP next_characterInfoShell
	exit_printInfoShell:
	POPA
	RET

Read:
	MOV AH, 0x0
	INT 0x16
	CMP AL, 0x52
	JE read_exit
	CMP AL, 0x49
	JE read_exit_shell
	JMP Read
read_exit:
	MOV BL, 0xC
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, SEARA_BUNA
	CALL PrintWarning
	CALL Delay
	MOV SI, POINT
	CALL PrintWarning
	CALL Delay
	MOV SI, POINT
	CALL PrintWarning
	CALL Delay
	MOV SI, POINT
	CALL PrintWarning
	CALL Delay
	JMP Reboot
read_exit_shell:
	MOV BL, 0xB
	MOV BH, 0
	MOV AH, 0xE
	INT 0x10
	MOV SI, EMPTY_LINE
	CALL PrintShell
	JMP NewLineShell

ReadShell:
	MOV AH, 0x0
	INT 0x16
	CMP AL, 0x08  ;Backspace
	JE Delete_char
	CMP AL, 0xD  ;CRLF
	JE Run_command
	CALL PrintCharShell
	MOV CL, AL
	JMP ReadShell
Run_command:
	CMP CL, 0
	JE NewLineShell
	CMP CL, 0x3F ;?
	JE h_print_help
	; Make any character lowercase by flipping bits 6 and 7 to 1
	OR CL, 01100000b
	CMP CL, 0x6E ;n
	JE Note_pad_init
	CMP CL, 0x67 ;g
	JE FreeDraw
	CMP CL, 0x74 ;t
	JE t_print_time
	CMP CL, 0x64 ;d
	JE d_print_date
	CMP CL, 0x68 ;h
	JE h_print_help
	CMP CL, 0x72 ;r
	JE read_exit
	CMP CL, 0x63 ;c
	JE c_clear_screen
	JMP Bad_command
Delete_char:
	CALL Check_cursor_position
	CALL PrintCharShell
	MOV AL, ' '
	CALL PrintCharShell
	MOV AL, 0x08
	CALL PrintCharShell
	XOR CL, CL
	JMP ReadShell
Bad_command:
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, ERR_COMMAND
	CALL PrintWarningShell
	JMP NewLineShell
NewLineShell:
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell
t_print_time:
	CALL Time
	CALL Convert_hours
	CALL Convert_minutes
	CALL Convert_seconds
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, TIME_FIELD
	CALL PrintInfoShell
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell
d_print_date:
	CALL Date
	CALL Convert_month
	CALL Convert_day
	CALL Convert_century
	CALL Convert_year
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, DATE_FIELD
	CALL PrintInfoShell
	MOV SI, EMPTY_LINE
	CALL PrintShell
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell
h_print_help:
	MOV SI, HELP
	CALL PrintInfoShell
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell
c_clear_screen:
	CALL Clear_screen
	CALL Reset_cursor
	MOV SI, SHELL_PROMPT
	CALL PrintShell
	XOR CL, CL
	JMP ReadShell

Check_cursor_position:
	PUSHA
	MOV AH, 0x3
	MOV BH, 0
	INT 0x10
	CMP DL, 3
	JE Fix_cursor_position
	POPA
	RET
Fix_cursor_position:
	MOV AH, 0x2
	MOV DL, 3
	INT 0x10
	POPA
	XOR CL, CL
	JMP ReadShell

Reboot:
	INT 19h

Delay:
	PUSHA
	MOV BL, 10h
	MOV AH, 0
	INT 0x1A
	ADD BX, DX
	tloop:
		MOV AH, 0
		INT 0x1A
		CMP DX, BX
		JG tloop2
		JMP tloop
	tloop2:
		POPA
		RET

Delay_char:
	PUSHA
	MOV BL, 0
	MOV AH, 0
	INT 0x1A
	ADD BX, DX
	tloop3:
		MOV AH, 0
		INT 0x1A
		CMP DX, BX
		JG tloop4
		JMP tloop3
	tloop4:
		POPA
		RET

Date:
	MOV AH, 0x04
	INT 0x1A
	RET
	;leaves CF unchanged if successful
	; TODO: CF should be cleared before calling this function
	;CH - CENTURY
	;CL - YEAR
	;DH - MONTH
	;DL - DAY
Convert_month:
	MOV BH, DH
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD + 3], BH
	MOV BH, DH
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD + 4], BH
	RET
Convert_day:
	MOV BH, DL
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD], BH
	MOV BH, DL
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD + 1],BH
	RET
Convert_century:
	MOV BH, CH
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD + 6], BH
	MOV BH, CH
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD + 7], BH
	RET
Convert_year:
	MOV BH, CL
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD + 8], BH
	MOV BH, CL
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD + 9], BH
	RET
Convert_century_os:
	PUSHA
	MOV BH, CH
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD_OS], BH
	MOV BH, CH
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD_OS + 1], BH
	POPA
	RET
Convert_year_os:
	PUSHA
	MOV BH, CL
	SHR BH, 4
	ADD BH, 30H
	MOV [DATE_FIELD_OS + 2], BH
	MOV BH, CL
	AND BH, 0FH
	ADD BH, 30H
	MOV [DATE_FIELD_OS + 3], BH
	POPA
	RET

Time:
	MOV AH, 02H
	INT 1AH
	RET
	;leaves CF unchanged if successful
	; TODO: CF should be cleared before calling this function
	;CH - HOURS
	;CL - MINUTES
	;DH - SECONDS
Convert_hours:
	MOV BH, CH
	SHR BH, 4
	ADD BH, 30H
	MOV [TIME_FIELD], BH
	MOV BH, CH
	AND BH, 0FH
	ADD BH, 30H
	MOV [TIME_FIELD + 1], BH
	RET
Convert_minutes:
	MOV BH, CL
	SHR BH, 4
	ADD BH, 30H
	MOV [TIME_FIELD + 3], BH
	MOV BH, CL
	AND BH, 0FH
	ADD BH, 30H
	MOV [TIME_FIELD + 4], BH
	RET
Convert_seconds:
	MOV BH, DH
	SHR BH, 4
	ADD BH, 30H
	MOV [TIME_FIELD + 6], BH
	MOV BH, DH
	AND BH, 0FH
	ADD BH, 30H
	MOV [TIME_FIELD + 7], BH
	RET



INTRO:
    db 0x40, "BLidOS", 0
BUNA_SEARA:
    db "Hello, World!", 0
OPTIONS:
	db "(R) Reboot", 0dh, 0ah,"(I) Interactive shell / Recovery", 0dh, 0ah,"(B) Boot to OS", 0
PLEASE:
	db "Please choose an option: ", 0
SEARA_BUNA:
    db 'Shutting down', 0
SHELL_PROMPT:
    db '#> ', 0
EMPTY_LINE:
    db 0dh, 0ah, 0
EMPTY_CHAR:
	db ' ',0
POINT:
	db '.', 0
BAR:
	db 0x7C, 0
HELP:
	db 0dh, 0ah, 0dh, 0ah, "    h -- show this help summary", 0dh, 0ah, "    g -- load PaintBleed (press c to exit)", 0dh, 0ah, "    n -- load bVim (press ESC to exit)", 0dh, 0ah, "    t -- print time", 0dh, 0ah, "    d -- print date", 0dh, 0ah, "    c -- clear screen", 0dh, 0ah, "    r -- reboot", 0dh, 0ah, 0dh, 0ah, 0
DATE_FIELD:
	db '00/00/0000', 0
TIME_FIELD:
	db '00:00:00', 0
DATE_FIELD_OS:
	db '0000', 0
ERR_COMMAND:
	db 'Invalid command! Type h for help', 0

times 2048 -( $ - $$ ) db 0