logo

utils-extra

Collection of extra tools for Unixes git clone https://anongit.hacktivis.me/git/utils-extra.git/

disk_id.c (2473B)

    

  1. // utils-extra: Collection of extra tools for Unixes
  2. // SPDX-FileCopyrightText: 2017-2022 Haelwenn (lanodan) Monnier <contact+utils@hacktivis.me>
  3. // SPDX-License-Identifier: MPL-2.0
  5. #define _DEFAULT_SOURCE
  6. #include <string.h> // strerror
  7. #include <stdio.h>
  8. #include <errno.h>
  9. #include <stropts.h> // ioctl
  10. #include <fcntl.h> // open
  11. #include <linux/hdreg.h> // HDIO_GET_IDENTITY
  12. #include <unistd.h> // fsync, write
  13. #include <stdint.h> // uint16_t
  14. #include <ctype.h> // isalnum
  16. static char *
  17. strip(char *s, size_t *len)
  18. {
  19. 	while(*len > 0 && *s == ' ')
  20. 	{
  21. 		s++;
  22. 		(*len)--;
  23. 	}
  25. 	while(*len > 0 && s[(*len)-1] == ' ')
  26. 	{
  27. 		s[(*len)-1] = '\0';
  28. 		(*len)--;
  29. 	}
  31. 	return s;
  32. }
  34. static size_t
  35. strncpyalnum(char *dest, char *src, size_t len)
  36. {
  37. 	size_t i = 0;
  39. 	for(; i < len; i++)
  40. 	{
  41. 		if(src[i] == '\0') break;
  43. 		dest[i] = isalnum(src[i]) ? src[i] : '_';
  44. 	}
  46. 	return i;
  47. }
  49. int
  50. main(int argc, char *argv[]) {
  51. 	if(argc != 2)
  52. 	{
  53. 		fprintf(stderr, "disk_id: error: Need 1 argument, got %d\n", argc-1);
  54. 		fprintf(stderr, "Usage: disk_id <block_device_path>\n");
  55. 		return 1;
  56. 	}
  58. 	const char *dev = argv[1];
  60. 	int fd = open(dev, O_RDONLY|O_NONBLOCK|O_LARGEFILE);
  62. 	// IDENTITY is 512 bytes
  63. 	static uint16_t identity[256];
  64. 	int ret = ioctl(fd, HDIO_GET_IDENTITY, identity);
  65. 	if(ret < 0)
  66. 	{
  67. 		switch(errno)
  68. 		{
  69. 		case EINVAL:
  70. 			fprintf(stderr, "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): [EINVAL] Is a partition not a disk\n", dev);
  71. 			return 1;
  72. 		case ENOMSG:
  73. 			fprintf(stderr, "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): [ENOMSG] Information not available\n", dev);
  74. 			return 1;
  75. 		default:
  76. 			fprintf(stderr, "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): %s\n", dev, strerror(errno));
  77. 			return 1;
  78. 		}
  79. 	}
  81. 	// TODO: Handle errno
  82. 	close(fd);
  84. 	static char serno_buf[20] = "";
  85. 	static char model_buf[40] = "";
  86. 	static char disk_id[40+20] = "";
  88. 	size_t serno_len = 20;
  89. 	memcpy(serno_buf, (char *)&identity[10], serno_len);
  90. 	char *serno = strip(serno_buf, &serno_len);
  91. 	//printf("serno: [%s] (%zd)\n", serno, serno_len);
  93. 	size_t model_len = 40;
  94. 	memcpy(model_buf, (char *)&identity[27], model_len);
  95. 	char *model = strip(model_buf, &model_len);
  96. 	//printf("model: [%s] (%zd)\n", model, model_len);
  98. 	size_t modelz = strncpyalnum(disk_id, model, model_len);
  99. 	if(modelz == 0) return 1;
  101. 	disk_id[modelz++] = '_';
  103. 	size_t serialz = strncpyalnum(disk_id+modelz, serno, serno_len);
  104. 	if(serialz == 0) return 1;
  106. 	printf("%s\n", disk_id);
  108. 	return 0;
  109. }