fsck.cramfs: use xalloc.h

Signed-off-by: Sami Kerola <kerolasa@iki.fi>
This commit is contained in:
Sami Kerola 2011-07-20 22:03:47 +02:00
parent 11b7698c27
commit fc8dc41054
1 changed files with 16 additions and 24 deletions

View File

@ -36,6 +36,15 @@
/* compile-time options */ /* compile-time options */
//#define INCLUDE_FS_TESTS /* include cramfs checking and extraction */ //#define INCLUDE_FS_TESTS /* include cramfs checking and extraction */
/* Exit codes used by fsck-type programs */
#define FSCK_OK 0 /* No errors */
#define FSCK_NONDESTRUCT 1 /* File system errors corrected */
#define FSCK_REBOOT 2 /* System should be rebooted */
#define FSCK_UNCORRECTED 4 /* File system errors left uncorrected */
#define FSCK_ERROR 8 /* Operational error */
#define FSCK_USAGE 16 /* Usage or syntax error */
#define FSCK_LIBRARY 128 /* Shared library error */
#include <stdio.h> #include <stdio.h>
#include <stdarg.h> #include <stdarg.h>
#include <stdint.h> #include <stdint.h>
@ -59,6 +68,9 @@
#include "blkdev.h" #include "blkdev.h"
#include "c.h" #include "c.h"
#define XALLOC_EXIT_CODE FSCK_ERROR
#include "xalloc.h"
static const char *progname = "cramfsck"; static const char *progname = "cramfsck";
static int fd; /* ROM image file descriptor */ static int fd; /* ROM image file descriptor */
@ -69,15 +81,6 @@ static int opt_verbose = 0; /* 1 = verbose (-v), 2+ = very verbose (-vv) */
char *extract_dir = ""; /* extraction directory (-x) */ char *extract_dir = ""; /* extraction directory (-x) */
/* Exit codes used by fsck-type programs */
#define FSCK_OK 0 /* No errors */
#define FSCK_NONDESTRUCT 1 /* File system errors corrected */
#define FSCK_REBOOT 2 /* System should be rebooted */
#define FSCK_UNCORRECTED 4 /* File system errors left uncorrected */
#define FSCK_ERROR 8 /* Operational error */
#define FSCK_USAGE 16 /* Usage or syntax error */
#define FSCK_LIBRARY 128 /* Shared library error */
#define PAD_SIZE 512 #define PAD_SIZE 512
#ifdef INCLUDE_FS_TESTS #ifdef INCLUDE_FS_TESTS
@ -251,10 +254,7 @@ static void test_crc(int start)
int retval; int retval;
size_t length = 0; size_t length = 0;
buf = malloc(4096); buf = xmalloc(4096);
if (!buf) {
err(FSCK_ERROR, _("malloc failed"));
}
lseek(fd, start, SEEK_SET); lseek(fd, start, SEEK_SET);
for (;;) { for (;;) {
retval = read(fd, buf, 4096); retval = read(fd, buf, 4096);
@ -317,11 +317,8 @@ static void *romfs_read(unsigned long offset)
static struct cramfs_inode *cramfs_iget(struct cramfs_inode * i) static struct cramfs_inode *cramfs_iget(struct cramfs_inode * i)
{ {
struct cramfs_inode *inode = malloc(sizeof(struct cramfs_inode)); struct cramfs_inode *inode = xmalloc(sizeof(struct cramfs_inode));
if (!inode) {
err(FSCK_ERROR, _("malloc failed"));
}
inode_to_host(cramfs_is_big_endian, i, inode); inode_to_host(cramfs_is_big_endian, i, inode);
return inode; return inode;
} }
@ -455,11 +452,8 @@ static void do_directory(char *path, struct cramfs_inode *i)
int pathlen = strlen(path); int pathlen = strlen(path);
int count = i->size; int count = i->size;
unsigned long offset = i->offset << 2; unsigned long offset = i->offset << 2;
char *newpath = malloc(pathlen + 256); char *newpath = xmalloc(pathlen + 256);
if (!newpath) {
err(FSCK_ERROR, _("malloc failed"));
}
if (offset == 0 && count != 0) { if (offset == 0 && count != 0) {
errx(FSCK_UNCORRECTED, _("directory inode has zero offset and non-zero size: %s"), path); errx(FSCK_UNCORRECTED, _("directory inode has zero offset and non-zero size: %s"), path);
} }
@ -693,9 +687,7 @@ int main(int argc, char **argv)
if (argc) if (argc)
progname = argv[0]; progname = argv[0];
outbuffer = malloc(page_size * 2); outbuffer = xmalloc(page_size * 2);
if (!outbuffer)
err(FSCK_ERROR, _("failed to allocate outbuffer"));
/* command line options */ /* command line options */
while ((c = getopt(argc, argv, "hx:v")) != EOF) { while ((c = getopt(argc, argv, "hx:v")) != EOF) {