Add the files

1 files changed, 279 insertions, 0 deletions

diff --git a/src/detection/physicaldisk/physicaldisk_obsd.c b/src/detection/physicaldisk/physicaldisk_obsd.c
new file mode 100644
index 0000000..6e89a99
--- /dev/null
+++ b/src/detection/physicaldisk/physicaldisk_obsd.c
@@ -0,0 +1,279 @@
+#include "physicaldisk.h"
+#include "common/io.h"
+#include "common/sysctl.h"
+#include "common/debug.h"
+
+#include <util.h>
+#include <sys/ioctl.h>
+#include <sys/disklabel.h>
+#include <sys/dkio.h>
+#include <sys/scsiio.h>
+#include <scsi/scsi_all.h>
+#include <fcntl.h>
+#include <errno.h>
+
+static inline const char* retstsToStr(uint8_t retsts) {
+    switch (retsts) {
+        case SCCMD_OK:
+            return "OK";
+        case SCCMD_TIMEOUT:
+            return "TIMEOUT";
+        case SCCMD_BUSY:
+            return "BUSY";
+        case SCCMD_SENSE:
+            return "SENSE";
+        case SCCMD_UNKNOWN:
+            return "UNKNOWN";
+        default:
+            return "?";
+    }
+}
+
+#ifndef NDEBUG
+static inline const char* senseKeyToStr(uint8_t key) {
+    switch (key) {
+        case 0x0:
+            return "No Sense";
+        case 0x1:
+            return "Recovered Error";
+        case 0x2:
+            return "Not Ready";
+        case 0x3:
+            return "Medium Error";
+        case 0x4:
+            return "Hardware Error";
+        case 0x5:
+            return "Illegal Request";
+        case 0x6:
+            return "Unit Attention";
+        case 0x7:
+            return "Data Protect";
+        case 0x8:
+            return "Blank Check";
+        case 0x9:
+            return "Vendor Specific";
+        case 0xA:
+            return "Copy Aborted";
+        case 0xB:
+            return "Aborted Command";
+        case 0xC:
+            return "Equal";
+        case 0xD:
+            return "Volume Overflow";
+        case 0xE:
+            return "Miscompare";
+        case 0xF:
+            return "Completed";
+        default:
+            return "Unknown";
+    }
+}
+
+static void logScsiSense(const char* diskName, const char* operation, const scsireq_t* req) {
+    uint8_t responseCode = req->sense[0] & 0x7F;
+    uint8_t senseKey;
+    uint8_t asc;
+    uint8_t ascq;
+
+    if (responseCode == 0x72 || responseCode == 0x73) {
+        senseKey = req->sense[1] & 0x0F;
+        asc = req->sense[2];
+        ascq = req->sense[3];
+    } else {
+        senseKey = req->sense[2] & 0x0F;
+        asc = req->sense[12];
+        ascq = req->sense[13];
+    }
+
+    FF_STRBUF_AUTO_DESTROY rawSense = ffStrbufCreate();
+    for (size_t i = 0; i < req->senselen_used; ++i) {
+        if (i) {
+            ffStrbufAppendC(&rawSense, ' ');
+        }
+        ffStrbufAppendF(&rawSense, "%02X", req->sense[i]);
+    }
+
+    FF_DEBUG(
+        "%s for %s reported SENSE: response=0x%02X, key=0x%X (%s), ASC=0x%02X, ASCQ=0x%02X, raw=[%s]",
+        operation,
+        diskName,
+        responseCode,
+        senseKey,
+        senseKeyToStr(senseKey),
+        asc,
+        ascq,
+        rawSense.length ? rawSense.chars : "empty");
+}
+#else
+    #define logScsiSense(...) ((void) 0)
+#endif
+
+const char* ffDetectPhysicalDisk(FFlist* result, FFPhysicalDiskOptions* options) {
+    FF_STRBUF_AUTO_DESTROY diskNames = ffStrbufCreate();
+    FF_DEBUG("Querying disk names via sysctl hw.disknames");
+    const char* error = ffSysctlGetString(CTL_HW, HW_DISKNAMES, &diskNames);
+    if (error) {
+        FF_DEBUG("ffSysctlGetString(HW_DISKNAMES) failed: %s", error);
+        return error;
+    }
+
+    FF_DEBUG("Disk names: %s", diskNames.chars);
+
+    char* diskName = NULL;
+    size_t len = 0;
+    while (ffStrbufGetdelim(&diskName, &len, ',', &diskNames)) {
+        char* colon = strchr(diskName, ':');
+        if (colon) {
+            *colon = '\0';
+        }
+
+        FF_DEBUG("Probing disk: %s", diskName);
+
+        if (options->namePrefix.length && !ffStrbufStartsWith(&(FFstrbuf) {
+                                                                  .chars = diskName,
+                                                                  .length = (uint32_t) len,
+                                                              },
+                                              &options->namePrefix)) {
+            FF_DEBUG("Skipping %s due to namePrefix filter", diskName);
+            continue;
+        }
+
+        char* devPath = NULL;
+        FF_AUTO_CLOSE_FD int f = opendev(diskName, O_RDONLY, OPENDEV_PART, &devPath);
+        if (f < 0) {
+            if (errno == EACCES) {
+                FF_DEBUG("opendev(%s) failed: permission denied", diskName);
+                return "Permission denied; root required";
+            } else {
+                FF_DEBUG("opendev(%s) failed: %s", diskName, strerror(errno));
+                continue; // Unknown error
+            }
+        }
+
+        FF_DEBUG("Opened %s as %s", diskName, devPath);
+
+        struct disklabel dl;
+        if (ioctl(f, DIOCGPDINFO, &dl) < 0) {
+            FF_DEBUG("ioctl(DIOCGPDINFO) failed for %s: %s", diskName, strerror(errno));
+            continue;
+        }
+
+        FFPhysicalDiskType type = FF_PHYSICALDISK_TYPE_NONE;
+        if (dl.d_type == DTYPE_VND || dl.d_type == DTYPE_RDROOT) {
+            if (options->hideType & FF_PHYSICALDISK_TYPE_VIRTUAL) {
+                FF_DEBUG("Skipping virtual disk %s due to hideType", diskName);
+                continue;
+            }
+
+            type |= FF_PHYSICALDISK_TYPE_VIRTUAL;
+        }
+
+        uint64_t size = DL_GETDSIZE(&dl) * dl.d_secsize;
+        if (size == 0) {
+            if (options->hideType & FF_PHYSICALDISK_TYPE_UNUSED) {
+                FF_DEBUG("Skipping unused disk %s due to hideType", diskName);
+                continue;
+            }
+
+            type |= FF_PHYSICALDISK_TYPE_UNUSED;
+        }
+
+        FFPhysicalDiskResult* device = FF_LIST_ADD(FFPhysicalDiskResult, *result);
+        ffStrbufInitS(&device->name, dl.d_packname);
+        ffStrbufInitS(&device->devPath, devPath);
+        ffStrbufInit(&device->serial);
+        ffStrbufInit(&device->revision);
+        ffStrbufInitS(&device->interconnect, dl.d_typename);
+        device->type = type;
+        device->size = size;
+        device->temperature = FF_PHYSICALDISK_TEMP_UNSET;
+        FF_DEBUG("Added disk entry: name='%s', devPath='%s', interconnect='%s', size=%llu",
+            device->name.chars,
+            device->devPath.chars,
+            device->interconnect.chars,
+            (unsigned long long) device->size);
+
+        struct scsi_inquiry_data inquiry = {};
+        struct scsireq req = {
+            .cmd = { [0] = INQUIRY, [4] = sizeof(inquiry) },
+            .cmdlen = 6,
+            .databuf = (caddr_t) &inquiry,
+            .datalen = sizeof(inquiry),
+            .timeout = 1000,
+            .flags = SCCMD_READ,
+        };
+        if (ioctl(f, SCIOCCOMMAND, &req) == 0) {
+            if (req.retsts == SCCMD_OK) {
+                ffStrbufClear(&device->name);
+                ffStrbufAppendNS(&device->name, (uint32_t) ARRAY_SIZE(inquiry.vendor), inquiry.vendor);
+                ffStrbufTrimRight(&device->name, '\0');
+                ffStrbufTrimRight(&device->name, ' ');
+                ffStrbufAppendC(&device->name, ' ');
+                ffStrbufAppendNS(&device->name, (uint32_t) ARRAY_SIZE(inquiry.product), inquiry.product);
+                ffStrbufTrimRight(&device->name, '\0');
+                ffStrbufTrimRight(&device->name, ' ');
+
+                ffStrbufSetNS(&device->revision, (uint32_t) ARRAY_SIZE(inquiry.revision), inquiry.revision);
+                ffStrbufTrimRight(&device->revision, '\0');
+                ffStrbufTrimRight(&device->revision, ' ');
+                device->type |= inquiry.dev_qual2 & SID_REMOVABLE ? FF_PHYSICALDISK_TYPE_REMOVABLE : FF_PHYSICALDISK_TYPE_FIXED;
+                FF_DEBUG("SCSI inquiry for %s: name='%s', revision='%s'", diskName, device->name.chars, device->revision.chars);
+            } else {
+                FF_DEBUG("SCSI inquiry retsts != SCCMD_OK for %s (%d)", diskName, req.retsts);
+            }
+        } else {
+            FF_DEBUG("ioctl(SCIOCCOMMAND) failed for SCSI inquiry on %s: %s", diskName, strerror(errno));
+        }
+
+        struct scsi_vpd_serial evpd = {};
+        req = (struct scsireq) {
+            .cmd = { [0] = INQUIRY, [1] = SI_EVPD, [2] = SI_PG_SERIAL, [4] = sizeof(evpd) },
+            .cmdlen = 6,
+            .databuf = (caddr_t) &evpd,
+            .datalen = sizeof(evpd),
+            .timeout = 1000,
+            .flags = SCCMD_READ,
+        };
+        if (ioctl(f, SCIOCCOMMAND, &req) == 0) {
+            if (req.retsts == SCCMD_OK && evpd.hdr.page_code == SI_PG_SERIAL) {
+                for (uint8_t i = 0; i < evpd.hdr.page_length[1]; ++i) {
+                    ffStrbufAppendF(&device->serial, "%02X", evpd.serial[i]);
+                }
+                ffStrbufTrimSpace(&device->serial);
+                FF_DEBUG("SCSI serial for %s: %s", diskName, device->serial.chars);
+            } else if (req.retsts == SCCMD_SENSE) {
+                logScsiSense(diskName, "SCSI serial page", &req);
+            } else {
+                FF_DEBUG("SCSI serial page unavailable for %s (retsts=%s, pagecode=%u)", diskName, retstsToStr(req.retsts), (unsigned) evpd.hdr.page_code);
+            }
+        } else {
+            FF_DEBUG("ioctl(SCIOCCOMMAND) failed for SCSI serial on %s: %s", diskName, strerror(errno));
+        }
+
+        struct scsi_mode_header mode = {};
+        req = (scsireq_t) {
+            .cmd = { [0] = MODE_SENSE, [2] = SMS_PAGE_CODE, [4] = sizeof(mode) },
+            .cmdlen = 6,
+            .databuf = (caddr_t) &mode,
+            .datalen = sizeof(mode),
+            .timeout = 1000,
+            .flags = SCCMD_READ,
+        };
+        if (ioctl(f, SCIOCCOMMAND, &req) == 0) {
+            if (req.retsts == SCCMD_OK) {
+                device->type |= mode.dev_spec & 0x80 ? FF_PHYSICALDISK_TYPE_READONLY : FF_PHYSICALDISK_TYPE_READWRITE;
+                FF_DEBUG("SCSI mode for %s indicates: %s", diskName, mode.dev_spec & 0x80 ? "readonly" : "readwrite");
+            } else if (req.retsts == SCCMD_SENSE) {
+                logScsiSense(diskName, "SCSI mode sense", &req);
+            } else {
+                FF_DEBUG("SCSI mode sense unavailable for %s (retsts=%s, data_length=%u)", diskName, retstsToStr(req.retsts), (unsigned) mode.data_length);
+            }
+        } else {
+            FF_DEBUG("ioctl(SCIOCCOMMAND) failed for SCSI mode sense on %s: %s", diskName, strerror(errno));
+        }
+
+        FF_DEBUG("Detected disk '%s' (%s), type=%u", device->name.chars, diskName, (unsigned) device->type);
+    }
+
+    return NULL;
+}