Форум программистов [Powered by Invision Power Board]

Код

#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <linux/types.h>

#define SIGNATURE 0xAA55
#define DEVICE "/dev/hda"

#define PT_SIZE 0x10

    struct systypes {
        __u8 part_type;
        __u8 *part_name;
    };

    struct systypes i386_sys_types[] = {
        {0x00, "Empty"},
        {0x01, "FAT12"},
        {0x04, "FAT16 <32M"},
        {0x05, "Extended"},
        {0x06, "FAT16"},
        {0x0b, "FAT32"},
        {0x0c, "FAT32 (LBA)"},
        {0x0e, "FAT16 (LBA)"},
        {0x0f, "Win Ext'd (LBA)"},
        {0x82, "Linux swap"},
        {0x83, "Linux"},
        {0x85, "Linux extended"},
        {0x07, "HPFS/NTFS"}
    };


#define PART_NUM (sizeof(i386_sys_types) / sizeof(i386_sys_types[0]))

int hard;    
 __u8 mbr[512];    

#define MAX_PART 20

    struct pt_struct {
        __u8 bootable;
        __u8 start_part[3];
        __u8 type_part;
        __u8 end_part[3];
        __u32 sect_before;
        __u32 sect_total;
    } pt_t[MAX_PART];

    int main()
    {
        int i = 0;
        __u64 seek;

        hard = open(DEVICE, O_RDONLY);
        if(hard < 0) {
            perror("open");
            exit(-1);
        }
    
        read_main_ptable();

        if(check_sign() < 0) {
            printf("Not valid signature!\n");
            exit(-1);
        }

        for(; i < 4; i++) {
            if((pt_t[i].type_part == 0xF) || \
        (pt_t[i].type_part == 0x5) || \
        (pt_t[i].type_part == 0x0C)) {
            seek = (__u64)pt_t[i].sect_before * 512;
            read_ext_ptable(seek);
            break;
            }
        }

        pt_info();
        return 0;
    }

    int check_sign()
    {
        __u16 sign = 0;

        memcpy((void *)&sign, (void *)(mbr + 0x1FE), 2);

        printf("Сигнатура - 0x%X\n", sign);

        if(sign != SIGNATURE) return -1;
        return 0;
    }

    void read_main_ptable()
    {
        if(read(hard, mbr, 512) < 0) {
            perror("read");
            close(hard);
            exit(-1);
        }

        memset((void *)pt_t, 0, (PT_SIZE * 4));
        memcpy((void *)pt_t, mbr + 0x1BE, (PT_SIZE * 4));

        return;
    }

    void read_ext_ptable(__u64 seek)
    {
        int num = 4;
           
        __u8 smbr[512];

        for(;;num++) {

            memset((void *)smbr, 0, 512);
            pread64(hard, smbr, 512, seek);

            memset((void *)&pt_t[num], 0, PT_SIZE * 2);
            memcpy((void *)&pt_t[num], smbr + 0x1BE, PT_SIZE * 2);

            pt_t[num].sect_before += (seek / 512);
     
            if(!(pt_t[num + 1].type_part)) break;
        
            seek = ((__u64)(pt_t[num].sect_before + pt_t[num].sect_total)) * 512;
        }
        return;
    }

    void pt_info()
    {
        int i, n;
     
        printf("\nNum\tBootable\tStart\tTotal\t\tId\tType\n");

        for(i = 0; i < MAX_PART; i++) {

            if(!pt_t[i].type_part) {
        if(i < 4) continue;
        else break;
            }

            printf("%d\t", i + 1);

            if(pt_t[i].bootable == 0x80) printf("*\t");
            else printf("\t");

            printf("%12u\t", pt_t[i].sect_before);
            printf("%8u\t", pt_t[i].sect_total);
            printf("0x%.2X\t", pt_t[i].type_part);

            for(n = 0; n < PART_NUM; n++) {
         if(pt_t[i].type_part == i386_sys_types[n].part_type) {
                 printf("%s\n", i386_sys_types[n].part_name);
                 break;
         }
            }
            if(n == PART_NUM) printf("unknown type\n");
        }
        return;
    }