Page Menu
Home
desp's stash
Search
Configure Global Search
Log In
Files
F225365
DeviceNameResolver.cpp
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Flag For Later
Size
2 KB
Subscribers
None
DeviceNameResolver.cpp
View Options
#include
"DeviceNameResolver.h"
#include
"NativeWinApi.h"
DeviceNameResolver
::
DeviceNameResolver
()
{
NativeWinApi
::
initialize
();
initDeviceNameList
();
}
DeviceNameResolver
::~
DeviceNameResolver
()
{
deviceNameList
.
clear
();
}
void
DeviceNameResolver
::
initDeviceNameList
()
{
TCHAR
shortName
[
3
]
=
{
0
};
TCHAR
longName
[
MAX_PATH
]
=
{
0
};
HardDisk
hardDisk
;
shortName
[
1
]
=
TEXT
(
':'
);
deviceNameList
.
reserve
(
3
);
for
(
TCHAR
shortD
=
TEXT
(
'a'
);
shortD
<
TEXT
(
'z'
);
shortD
++
)
{
shortName
[
0
]
=
shortD
;
if
(
QueryDosDevice
(
shortName
,
longName
,
MAX_PATH
)
>
0
)
{
hardDisk
.
shortName
[
0
]
=
_totupper
(
shortD
);
hardDisk
.
shortName
[
1
]
=
TEXT
(
':'
);
hardDisk
.
shortName
[
2
]
=
0
;
_tcscat_s
(
longName
,
TEXT
(
"
\\
"
));
hardDisk
.
longNameLength
=
_tcslen
(
longName
);
_tcscpy_s
(
hardDisk
.
longName
,
longName
);
deviceNameList
.
push_back
(
hardDisk
);
}
}
fixVirtualDevices
();
}
bool
DeviceNameResolver
::
resolveDeviceLongNameToShort
(
const
TCHAR
*
sourcePath
,
TCHAR
*
targetPath
)
{
for
(
unsigned
int
i
=
0
;
i
<
deviceNameList
.
size
();
i
++
)
{
if
(
!
_tcsnicmp
(
deviceNameList
[
i
].
longName
,
sourcePath
,
deviceNameList
[
i
].
longNameLength
))
{
_tcscpy_s
(
targetPath
,
MAX_PATH
,
deviceNameList
[
i
].
shortName
);
_tcscat_s
(
targetPath
,
MAX_PATH
,
sourcePath
+
deviceNameList
[
i
].
longNameLength
);
return
true
;
}
}
return
false
;
}
void
DeviceNameResolver
::
fixVirtualDevices
()
{
const
USHORT
BufferSize
=
MAX_PATH
*
2
*
sizeof
(
WCHAR
);
WCHAR
longCopy
[
MAX_PATH
]
=
{
0
};
OBJECT_ATTRIBUTES
oa
=
{
0
};
UNICODE_STRING
unicodeInput
=
{
0
};
UNICODE_STRING
unicodeOutput
=
{
0
};
HANDLE
hFile
=
0
;
ULONG
retLen
=
0
;
HardDisk
hardDisk
;
unicodeOutput
.
Buffer
=
(
PWSTR
)
malloc
(
BufferSize
);
if
(
!
unicodeOutput
.
Buffer
)
return
;
for
(
unsigned
int
i
=
0
;
i
<
deviceNameList
.
size
();
i
++
)
{
wcscpy_s
(
longCopy
,
deviceNameList
[
i
].
longName
);
NativeWinApi
::
RtlInitUnicodeString
(
&
unicodeInput
,
longCopy
);
InitializeObjectAttributes
(
&
oa
,
&
unicodeInput
,
0
,
0
,
0
);
if
(
NT_SUCCESS
(
NativeWinApi
::
NtOpenSymbolicLinkObject
(
&
hFile
,
SYMBOLIC_LINK_QUERY
,
&
oa
)))
{
unicodeOutput
.
Length
=
BufferSize
;
unicodeOutput
.
MaximumLength
=
unicodeOutput
.
Length
;
ZeroMemory
(
unicodeOutput
.
Buffer
,
unicodeOutput
.
Length
);
if
(
NT_SUCCESS
(
NativeWinApi
::
NtQuerySymbolicLinkObject
(
hFile
,
&
unicodeOutput
,
&
retLen
)))
{
hardDisk
.
longNameLength
=
wcslen
(
unicodeOutput
.
Buffer
);
wcscpy_s
(
hardDisk
.
shortName
,
deviceNameList
[
i
].
shortName
);
wcscpy_s
(
hardDisk
.
longName
,
unicodeOutput
.
Buffer
);
deviceNameList
.
push_back
(
hardDisk
);
}
NativeWinApi
::
NtClose
(
hFile
);
}
}
free
(
unicodeOutput
.
Buffer
);
}
File Metadata
Details
Attached
Mime Type
text/x-c
Expires
Sun, Mar 16, 9:56 AM (1 d, 18 h)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
7b/d0/62f0c151084008da94fc1437ee86
Attached To
rSCY Scylla
Event Timeline
Log In to Comment