mirror of
https://github.com/aaru-dps/aaruremote.git
synced 2025-12-16 19:24:37 +00:00
305 lines
9.9 KiB
C
305 lines
9.9 KiB
C
/*
|
|
* This file is part of the DiscImageChef Remote Server.
|
|
* Copyright (c) 2019 Natalia Portillo.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, version 3.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "../dicmote.h"
|
|
#include "linux.h"
|
|
|
|
#include <ctype.h>
|
|
#include <dirent.h>
|
|
#include <libudev.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
DeviceInfoList* LinuxListDevices()
|
|
{
|
|
DIR* dir;
|
|
struct dirent* dirent;
|
|
struct udev* udev;
|
|
int has_udev = false, i;
|
|
DeviceInfoList * list_start = NULL, *list_current = NULL, *list_next = NULL;
|
|
struct udev_device* udev_device;
|
|
const char* tmp_string;
|
|
FILE* file;
|
|
char* line_str;
|
|
size_t n, ret;
|
|
char* chrptr;
|
|
|
|
udev = udev_new();
|
|
|
|
has_udev = udev != 0;
|
|
|
|
dir = opendir(PATH_SYS_DEVBLOCK);
|
|
if(!dir) return NULL;
|
|
|
|
dirent = readdir(dir);
|
|
|
|
while(dirent)
|
|
{
|
|
if(dirent->d_type != DT_REG && dirent->d_type != DT_LNK)
|
|
{
|
|
dirent = readdir(dir);
|
|
continue;
|
|
}
|
|
|
|
list_next = malloc(sizeof(DeviceInfoList));
|
|
memset(list_next, 0, sizeof(DeviceInfoList));
|
|
|
|
if(!list_next)
|
|
{
|
|
closedir(dir);
|
|
|
|
if(has_udev) udev_unref(udev);
|
|
|
|
return list_start;
|
|
}
|
|
|
|
if(!list_start) list_start = list_next;
|
|
|
|
if(list_current) list_current->next = list_next;
|
|
|
|
snprintf(list_next->this.path, 1024, "/dev/%s", dirent->d_name);
|
|
|
|
if(has_udev)
|
|
{
|
|
udev_device = udev_device_new_from_subsystem_sysname(udev, "block", dirent->d_name);
|
|
if(udev_device)
|
|
{
|
|
tmp_string = udev_device_get_property_value(udev_device, "ID_VENDOR");
|
|
if(tmp_string)
|
|
{
|
|
strncpy(list_next->this.vendor, tmp_string, 256);
|
|
free((void*)tmp_string);
|
|
}
|
|
|
|
tmp_string = udev_device_get_property_value(udev_device, "ID_MODEL");
|
|
if(tmp_string)
|
|
{
|
|
strncpy(list_next->this.model, tmp_string, 256);
|
|
free((void*)tmp_string);
|
|
|
|
for(i = 0; i < 256; i++)
|
|
{
|
|
if(list_next->this.model[i] == 0) break;
|
|
|
|
if(list_next->this.model[i] == '_') list_next->this.model[i] = ' ';
|
|
}
|
|
}
|
|
|
|
tmp_string = udev_device_get_property_value(udev_device, "ID_SCSI_SERIAL");
|
|
if(tmp_string)
|
|
{
|
|
strncpy(list_next->this.serial, tmp_string, 256);
|
|
free((void*)tmp_string);
|
|
}
|
|
else
|
|
{
|
|
tmp_string = udev_device_get_property_value(udev_device, "ID_SERIAL_SHORT");
|
|
if(tmp_string)
|
|
{
|
|
strncpy(list_next->this.serial, tmp_string, 256);
|
|
free((void*)tmp_string);
|
|
}
|
|
}
|
|
|
|
tmp_string = udev_device_get_property_value(udev_device, "ID_BUS");
|
|
if(tmp_string)
|
|
{
|
|
strncpy(list_next->this.bus, tmp_string, 256);
|
|
free((void*)tmp_string);
|
|
}
|
|
}
|
|
}
|
|
|
|
tmp_string = malloc(1024);
|
|
memset((void*)tmp_string, 0, 1024);
|
|
snprintf((char*)tmp_string, 1024, "%s/%s/device/vendor", PATH_SYS_DEVBLOCK, dirent->d_name);
|
|
|
|
if(access(tmp_string, R_OK) == 0 && strlen(list_next->this.vendor) == 0)
|
|
{
|
|
file = fopen(tmp_string, "rb");
|
|
|
|
if(file != NULL)
|
|
{
|
|
line_str = malloc(256);
|
|
memset(line_str, 0, 256);
|
|
n = 256;
|
|
ret = getline(&line_str, &n, file);
|
|
|
|
if(ret > 0 && line_str != NULL)
|
|
{
|
|
strncpy(list_next->this.vendor, line_str, 256);
|
|
for(i = 255; i >= 0; i--)
|
|
{
|
|
if(list_next->this.vendor[i] == 0) continue;
|
|
|
|
else if(list_next->this.vendor[i] == 0x0A || list_next->this.vendor[i] == 0x0D ||
|
|
list_next->this.vendor[i] == ' ')
|
|
list_next->this.vendor[i] = 0;
|
|
else
|
|
break;
|
|
}
|
|
}
|
|
|
|
free(line_str);
|
|
fclose(file);
|
|
}
|
|
}
|
|
else if(strncmp(dirent->d_name, "loop", 4) == 0)
|
|
{
|
|
strncpy(list_next->this.vendor, "Linux", 256);
|
|
}
|
|
free((void*)tmp_string);
|
|
|
|
tmp_string = malloc(1024);
|
|
memset((void*)tmp_string, 0, 1024);
|
|
snprintf((char*)tmp_string, 1024, "%s/%s/device/model", PATH_SYS_DEVBLOCK, dirent->d_name);
|
|
|
|
if(access(tmp_string, R_OK) == 0 &&
|
|
(strlen(list_next->this.model) == 0 || strncmp(list_next->this.bus, "ata", 3) == 0))
|
|
{
|
|
file = fopen(tmp_string, "rb");
|
|
|
|
if(file != NULL)
|
|
{
|
|
line_str = malloc(256);
|
|
memset(line_str, 0, 256);
|
|
n = 256;
|
|
ret = getline(&line_str, &n, file);
|
|
|
|
if(ret > 0 && line_str != NULL)
|
|
{
|
|
strncpy(list_next->this.model, line_str, 256);
|
|
for(i = 255; i >= 0; i--)
|
|
{
|
|
if(list_next->this.model[i] == 0) continue;
|
|
|
|
else if(list_next->this.model[i] == 0x0A || list_next->this.model[i] == 0x0D ||
|
|
list_next->this.model[i] == ' ')
|
|
list_next->this.model[i] = 0;
|
|
else
|
|
break;
|
|
}
|
|
}
|
|
|
|
free(line_str);
|
|
fclose(file);
|
|
}
|
|
}
|
|
else if(strncmp(dirent->d_name, "loop", 4) == 0)
|
|
{
|
|
strncpy(list_next->this.model, "Linux", 256);
|
|
}
|
|
free((void*)tmp_string);
|
|
|
|
tmp_string = malloc(1024);
|
|
memset((void*)tmp_string, 0, 1024);
|
|
snprintf((char*)tmp_string, 1024, "%s/%s/device/serial", PATH_SYS_DEVBLOCK, dirent->d_name);
|
|
|
|
if(access(tmp_string, R_OK) == 0 && (strlen(list_next->this.serial) == 0))
|
|
{
|
|
file = fopen(tmp_string, "rb");
|
|
|
|
if(file != NULL)
|
|
{
|
|
line_str = malloc(256);
|
|
memset(line_str, 0, 256);
|
|
n = 256;
|
|
ret = getline(&line_str, &n, file);
|
|
|
|
if(ret > 0 && line_str != NULL)
|
|
{
|
|
strncpy(list_next->this.serial, line_str, 256);
|
|
for(i = 255; i >= 0; i--)
|
|
{
|
|
if(list_next->this.serial[i] == 0) continue;
|
|
|
|
else if(list_next->this.serial[i] == 0x0A || list_next->this.serial[i] == 0x0D ||
|
|
list_next->this.serial[i] == ' ')
|
|
list_next->this.serial[i] = 0;
|
|
else
|
|
break;
|
|
}
|
|
}
|
|
|
|
free(line_str);
|
|
fclose(file);
|
|
}
|
|
}
|
|
free((void*)tmp_string);
|
|
|
|
if(strlen(list_next->this.vendor) == 0 || strncmp(list_next->this.vendor, "ATA", 3) == 0)
|
|
{
|
|
if(strlen(list_next->this.model) > 0)
|
|
{
|
|
tmp_string = malloc(256);
|
|
strncpy((void*)tmp_string, list_next->this.model, 256);
|
|
|
|
chrptr = strchr(tmp_string, ' ');
|
|
|
|
if(chrptr)
|
|
{
|
|
memset(&list_next->this.vendor, 0, 256);
|
|
memset(&list_next->this.model, 0, 256);
|
|
strncpy(list_next->this.vendor, tmp_string, chrptr - tmp_string);
|
|
strncpy(list_next->this.model, chrptr + 1, 256 - (chrptr - tmp_string) - 1);
|
|
}
|
|
|
|
free((void*)tmp_string);
|
|
}
|
|
}
|
|
|
|
// TODO: Get better device type from sysfs paths
|
|
if(strlen(list_next->this.bus) == 0)
|
|
{
|
|
if(strncmp(dirent->d_name, "loop", 4) == 0) strncpy(list_next->this.bus, "loop", 4);
|
|
else if(strncmp(dirent->d_name, "nvme", 4) == 0)
|
|
strncpy(list_next->this.bus, "NVMe", 4);
|
|
else if(strncmp(dirent->d_name, "mmc", 3) == 0)
|
|
strncpy(list_next->this.bus, "MMC/SD", 6);
|
|
}
|
|
else
|
|
{
|
|
for(i = 0; i < 256; i++)
|
|
{
|
|
if(list_next->this.bus[i] == 0) break;
|
|
|
|
list_next->this.bus[i] = (char)toupper(list_next->this.bus[i]);
|
|
}
|
|
}
|
|
|
|
if(strncmp(list_next->this.bus, "ATA", 3) == 0 || strncmp(list_next->this.bus, "ATAPI", 5) == 0 ||
|
|
strncmp(list_next->this.bus, "SCSI", 4) == 0 || strncmp(list_next->this.bus, "USB", 3) == 0 ||
|
|
strncmp(list_next->this.bus, "PCMCIA", 6) == 0 || strncmp(list_next->this.bus, "FireWire", 8) == 0 ||
|
|
strncmp(list_next->this.bus, "MMC/SD", 6) == 0)
|
|
list_next->this.supported = true;
|
|
else
|
|
list_next->this.supported = false;
|
|
|
|
list_current = list_next;
|
|
dirent = readdir(dir);
|
|
}
|
|
|
|
closedir(dir);
|
|
|
|
if(has_udev) udev_unref(udev);
|
|
|
|
return list_start;
|
|
} |