1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <string.h>
#include <check.h>
#include "common.h"
#include "xstrtol.h"
#define STREQ(a, b) (strcmp (a, b) == 0)
size_t get_sector_size (void)
{
char *p = getenv ("PARTED_SECTOR_SIZE");
size_t ss = 512;
unsigned long val;
if (p
&& xstrtoul (p, NULL, 10, &val, NULL) == LONGINT_OK
&& val % 512 == 0)
ss = val;
return ss;
}
PedExceptionOption
_test_exception_handler (PedException* e)
{
ck_abort_msg("Exception of type %s has been raised: %s",
ped_exception_get_type_string (e->type),
e->message);
return PED_EXCEPTION_UNHANDLED;
}
char*
_create_disk (const off_t n_bytes)
{
char* filename = strdup ("parted-test-XXXXXX");
if (filename == NULL)
return NULL;
int fd = mkstemp (filename);
if (fd < 0) {
free_filename:
free (filename);
return NULL;
}
FILE* disk = fdopen (fd, "w");
if (disk == NULL)
goto free_filename;
int fail = (fseek (disk, n_bytes, SEEK_SET) != 0
|| fwrite ("", sizeof (char), 1, disk) != 1);
if (fclose (disk) != 0 || fail)
goto free_filename;
return filename;
}
PedDisk*
_create_disk_label (PedDevice *dev, PedDiskType *type)
{
PedDisk* disk = NULL;
/* Create the label */
disk = ped_disk_new_fresh (dev, type);
ck_assert_msg(disk != NULL, "Failed to create a label of type: %s",
type->name);
ck_assert_msg(ped_disk_commit(disk) != 0,
"Failed to commit label to device");
return disk;
}
int
_implemented_disk_label (const char *label)
{
/* FIXME: these have minor problems, so skip them, temporarily. */
if (STREQ (label, "amiga")) return 0;
if (STREQ (label, "atari") && get_sector_size() != 512)
return 0;
/* Not implemented yet */
if (STREQ (label, "aix")) return 0;
if (STREQ (label, "pc98")) return 0;
return 1;
}
|
