common

type AtaDriverHandle interface {
	DriverHandle
	GetIdentity() []byte
	DoTaskFileCmd(rw bool, dma bool, tf *ata.Tf, data []byte, timeoutSecs int) error
}

type DparmError struct {
	Message      string
	DriverStatus byte
	SenseData    *scsi.SENSE_DATA
	// contains filtered or unexported fields
}

type DriveFactory interface {
	OpenByPath(path string) (DriveHandle, error)
	EnumDrives() ([]DriveInfo, error)
	EnumVolumes() (EnumVolumeContext, error)
}

type DriveHandle interface {
	GetDriverHandle() DriverHandle
	Close()
	GetDevicePath() string
	GetDrivingType() DrivingType
	GetDriverName() string

	GetDriveInfo() *DriveInfo

	// ATA
	AtaDoTaskFileCmd(rw bool, dma bool, tf *ata.Tf, data []byte, timeoutSecs int) error

	// NVME
	NvmeGetLogPage(nsid uint32, logId uint32, rae bool, size int) ([]byte, error)

	// COMMON
	SecurityCommand(rw bool, dma bool, protocol uint8, comId uint16, buffer []byte, timeoutSecs int) error

	// TCG
	TcgDiscovery0() error
	GetTcgLevel0InfoAndSerial() (tcg.TcgLevel0Info, string)
}

type DriveHandleImpl struct {
	DriveHandle
	Dh   DriverHandle
	Info DriveInfo
}

type DriveInfo struct {
	DevicePath  string
	DrivingType DrivingType
	DriverName  string

	PartitionStyle   PartitionStyle
	GptDiskId        string // uuid format
	MbrDiskSignature uint32

	Model            string
	Serial           string
	FirmwareRevision string
	RawSerial        [20]byte

	VendorId        string
	ProductRevision string

	WindowsDevNum   int
	SmartEnabled    bool
	AtaIdentity     *ata.IdentityDeviceData
	AtaIdentityRaw  []byte
	NvmeIdentity    *nvme.IdentifyController
	NvmeIdentityRaw []byte

	IsSsd          bool
	SsdCheckWeight int
	TotalCapacity  int64

	TcgSupport int
	tcg.TcgLevel0Info

	Removable  int
	DriveType  DriveType
	Partitions []Partition
}

type DriveType int

type DriverHandle interface {
	GetDriverName() string
	GetDrivingType() DrivingType
	ReopenWritable() error
	Close()

	SecurityCommand(rw bool, dma bool, protocol uint8, comId uint16, buffer []byte, timeoutSecs int) error
}

type DrivingType int

type EnumVolumeContext interface {
	GetList() []VolumeInfo
	FindVolumesByDrive(driveInfo *DriveInfo) []VolumeInfo
	OpenDriveByVolumePath(volumePath string) (DriveHandle, error)
	OpenDriveByPartition(partition Partition) (DriveHandle, error)
}

type GptPartitionInfo struct {
	// PartitionType GUID Upper String Format {...}
	PartitionType string
	// PartitionId GUID Upper String Format {...}
	PartitionId string
}

type MbrPartitionInfo struct {
	PartitionType byte
	BootIndicator bool
}

type NestedError struct {
	// contains filtered or unexported fields
}

type NvmeDriverHandle interface {
	DriverHandle
	GetIdentity() []byte
	DoNvmeAdminPassthru(cmd *nvme.NvmeAdminCmd) error
	NvmeGetLogPage(nsid uint32, logId uint32, rae bool, size int) ([]byte, error)
}

type Partition interface {
	// GetStart bytes
	GetStart() uint64
	// GetEnd bytes
	GetSize() uint64

	GetPartitionStyle() PartitionStyle

	GetMbrInfo() *MbrPartitionInfo
	GetGptInfo() *GptPartitionInfo
}

type PartitionStyle int

type VolumeInfo struct {
	Path        string
	Filesystem  string
	MountPoints []string
	Partitions  []Partition
	DriveType   DriveType
}

type WindowsPhysicalDrive struct {
	DeviceIndex        int
	PhysicalDiskPath   string
	SetupApiDevicePath string
}