summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorPali Rohár <pali.rohar@gmail.com>2014-11-21 15:01:30 +0100
committerPali Rohár <pali.rohar@gmail.com>2014-11-21 15:01:30 +0100
commite00331fb42561669c198cbe3715b1dafdb53bee9 (patch)
tree01de76daaa130539bb176ee21159fa0c71ea9026
parenteb23177014ccfdd96d604f4c6e1cc4bfa5c24ffe (diff)
download0xFFFF-e00331fb42561669c198cbe3715b1dafdb53bee9.tar.bz2
disk: Implement new functions for raw disk access
-rw-r--r--src/disk.c186
-rw-r--r--src/disk.h3
2 files changed, 189 insertions, 0 deletions
diff --git a/src/disk.c b/src/disk.c
index d6390c4..76e2634 100644
--- a/src/disk.c
+++ b/src/disk.c
@@ -35,6 +35,192 @@
static char global_buf[1 << 22]; /* 4MB */
+int disk_open_dev(int maj, int min, int partition, int readonly) {
+
+ int fd;
+ struct stat st;
+ DIR * dir;
+ struct dirent * dirent;
+ int found;
+ size_t len;
+ char blkdev[1024];
+
+ dir = opendir("/dev/");
+ if ( ! dir ) {
+ ERROR_INFO("Cannot open '/dev/' directory");
+ return -1;
+ }
+
+ found = 0;
+
+ while ( ( dirent = readdir(dir) ) ) {
+
+ if ( snprintf(blkdev, sizeof(blkdev), "/dev/%s", dirent->d_name) <= 0 )
+ continue;
+
+ if ( stat(blkdev, &st) != 0 )
+ continue;
+
+ if ( ! S_ISBLK(st.st_mode) )
+ continue;
+
+ if ( makedev(maj, min) != st.st_rdev )
+ continue;
+
+ found = 1;
+ break;
+
+ }
+
+ closedir(dir);
+
+ if ( ! found ) {
+ ERROR("Cannot find block device with id %d:%d", maj, min);
+ return -1;
+ }
+
+ printf("Found block device %s with id %d:%d\n", blkdev, maj, min);
+
+ if ( partition == -1 ) {
+
+ /* Check if block device does not have partitions */
+
+ len = strlen(blkdev);
+ if ( sizeof(blkdev) <= len+2 ) {
+ ERROR("Block device name is too long");
+ return -1;
+ }
+
+ memcpy(blkdev+len, "p1", 3);
+ if ( stat(blkdev, &st) == 0 ) {
+ ERROR("Block device has partitions");
+ return -1;
+ }
+
+ memcpy(blkdev+len, "1", 2);
+ if ( stat(blkdev, &st) == 0 ) {
+ ERROR("Block device has partitions");
+ return -1;
+ }
+
+ blkdev[len] = 0;
+
+ } else if ( partition > 0 ) {
+
+ /* Select partition */
+
+ len = strlen(blkdev);
+ if ( sizeof(blkdev) <= len+2 ) {
+ ERROR("Block device name is too long");
+ return -1;
+ }
+
+ memcpy(blkdev+len, "p1", 3);
+ if ( stat(blkdev, &st) != 0 || ! S_ISBLK(st.st_mode) ) {
+ memcpy(blkdev+len, "1", 2);
+ if ( stat(blkdev, &st) != 0 || ! S_ISBLK(st.st_mode) ) {
+ ERROR("Block device has partitions");
+ return -1;
+ }
+ }
+
+ printf("Found block device %s for partition %d\n", blkdev, partition);
+
+ }
+
+ fd = open(blkdev, (readonly ? O_RDONLY : O_RDWR) | O_EXCL);
+
+ if ( fd < 0 ) {
+ ERROR_INFO("Cannot open block device %s", blkdev);
+ return -1;
+ }
+
+ return fd;
+
+}
+
+int disk_dump_dev(int fd, const char * file) {
+
+ int fd2;
+ int ret;
+ char * path;
+ uint64_t blksize;
+ size_t need, readed;
+ ssize_t size;
+ struct statvfs buf;
+
+ printf("Dump block device to file %s...\n", file);
+
+ if ( ioctl(fd, BLKGETSIZE64, &blksize) != 0 ) {
+ ERROR_INFO("Cannot get size of block device");
+ return -1;
+ }
+
+ if ( blksize > ULLONG_MAX ) {
+ ERROR("Block device is too big");
+ return -1;
+ }
+
+ if ( blksize == 0 ) {
+ ERROR("Block device has zero size");
+ return -1;
+ }
+
+ path = strdup(file);
+ if ( ! path ) {
+ ALLOC_ERROR();
+ return -1;
+ }
+
+ ret = statvfs(dirname(path), &buf);
+
+ free(path);
+
+ if ( ret == 0 && buf.f_bsize * buf.f_bfree < blksize ) {
+ ERROR("Not enough free space (have: %llu, need: %llu)", (unsigned long long int)(buf.f_bsize) * buf.f_bfree, (unsigned long long int)blksize);
+ return -1;
+ }
+
+ fd2 = creat(file, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
+
+ if ( fd2 < 0 ) {
+ ERROR_INFO("Cannot create file %s", file);
+ return -1;
+ }
+
+ readed = 0;
+ printf_progressbar(0, blksize);
+
+ while ( readed < blksize ) {
+ need = blksize - readed;
+ if ( need > sizeof(global_buf) )
+ need = sizeof(global_buf);
+ size = read(fd, global_buf, need);
+ if ( size == 0 )
+ break;
+ if ( write(fd2, global_buf, size) != size ) {
+ PRINTF_ERROR("Dumping image failed");
+ close(fd2);
+ return -1;
+ }
+ readed += size;
+ printf_progressbar(readed, blksize);
+ }
+
+ close(fd2);
+ return 0;
+
+}
+
+int disk_flash_dev(int fd, const char * file) {
+
+ ERROR("Not implemented yet");
+ (void)fd;
+ (void)file;
+ return -1;
+
+}
+
int disk_init(struct usb_device_info * dev) {
(void)dev;
diff --git a/src/disk.h b/src/disk.h
index d67b208..bb61642 100644
--- a/src/disk.h
+++ b/src/disk.h
@@ -29,6 +29,9 @@ enum device disk_get_device(struct usb_device_info * dev);
int disk_flash_raw(const char * blkdev, const char * file);
int disk_dump_raw(const char * blkdev, const char * file);
+int disk_open_dev(int maj, int min, int partition, int readonly);
+int disk_dump_dev(int fd, const char * file);
+int disk_flash_dev(int fd, const char * file);
int disk_flash_image(struct usb_device_info * dev, struct image * image);
int disk_dump_image(struct usb_device_info * dev, enum image_type image, const char * file);