mkgpt.c

#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <string.h>

#include "device.h"
#include "gpt.h"
#include "mbr.h"

void usage()
{
    printf("mkgpt [--force] [device]\n");
}

int main(int argc, char *argv[])
{
    int force = 0;
    char *devname = NULL;
    struct device dev;

    if (argc > 3)
    {
        printf("Too many arguments.\n");
        usage();
        return -1;
    }
    else if(argc == 1)
    {
        printf("Missing parameters.\n");
        usage();
        return -1;
    }
    else
    {
        for(int i = 1; i < argc; i++)
        {
            if(argv[i][0] == '-')
            {
                if(strcmp("--force", argv[i]) == 0)
                {
                    if(argc == 2)
                    {
                        printf("You must specify a device.\n");
                        usage();
                        return -1;
                    }
                    force = 1;
                }
                else
                {
                    printf("Unknown parameter %s\n", argv[i]);
                    usage();
                    return -1;
                }
            }
            else
            {
                devname = argv[i];
            }
        }
    }
    if(open_device(devname, &dev) < 0)
    {
        printf("Unable to open device: %s\n", strerror(errno));
        return -1;
    }

    if(!force)
    {
        if(has_partition_scheme(&dev) != PARTSCHEME_IS_NOTHING)
        {
            printf("An existing partition scheme was found on this device. If you really want to proceed and DESTROY YOUR PARTITIONS, use the --force parameter.\n");
            close_device(&dev);
            return -1;
        }
    }
    if(initialize_gpt(&dev) == -1)
    {
       printf("Error initializing GPT: %s\n", strerror(errno));
       close_device(&dev);
       return -1;
    }
    close_device(&dev);
    return 0;
}