extras/volume_id/lib/util.c

   1 /*
   2  * volume_id - reads filesystem label and uuid
   3  *
   4  * Copyright (C) 2005 Kay Sievers <kay.sievers@vrfy.org>
   5  *
   6  *      This program is free software; you can redistribute it and/or modify it
   7  *      under the terms of the GNU General Public License as published by the
   8  *      Free Software Foundation version 2 of the License.
   9  */
  10
  11 #ifndef _GNU_SOURCE
  12 #define _GNU_SOURCE 1
  13 #endif
  14
  15 #ifdef HAVE_CONFIG_H
  16 #  include <config.h>
  17 #endif
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <string.h>
  23 #include <errno.h>
  24 #include <ctype.h>
  25 #include <fcntl.h>
  26 #include <sys/stat.h>
  27
  28 #include "libvolume_id.h"
  29 #include "util.h"
  30
  31 void volume_id_set_unicode16(char *str, size_t len, const uint8_t *buf, enum endian endianess, size_t count)
  32 {
  33         unsigned int i, j;
  34         uint16_t c;
  35
  36         j = 0;
  37         for (i = 0; i + 2 <= count; i += 2) {
  38                 if (endianess == LE)
  39                         c = (buf[i+1] << 8) | buf[i];
  40                 else
  41                         c = (buf[i] << 8) | buf[i+1];
  42                 if (c == 0) {
  43                         str[j] = '\0';
  44                         break;
  45                 } else if (c < 0x80) {
  46                         if (j+1 >= len)
  47                                 break;
  48                         str[j++] = (uint8_t) c;
  49                 } else if (c < 0x800) {
  50                         if (j+2 >= len)
  51                                 break;
  52                         str[j++] = (uint8_t) (0xc0 | (c >> 6));
  53                         str[j++] = (uint8_t) (0x80 | (c & 0x3f));
  54                 } else {
  55                         if (j+3 >= len)
  56                                 break;
  57                         str[j++] = (uint8_t) (0xe0 | (c >> 12));
  58                         str[j++] = (uint8_t) (0x80 | ((c >> 6) & 0x3f));
  59                         str[j++] = (uint8_t) (0x80 | (c & 0x3f));
  60                 }
  61         }
  62         str[j] = '\0';
  63 }
  64
  65 static char *usage_to_string(enum volume_id_usage usage_id)
  66 {
  67         switch (usage_id) {
  68         case VOLUME_ID_FILESYSTEM:
  69                 return "filesystem";
  70         case VOLUME_ID_OTHER:
  71                 return "other";
  72         case VOLUME_ID_RAID:
  73                 return "raid";
  74         case VOLUME_ID_DISKLABEL:
  75                 return "disklabel";
  76         case VOLUME_ID_CRYPTO:
  77                 return "crypto";
  78         case VOLUME_ID_UNPROBED:
  79                 return "unprobed";
  80         case VOLUME_ID_UNUSED:
  81                 return "unused";
  82         }
  83         return NULL;
  84 }
  85
  86 void volume_id_set_usage(struct volume_id *id, enum volume_id_usage usage_id)
  87 {
  88         id->usage_id = usage_id;
  89         id->usage = usage_to_string(usage_id);
  90 }
  91
  92 void volume_id_set_label_raw(struct volume_id *id, const uint8_t *buf, size_t count)
  93 {
  94         memcpy(id->label_raw, buf, count);
  95         id->label_raw_len = count;
  96 }
  97
  98 void volume_id_set_label_string(struct volume_id *id, const uint8_t *buf, size_t count)
  99 {
 100         unsigned int i;
 101
 102         memcpy(id->label, buf, count);
 103
 104         /* remove trailing whitespace */
 105         i = strnlen(id->label, count);
 106         while (i--) {
 107                 if (!isspace(id->label[i]))
 108                         break;
 109         }
 110         id->label[i+1] = '\0';
 111 }
 112
 113 void volume_id_set_label_unicode16(struct volume_id *id, const uint8_t *buf, enum endian endianess, size_t count)
 114 {
 115          volume_id_set_unicode16(id->label, sizeof(id->label), buf, endianess, count);
 116 }
 117
 118 void volume_id_set_uuid(struct volume_id *id, const uint8_t *buf, enum uuid_format format)
 119 {
 120         unsigned int i;
 121         unsigned int count = 0;
 122
 123         switch(format) {
 124         case UUID_DOS:
 125                 count = 4;
 126                 break;
 127         case UUID_NTFS:
 128         case UUID_HFS:
 129                 count = 8;
 130                 break;
 131         case UUID_DCE:
 132                 count = 16;
 133                 break;
 134         case UUID_DCE_STRING:
 135                 count = 36;
 136                 break;
 137         case UUID_FOURINT:
 138                 count = 35;
 139                 break;
 140         }
 141         memcpy(id->uuid_raw, buf, count);
 142         id->uuid_raw_len = count;
 143
 144         /* if set, create string in the same format, the native platform uses */
 145         for (i = 0; i < count; i++)
 146                 if (buf[i] != 0)
 147                         goto set;
 148         return;
 149
 150 set:
 151         switch(format) {
 152         case UUID_DOS:
 153                 sprintf(id->uuid, "%02X%02X-%02X%02X",
 154                         buf[3], buf[2], buf[1], buf[0]);
 155                 break;
 156         case UUID_NTFS:
 157                 sprintf(id->uuid,"%02X%02X%02X%02X%02X%02X%02X%02X",
 158                         buf[7], buf[6], buf[5], buf[4],
 159                         buf[3], buf[2], buf[1], buf[0]);
 160                 break;
 161         case UUID_HFS:
 162                 sprintf(id->uuid,"%02X%02X%02X%02X%02X%02X%02X%02X",
 163                         buf[0], buf[1], buf[2], buf[3],
 164                         buf[4], buf[5], buf[6], buf[7]);
 165                 break;
 166         case UUID_DCE:
 167                 sprintf(id->uuid,
 168                         "%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x",
 169                         buf[0], buf[1], buf[2], buf[3],
 170                         buf[4], buf[5],
 171                         buf[6], buf[7],
 172                         buf[8], buf[9],
 173                         buf[10], buf[11], buf[12], buf[13], buf[14],buf[15]);
 174                 break;
 175         case UUID_DCE_STRING:
 176                 memcpy(id->uuid, buf, count);
 177                 id->uuid[count] = '\0';
 178                 break;
 179         case UUID_FOURINT:
 180                 sprintf(id->uuid,
 181                         "%02x%02x%02x%02x:%02x%02x%02x%02x:%02x%02x%02x%02x:%02x%02x%02x%02x",
 182                         buf[0], buf[1], buf[2], buf[3],
 183                         buf[4], buf[5], buf[6], buf[7],
 184                         buf[8], buf[9], buf[10], buf[11],
 185                         buf[12], buf[13], buf[14],buf[15]);
 186                 break;
 187         }
 188 }
 189
 190 uint8_t *volume_id_get_buffer(struct volume_id *id, uint64_t off, size_t len)
 191 {
 192         ssize_t buf_len;
 193
 194         info("get buffer off 0x%llx(%llu), len 0x%zx", (unsigned long long) off, (unsigned long long) off, len);
 195         /* check if requested area fits in superblock buffer */
 196         if (off + len <= SB_BUFFER_SIZE) {
 197                 if (id->sbbuf == NULL) {
 198                         id->sbbuf = malloc(SB_BUFFER_SIZE);
 199                         if (id->sbbuf == NULL) {
 200                                 dbg("error malloc");
 201                                 return NULL;
 202                         }
 203                 }
 204
 205                 /* check if we need to read */
 206                 if ((off + len) > id->sbbuf_len) {
 207                         info("read sbbuf len:0x%llx", (unsigned long long) (off + len));
 208                         if (lseek(id->fd, 0, SEEK_SET) < 0) {
 209                                 dbg("lseek failed (%s)", strerror(errno));
 210                                 return NULL;
 211                         }
 212                         buf_len = read(id->fd, id->sbbuf, off + len);
 213                         if (buf_len < 0) {
 214                                 dbg("read failed (%s)", strerror(errno));
 215                                 return NULL;
 216                         }
 217                         dbg("got 0x%zx (%zi) bytes", buf_len, buf_len);
 218                         id->sbbuf_len = buf_len;
 219                         if ((size_t)buf_len < off + len) {
 220                                 dbg("requested 0x%zx bytes, got only 0x%zx bytes", len, buf_len);
 221                                 return NULL;
 222                         }
 223                 }
 224
 225                 return &(id->sbbuf[off]);
 226         } else {
 227                 if (len > SEEK_BUFFER_SIZE) {
 228                         dbg("seek buffer too small %d", SEEK_BUFFER_SIZE);
 229                         return NULL;
 230                 }
 231
 232                 /* get seek buffer */
 233                 if (id->seekbuf == NULL) {
 234                         id->seekbuf = malloc(SEEK_BUFFER_SIZE);
 235                         if (id->seekbuf == NULL) {
 236                                 dbg("error malloc");
 237                                 return NULL;
 238                         }
 239                 }
 240
 241                 /* check if we need to read */
 242                 if ((off < id->seekbuf_off) || ((off + len) > (id->seekbuf_off + id->seekbuf_len))) {
 243                         info("read seekbuf off:0x%llx len:0x%zx", (unsigned long long) off, len);
 244                         if (lseek(id->fd, off, SEEK_SET) < 0) {
 245                                 dbg("lseek failed (%s)", strerror(errno));
 246                                 return NULL;
 247                         }
 248                         buf_len = read(id->fd, id->seekbuf, len);
 249                         if (buf_len < 0) {
 250                                 dbg("read failed (%s)", strerror(errno));
 251                                 return NULL;
 252                         }
 253                         dbg("got 0x%zx (%zi) bytes", buf_len, buf_len);
 254                         id->seekbuf_off = off;
 255                         id->seekbuf_len = buf_len;
 256                         if ((size_t)buf_len < len) {
 257                                 dbg("requested 0x%zx bytes, got only 0x%zx bytes", len, buf_len);
 258                                 return NULL;
 259                         }
 260                 }
 261
 262                 return &(id->seekbuf[off - id->seekbuf_off]);
 263         }
 264 }
 265
 266 void volume_id_free_buffer(struct volume_id *id)
 267 {
 268         if (id->sbbuf != NULL) {
 269                 free(id->sbbuf);
 270                 id->sbbuf = NULL;
 271                 id->sbbuf_len = 0;
 272         }
 273         if (id->seekbuf != NULL) {
 274                 free(id->seekbuf);
 275                 id->seekbuf = NULL;
 276                 id->seekbuf_len = 0;
 277         }
 278 }