Code: Select all

#include "..\lib\kolibri.h" 
#include "..\lib\strings.h" 
#include "..\lib\file_system.h"

inline fastcall dword GetFreeRAM()
{
	$mov eax, 18
	$mov ebx, 16
	$int 0x40
	//return eax = размер свободной памяти в килобайтах
}

inline fastcall dword LoadDriver( ECX) //ECX - имя драйвера
{
	$mov eax, 68
	$mov ebx, 16
	$int 0x40
	//return 0 - неудача, иначе eax = хэндл драйвера 
}

inline fastcall dword RuleDriver( ECX) //указатель на управляющую структуру
{
	$mov eax, 68
	$mov ebx, 17
	$int 0x40
	//return eax = определяется драйвером
}

struct ioctl_struct
{
	dword	handle;
	dword	io_code;
	dword	input;
	dword	inp_size;
	dword	output;
	dword	out_size;
};

#define DEV_ADD_DISK 1	//input = structure add_disk_struc
#define DEV_DEL_DISK 2	//input = structure del_disk_struc

struct add_disk_struc
{
	dword DiskSize; // disk size in sectors, 1 sector = 512 bytes
	// Note: .DiskSize is the full size, including FAT service data.
	// Size for useful data is slightly less than this number.
	unsigned char DiskId; // from 0 to 9
};

struct del_disk_struc
{
	unsigned char DiskId; //from 0 to 9
};



void main()
{   
	int driver_handle;

	driver_handle = LoadDriver("tmpdisk");
	if (driver_handle==0)
	{
		debug("tmpdisk: driver loading failed");
		ExitProcess();
		return;
	}
	debug("tmpdisk: driver loaded successfully");
	
	//del disk
	del_disk_struc.DiskId   = 1;	
	//add disk
	add_disk_struc.DiskId   = 1;
	add_disk_struc.DiskSize = 200;
	
	ioctl_struct.handle   = driver_handle;
	ioctl_struct.io_code  = DEV_ADD_DISK;
	ioctl_struct.input    = #add_disk_struc;
	ioctl_struct.inp_size = sizeof(#add_disk_struc);
	ioctl_struct.output   = 0;
	ioctl_struct.out_size = 0;	
	
	RuleDriver(#ioctl_struct);
	
	
	switch(EAX)
	{
		case 0:
			debug("tmpdisk: disk added successfully");
			break;
		case 1:
			debug("tmpdisk: unknown IOCTL code, wrong input/output size...");
			break;
		case 2:
			debug("tmpdisk: DiskId must be from 0 to 9");
			break;
		case 3:
			debug("tmpdisk: DiskSize is too large");
			break;
		case 4:
			debug("tmpdisk: DiskSize is too small");
			break;
		case 5:
			debug("tmpdisk: memory allocation failed");
			break;
		default:
			debug("tmpdisk: unknown error O_o");
	}
	ExitProcess();
}