1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103

/// System error category, suitable for all environments.
///
/// On UNIX system it is equivalent of [Posix](struct.PosixCategory.html)
///
/// On Windows it uses winapi error functions
pub struct SystemCategory;

#[cfg(not(windows))]
use crate::posix::to_error;
#[cfg(not(windows))]
pub use crate::posix::{is_would_block, get_unimplemented_error, get_last_error};

#[cfg(windows)]
extern "system" {
    fn GetLastError() -> u32;
    fn FormatMessageW(dwFlags: u32, lpSource: *const u8, dwMessageId: u32, dwLanguageId: u32, lpBuffer: *mut u16, nSize: u32, Arguments: *mut i8) -> u32;
}

#[cfg(windows)]
#[inline]
pub fn get_last_error() -> i32 {
    unsafe {
        GetLastError() as i32
    }
}

#[cfg(windows)]
#[inline]
pub const fn get_unimplemented_error() -> i32 {
    //ERROR_INVALID_FUNCTION
    1
}

#[cfg(windows)]
#[inline]
pub const fn is_would_block(code: i32) -> bool {
    code == 10035 || crate::posix::is_would_block(code)
}

#[cfg(windows)]
#[inline(always)]
pub fn to_error(code: i32) -> crate::Str {
    use core::fmt::Write;

    const FORMAT_MESSAGE_FROM_SYSTEM: u32 = 0x00001000;
    const FORMAT_MESSAGE_IGNORE_INSERTS: u32 = 0x00000200;

    const BUF_SIZE: usize = 256;
    const FMT_FLAGS: u32 = FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_FROM_SYSTEM;
    let mut res = crate::Str::new();
    let mut buff: [u16; BUF_SIZE] = [0; BUF_SIZE];

    let mut num_chars: u32 = unsafe { FormatMessageW(FMT_FLAGS,
                                                     core::ptr::null(), code as u32,
                                                     0, buff.as_mut_ptr(),
                                                     BUF_SIZE as u32, core::ptr::null_mut()) };

    if num_chars == 0 {
        match get_last_error() {
            //Buffer doesn't have enough space
            //But it is completely written so we'll take what we can
            122 => num_chars = BUF_SIZE as u32,
            //System cannot find specified error code
            317 => {
                crate::posix::write_error(code, &mut res);
                return res;
            },
            _ => {
                res.push_str(crate::FAIL_FORMAT);
                return res;
            }
        }
    }

    let mut chars = core::char::decode_utf16(buff[..num_chars as usize].iter().cloned()).map(|r| r.unwrap_or(core::char::REPLACEMENT_CHARACTER));
    while let Some(ch) = chars.next() {
        if ch == '\r' {
            let _ = res.write_str(" ");
            let _ = chars.next();
        } else {
            let _ = res.write_char(ch);
        }

        if res.remaining() == 0 {
            break;
        }
    }

    unsafe {
        res.truncate(res.as_str().trim_end().len() as u8);
    }

    res
}

impl crate::Category for SystemCategory {
    const NAME: &'static str = "OS error";

    #[inline(always)]
    fn message<'a>(code: i32) -> crate::Str {
        to_error(code)
    }
}