Написание ОС: Многозадачность

Доброго времени суток, дорогой читатель, скорее всего, ты видел мою предыдущую статью о том, что самому можно написать работоспособную ОС за достаточно короткий срок. Что же, сегодня мы поговорим о реализации многозадачности в моей ОС.

Что же, вы скорее всего не можете представить себе однозадачную ОС в 2018 году, по этому я решил поговорить о реализации многозадачности в моей ОС. И так, первое — вам надо определиться с типом многозадачности, я выбрал вытесняющую.
Что она из себя представляет? Вытесняющая многозадачность представляет собой систему распределения вычислительной мощности процессора между процессами: у каждого есть свой квант времени, у кажого есть свой приоритет. И первая проблема — какой квант по длине выбрать, как останавливать выполнение процесса по истечению кванта? На самом деле всё легко как никогда! Мы будем использовать PIT с изначально выставленной частотой в 10026 с копейками прерываний в секунду, тут же мы решаем еще одну проблему: мы уже останавливаем предыдущий процесс. И так, начнем с PIT'а.

PIT

PIT — Programmable Interval Timer — счетчик, который по достижению какого-либо запрограммированного количества инкрементов выдаёт сигнал. Так же при помощи этого таймера можно пищать пищалкой в компьютере(той штукой, что пищит после прохождения теста устройств). И так, он считает с частотой 1193182 герц, это значит, что нам надо запрограммировать его на 119(1193182/119 примерно равно 10026). Для этого надо в порт первого генератора отправить 2 байта, сначала младший байт, а потом старший:

	unsigned short hz = 119;
	outportb(0x43, 0x34);
	outportb(0x40, (unsigned char)hz & 0xFF); //Low
	outportb(0x40, (unsigned char)(hz >> 8) & 0xFF); //Hight, about 10026 times per second

Теперь стоит приступить к программированию прерывания от PIT, оно имеет IRQ 0, и после ремапа PIC'а будет 0x20м. Для IRQ первого PIC'а я написал вот такой макрос:

//PIC#0; port 0x20
#define IRQ_HANDLER(func) char func = 0x90;\
__asm__(#func ": \npusha \n call __"#func " \n movb $0x20,\
 %al \n outb %al, $0x20 \n popa  \n iret \n");\
void _## func()

Структура и процессы

И так, как вы понимаете, нам надо разработать структуру для каждого процесса, а так же структуру, которая позволяет запомнить все мои выделения памяти.
Вот, что имею я:

typedef struct _pralloc
{
	void * addr;
	struct _pralloc * next;
} processAlloc;
typedef struct
{
	void * entry;
	processAlloc *allocs;
} ELF_Process;
typedef struct __attribute__((packed)) _E {
	unsigned int eax;//4
	unsigned int ebx;//8
	unsigned int ecx;//12
	unsigned int edx;//16
	unsigned int ebp;//20
	unsigned int esp;//24
	unsigned int esi;//28
	unsigned int edi;//32
	unsigned int eflags;//36
	unsigned int state;//40
	void * startAddr;//44
	void * currentAddr;//48
	void * stack;//52
	unsigned int sse[4 * 8];//
	unsigned int mmx[2 * 8];//244
	unsigned int priority;//248
	unsigned int priorityL;//252
	void * elf_process;//256
	char ** argv;//260
	unsigned int argc;//264
	unsigned int runnedFrom;//268
	char * workingDir;//272
	unsigned int cs;//276 - pop is 4 byte in IRET
	unsigned int ds;//280
} Process;

Для начала, нам надо понять следующее: мы можем где-нибудь по глобальному адресу, к примеру, по 0xDEAD положить номер текущего запущенного процесса, тогда при выполнении любого кода мы можем быть уверены: у нас есть номер текущего запущенного процесса, это значит, что при обращении к malloc мы знаем, кому выделяем память, и сразу можем добавить адрес выделенной памяти в список allocs.

void addProcessAlloc(ELF_Process * p, void * addr)
{
	void * z = p->allocs;
	p->allocs = malloc_wo_adding_to_process(sizeof(processAlloc));
	p->allocs->addr = addr;
	p->allocs->next = z;
}

Что же, структуру таблицы с описанием процессов мы написали, что дальше, как переключать задачи?
Для начала хочу заметить, что к примеру, в обработчике локальные переменные хранятся в стеке, а значит после входа в обработчик компилятор портит нам esp. Чтобы такого не произошло создадим переменную с абсолютным адресом, и перед вызовом обработчика будем засовывать ESP туда. В обработчике нам необходимо отослать EOI первому PIC'у и найти процесс, на который нам надо переключиться(не буду описывать механизм приоритетов: он прост, как пробка). Далее — нам надо сохранить все регистры и флаги текущего процесса, по этому сразу перед засовыванием ESP в переменную сохраним все регистры(в том числе сегментные) в стек. В самом обработчике нам очень аккуратно надо их вынуть из стека, так же сохранив флаги и адрес возврата. Хочу заметить, что стек растет вверх(т.е. ESP уменьшается), это значит, что последний регистр, который вы сохранили в стек будет лежать по адресу ESP, предпоследний — ESP +4 и т.п:

Теперь нам остаётся засунуть в регистры значения регистров процесса, на который мы переключились и выполнить IRET. Profit!

Запуск процессов

При запуске процесса нам достаточно выделить стек для процесса, после чего положить в него argc и argv, адрес функции, которая будет отдано управление после завершения процесса. Так же надо установить флаги процессора в нужное вам значение, к примеру, для моей ОС это 0x216, про регистр флагов можно прочитать на википедии.

Напоследок хочу пожелать успехов, в скором времени я напишу про работу с памятью и другие интересующие вас статьи.
Удачи, и этичного хакинга!