Esempio n. 1
0
        public static ThreadPriority GetCurrentThreadPriority()
        {
            if (PlatformDetails.RunningOnPosix)
            {
                var threadId = Syscall.gettid();
                var value    = Syscall.getpriority((int)Prio.PROCESS, threadId);
                // we assume no failure here, since clearing errno isn't really possible here
                return(FixLinuxPriority(value));
            }

            IntPtr handle = IntPtr.Zero;

            try
            {
                uint threadId = Win32ThreadsMethods.GetCurrentThreadId();
                handle = Win32ThreadsMethods.OpenThread(ThreadAccess.QueryInformation, false, threadId);
                if (handle == IntPtr.Zero)
                {
                    throw new Win32Exception("Failed to setup thread priority, couldn't open the current thread");
                }
                return(Win32ThreadsMethods.GetThreadPriority(handle));
            }
            finally
            {
                if (handle != IntPtr.Zero)
                {
                    Win32ThreadsMethods.CloseHandle(handle);
                }
            }
        }
Esempio n. 2
0
 public ThreadStats()
 {
     ThreadInstance    = Thread.CurrentThread;
     Name              = ThreadInstance.Name;
     Id                = ThreadInstance.ManagedThreadId;
     UnmanagedThreadId = PlatformDetails.RunningOnPosix ?
                         Syscall.gettid() :
                         (int)Win32ThreadsMethods.GetCurrentThreadId();
 }
Esempio n. 3
0
        //method for no seek penalty
        private static RavenDriveType HasNoSeekPenalty(string sDrive)
        {
            var hDrive = CreateFileW(sDrive, 0, // No access to drive
                                     FILE_SHARE_READ | FILE_SHARE_WRITE, IntPtr.Zero, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, IntPtr.Zero);

            DEVICE_SEEK_PENALTY_DESCRIPTOR querySeekPenaltyDesc;
            bool querySeekPenaltyResult;

            try
            {
                if (hDrive == INVALID_HANDLE_VALUE)
                {
                    var message = GetErrorMessage(Marshal.GetLastWin32Error());
                    if (Log.IsInfoEnabled)
                    {
                        Log.Info("CreateFile failed. " + message);
                    }
                    return(RavenDriveType.Unknown);
                }

                var IOCTL_STORAGE_QUERY_PROPERTY = CTL_CODE(IOCTL_STORAGE_BASE, 0x500, METHOD_BUFFERED, FILE_ANY_ACCESS); // From winioctl.h

                var query_seek_penalty = new STORAGE_PROPERTY_QUERY
                {
                    PropertyId = StorageDeviceSeekPenaltyProperty,
                    QueryType  = PropertyStandardQuery
                };

                querySeekPenaltyDesc = new DEVICE_SEEK_PENALTY_DESCRIPTOR();

                uint returnedQuerySeekPenaltySize;

                querySeekPenaltyResult = DeviceIoControl(hDrive, IOCTL_STORAGE_QUERY_PROPERTY, ref query_seek_penalty, (uint)Marshal.SizeOf(query_seek_penalty), ref querySeekPenaltyDesc, (uint)Marshal.SizeOf(querySeekPenaltyDesc), out returnedQuerySeekPenaltySize, IntPtr.Zero);
            }
            finally
            {
                Win32ThreadsMethods.CloseHandle(hDrive);
            }

            if (querySeekPenaltyResult == false)
            {
                var message = GetErrorMessage(Marshal.GetLastWin32Error());
                if (Log.IsInfoEnabled)
                {
                    Log.Info("DeviceIoControl failed. " + message);
                }
                return(RavenDriveType.Unknown);
            }

            return(querySeekPenaltyDesc.IncursSeekPenalty == false ? RavenDriveType.SSD : RavenDriveType.HDD);
        }
Esempio n. 4
0
        public static unsafe void SetCurrentThreadPriority(ThreadPriority priority)
        {
            if (PlatformDetails.RunningOnPosix)
            {
                var nice = FixLinuxPriority(priority);

                var threadId = Syscall.gettid();
                var success  = Syscall.setpriority((int)Prio.PROCESS, threadId, nice);
                if (success != 0)
                {
                    int lastError = Marshal.GetLastWin32Error();
                    if (_log.IsInfoEnabled)
                    {
                        _log.Info($"SetThreadPriority failed to set thread priority. threadId:{threadId}, error code {lastError} - {(Errno)lastError}");
                    }

                    throw new InvalidOperationException("Failed to set priority to thread " + threadId + " with " + (Errno)lastError);
                }
            }
            else
            {
                IntPtr handle = IntPtr.Zero;
                try
                {
                    uint threadId = Win32ThreadsMethods.GetCurrentThreadId();
                    handle = Win32ThreadsMethods.OpenThread(ThreadAccess.SetInformation, false, threadId);
                    if (handle == IntPtr.Zero)
                    {
                        throw new Win32Exception("Failed to setup thread priority, couldn't open the current thread");
                    }

                    var success = Win32ThreadsMethods.SetThreadPriority(handle, (int)priority);
                    if (success == 0)
                    {
                        int lastWin32ErrorCode = Marshal.GetLastWin32Error();

                        if (_log.IsInfoEnabled)
                        {
                            _log.Info($"SetThreadPriority failed to set thread priority. threadId:{threadId}, error code {lastWin32ErrorCode}");
                        }

                        throw new Win32Exception(lastWin32ErrorCode, "Failed to set priority to thread " + threadId);
                    }
                }
                finally
                {
                    Win32ThreadsMethods.CloseHandle(handle);
                }
            }
        }
