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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
* vim:expandtab:autoindent:tabstop=4:shiftwidth=4:filetype=c:cindent:textwidth=0:
*
* Copyright (C) 2005 Dell Inc.
* by Michael Brown <Michael_E_Brown@dell.com>
* Licensed under the Open Software License version 2.1
*
* Alternatively, you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published
* by the Free Software Foundation; either version 2 of the License,
* or (at your option) any later version.
* 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.
*/
// compat header should always be first header if including system headers
#define LIBSMBIOS_SOURCE
#include "smbios/compat.h"
#include <stdio.h>
#include <conio.h>
#include "CmosRWImpl.h"
#include "miniddk.h"
using namespace std;
namespace cmos
{
// Non Member Functions
ZwSystemDebugControlPtr ZwSystemDebugControl;
int LoadNtdllFuncs(void)
{
HMODULE hNtdll;
hNtdll = GetModuleHandle(L"ntdll.dll");
if (!hNtdll)
return FALSE;
ZwSystemDebugControl = (ZwSystemDebugControlPtr) GetProcAddress(hNtdll, "ZwSystemDebugControl");
if (!ZwSystemDebugControl)
return FALSE;
return TRUE;
}
int EnableDebug(void)
{
HANDLE hToken;
TOKEN_PRIVILEGES TP;
NTSTATUS status;
status = OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken);
if (!NT_SUCCESS(status))
{
return FALSE;
}
TP.PrivilegeCount = 1;
TP.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
status = LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &TP.Privileges[0].Luid);
if (!NT_SUCCESS(status))
{
return FALSE;
}
status = AdjustTokenPrivileges(hToken, FALSE, &TP, sizeof(TP), NULL, NULL);
if (!NT_SUCCESS(status))
{
return FALSE;
}
return TRUE;
}
//
// CmosRWIo functions
//
// REGULAR CONSTRUCTOR
CmosRWIo::CmosRWIo ()
: ICmosRW(), Suppressable()
{
if (!LoadNtdllFuncs())
throw smbios::NotImplementedImpl("Error loading DLLs needed to call functions to read CMOS.");
if (!EnableDebug())
throw smbios::NotImplementedImpl("Error loading DLLs needed to call functions to read CMOS.");
}
// COPY CONSTRUCTOR
CmosRWIo::CmosRWIo (const CmosRWIo &)
: ICmosRW(), Suppressable()
{}
// OVERLOADED ASSIGNMENT
CmosRWIo & CmosRWIo::operator = (const CmosRWIo &)
{
return *this;
}
// TODO: need to throw exception on problem with iopl
//
u8 CmosRWIo::readByte (u32 indexPort, u32 dataPort, u32 offset) const
{
NTSTATUS status;
u8 Buf;
IO_STRUCT io;
memset(&io, 0, sizeof(io));
io.Addr = indexPort;
io.pBuf = &offset;
io.NumBytes = 1;
io.Reserved4 = 1;
io.Reserved6 = 1;
status = ZwSystemDebugControl(DebugSysWriteIoSpace, &io, sizeof(io), NULL, 0, NULL);
if (!NT_SUCCESS(status))
throw smbios::NotImplementedImpl();
memset(&io, 0, sizeof(io));
io.Addr = dataPort;
io.pBuf = &Buf;
io.NumBytes = 1;
io.Reserved4 = 1;
io.Reserved6 = 1;
status = ZwSystemDebugControl(DebugSysReadIoSpace, &io, sizeof(io), NULL, 0, NULL);
if (!NT_SUCCESS(status))
throw smbios::NotImplementedImpl();
return Buf;
}
// TODO: need to throw exception on problem with iopl
//
void CmosRWIo::writeByte (u32 indexPort, u32 dataPort, u32 offset, u8 byte) const
{
(void) indexPort;
(void) dataPort;
(void) offset;
(void) byte;
throw smbios::NotImplementedImpl();
// enable this whenever we get write support on windows.
#if 0
if(! isNotifySuppressed() )
{
notify();
}
#endif
}
}
|
