logo

utils-extra

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

disk_id.c (2611B)

    

  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 <ctype.h> // isalnum
  7. #include <errno.h>
  8. #include <fcntl.h>       // open
  9. #include <linux/hdreg.h> // HDIO_GET_IDENTITY
  10. #include <stdint.h>      // uint16_t
  11. #include <stdio.h>
  12. #include <string.h>  // strerror
  13. #include <stropts.h> // ioctl
  14. #include <unistd.h>  // fsync, write
  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. {
  52. 	if(argc != 2)
  53. 	{
  54. 		fprintf(stderr, "disk_id: error: Need 1 argument, got %d\n", argc - 1);
  55. 		fprintf(stderr, "Usage: disk_id <block_device_path>\n");
  56. 		return 1;
  57. 	}
  59. 	const char *dev = argv[1];
  61. 	int fd = open(dev, O_RDONLY | O_NONBLOCK | O_LARGEFILE);
  63. 	// IDENTITY is 512 bytes
  64. 	static uint16_t identity[256];
  65. 	int ret = ioctl(fd, HDIO_GET_IDENTITY, identity);
  66. 	if(ret < 0)
  67. 	{
  68. 		switch(errno)
  69. 		{
  70. 		case EINVAL:
  71. 			fprintf(stderr,
  72. 			        "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): [EINVAL] Is a partition "
  73. 			        "not a disk\n",
  74. 			        dev);
  75. 			return 1;
  76. 		case ENOMSG:
  77. 			fprintf(stderr,
  78. 			        "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): [ENOMSG] Information not "
  79. 			        "available\n",
  80. 			        dev);
  81. 			return 1;
  82. 		default:
  83. 			fprintf(stderr,
  84. 			        "disk_id: error: Device '%s' -> ioctl(HDIO_GET_IDENTITY): %s\n",
  85. 			        dev,
  86. 			        strerror(errno));
  87. 			return 1;
  88. 		}
  89. 	}
  91. 	// TODO: Handle errno
  92. 	close(fd);
  94. 	static char serno_buf[20]    = "";
  95. 	static char model_buf[40]    = "";
  96. 	static char disk_id[40 + 20] = "";
  98. 	size_t serno_len = 20;
  99. 	memcpy(serno_buf, (char *)&identity[10], serno_len);
  100. 	char *serno = strip(serno_buf, &serno_len);
  101. 	//printf("serno: [%s] (%zd)\n", serno, serno_len);
  103. 	size_t model_len = 40;
  104. 	memcpy(model_buf, (char *)&identity[27], model_len);
  105. 	char *model = strip(model_buf, &model_len);
  106. 	//printf("model: [%s] (%zd)\n", model, model_len);
  108. 	size_t modelz = strncpyalnum(disk_id, model, model_len);
  109. 	if(modelz == 0) return 1;
  111. 	disk_id[modelz++] = '_';
  113. 	size_t serialz = strncpyalnum(disk_id + modelz, serno, serno_len);
  114. 	if(serialz == 0) return 1;
  116. 	printf("%s\n", disk_id);
  118. 	return 0;
  119. }