Esempio n. 5
0
        public static ulong GetCurrentThreadId()
        {
            if (RunningOnPosix == false)
            {
                return(Win32ThreadsMethods.GetCurrentThreadId());
            }

            if (RunningOnLinux)
            {
                return((ulong)Syscall.syscall0(PerPlatformValues.SyscallNumbers.SYS_gettid));
            }

            // OSX
            return(macSyscall.pthread_self());
        }
Esempio n. 6
0
        //method for disk extents
        public static uint?GetPhysicalDriveNumber(string driveLetter)
        {
            var sDrive = "\\\\.\\" + driveLetter + ":";

            var hDrive = CreateFileW(sDrive, 0, // No access to drive
                                     FILE_SHARE_READ | FILE_SHARE_WRITE, IntPtr.Zero, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, IntPtr.Zero);

            VOLUME_DISK_EXTENTS query_disk_extents;
            bool query_disk_extents_result;

            try
            {
                if (hDrive == INVALID_HANDLE_VALUE)
                {
                    var message = GetErrorMessage(Marshal.GetLastWin32Error());
                    if (Log.IsInfoEnabled)
                    {
                        Log.Info("CreateFile failed. " + message);
                    }
                    return(uint.MinValue);
                }

                uint IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS = CTL_CODE(IOCTL_VOLUME_BASE, 0, METHOD_BUFFERED, FILE_ANY_ACCESS); // From winioctl.h

                query_disk_extents = new VOLUME_DISK_EXTENTS();

                uint returned_query_disk_extents_size;

                query_disk_extents_result = DeviceIoControl(hDrive, IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS, IntPtr.Zero, 0, ref query_disk_extents,
                                                            (uint)Marshal.SizeOf(query_disk_extents), out returned_query_disk_extents_size, IntPtr.Zero);
            }
            finally
            {
                Win32ThreadsMethods.CloseHandle(hDrive);
            }

            if (query_disk_extents_result == false || query_disk_extents.Extents.Length != 1)
            {
                var message = GetErrorMessage(Marshal.GetLastWin32Error());
                if (Log.IsInfoEnabled)
                {
                    Log.Info("DeviceIoControl failed. " + message);
                }
                return(null);
            }

            return(query_disk_extents.Extents[0].DiskNumber);
        }
Esempio n. 7
0
        //method for nominal media rotation rate
        //(administrative privilege is required)
        private static RavenDriveType HasNominalMediaRotationRate(string sDrive)
        {
            var hDrive = CreateFileW(sDrive, GENERIC_READ | GENERIC_WRITE, //administrative privilege is required
                                     FILE_SHARE_READ | FILE_SHARE_WRITE, IntPtr.Zero, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, IntPtr.Zero);

            ATAIdentifyDeviceQuery id_query;
            bool result;

            try
            {
                if (hDrive == INVALID_HANDLE_VALUE)
                {
                    var message = GetErrorMessage(Marshal.GetLastWin32Error());
                    if (Log.IsInfoEnabled)
                    {
                        Log.Info("CreateFile failed. " + message);
                    }
                    return(RavenDriveType.Unknown);
                }

                var ioctlAtaPassThrough = CTL_CODE(IOCTL_SCSI_BASE, 0x040b, METHOD_BUFFERED, FILE_READ_ACCESS | FILE_WRITE_ACCESS); // From ntddscsi.h

                id_query      = new ATAIdentifyDeviceQuery();
                id_query.data = new ushort[256];

                id_query.header.Length             = (ushort)Marshal.SizeOf(id_query.header);
                id_query.header.AtaFlags           = (ushort)ATA_FLAGS_DATA_IN;
                id_query.header.DataTransferLength = (uint)(id_query.data.Length * 2); // Size of "data" in bytes
                id_query.header.TimeOutValue       = 3;                                // Sec
#pragma warning disable 618
                id_query.header.DataBufferOffset = Marshal.OffsetOf(typeof(ATAIdentifyDeviceQuery), "data");
#pragma warning restore 618
                id_query.header.PreviousTaskFile   = new byte[8];
                id_query.header.CurrentTaskFile    = new byte[8];
                id_query.header.CurrentTaskFile[6] = 0xec; // ATA IDENTIFY DEVICE

                uint retvalSize;

                result = DeviceIoControl(hDrive, ioctlAtaPassThrough, ref id_query, (uint)Marshal.SizeOf(id_query), ref id_query, (uint)Marshal.SizeOf(id_query),
                                         out retvalSize, IntPtr.Zero);
            }
            finally
            {
                Win32ThreadsMethods.CloseHandle(hDrive);
            }

            if (result == false)
            {
                var message = GetErrorMessage(Marshal.GetLastWin32Error());
                if (Log.IsInfoEnabled)
                {
                    Log.Info("DeviceIoControl failed. " + message);
                }
                return(RavenDriveType.Unknown);
            }

            //word index of nominal media rotation rate
            //(1 means non-rotate device)
            const int kNominalMediaRotRateWordIndex = 217;
            return(id_query.data[kNominalMediaRotRateWordIndex] == 1 ? RavenDriveType.SSD : RavenDriveType.HDD);
        }
Esempio n. 8
0
 public static int GetCurrentThreadId()
 {
     return(RunningOnPosix ?
            Syscall.gettid() :
            (int)Win32ThreadsMethods.GetCurrentThreadId());
 }