OLD | NEW |
1 /** | 1 /** |
2 * \file property.cpp Implementation of Property class etc. | 2 * \file property.cpp Implementation of Property class etc. |
3 */ | 3 */ |
4 | 4 |
| 5 #include "installer-lib.h" |
5 #include "property.h" | 6 #include "property.h" |
6 #include "session.h" | 7 #include "session.h" |
7 #include "msiquery.h" | 8 #include <msiquery.h> |
8 #include <memory> | 9 #include <memory> |
9 | 10 |
10 //------------------------------------------------------------------------------
----------- | 11 //------------------------------------------------------------------------------
----------- |
11 // Property | 12 // Property |
12 //------------------------------------------------------------------------------
----------- | 13 //------------------------------------------------------------------------------
----------- |
13 Property::Property( Session & session, std::wstring name ) | 14 Property::Property( Session & session, std::wstring name ) |
14 // VSE 2012 shows an IntelliSense error here. Ignore it. The compiler properly
sees the 'friend' declaration. | 15 // VSE 2012 shows an IntelliSense error here. Ignore it. The compiler properly
sees the 'friend' declaration. |
15 : handle( session.handle ), name( name ) | 16 : handle( session.handle ), name( name ) |
16 {} | 17 {} |
17 | 18 |
18 /** | 19 /** |
19 * \par Implementation | 20 * \par Implementation |
20 * The center of the implementation is the <a href="http://msdn.microsoft.com/en
-us/library/windows/desktop/aa370134%28v=vs.85%29.aspx">MsiGetProperty function<
/a>. | 21 * The center of the implementation is the <a href="http://msdn.microsoft.com/en
-us/library/windows/desktop/aa370134%28v=vs.85%29.aspx">MsiGetProperty function<
/a>. |
21 */ | 22 */ |
22 Property::operator std::wstring() const | 23 Property::operator std::wstring() const |
23 { | 24 { |
24 /* | 25 /* |
25 * The first call gets the size, but also the actual value if it's short enoug
h. | 26 * The first call gets the size, but also the actual value if it's short enoug
h. |
26 * A second call, if necessary, allocates a sufficiently-long buffer and then
gets the full value. | 27 * A second call, if necessary, allocates a sufficiently-long buffer and then
gets the full value. |
27 * We use only a modest fixed-size buffer for the first step, because we handl
e arbitrary-length property values in a second step. | 28 * We use only a modest fixed-size buffer for the first step, because we handl
e arbitrary-length property values in a second step. |
28 */ | 29 */ |
29 // This buffer allocates on the stack, so we don't want it too large; 64 chara
cters is enough for most properties anyway. | 30 // This buffer allocates on the stack, so we don't want it too large; 64 chara
cters is enough for most properties anyway. |
30 WCHAR buffer1[ 64 ] = { L'\0' } ; | 31 WCHAR buffer1[ 64 ] = { L'\0' } ; |
31 DWORD length = sizeof( buffer1 ) / sizeof( WCHAR ) ; | 32 DWORD length = sizeof( buffer1 ) / sizeof( WCHAR ) ; |
32 switch ( MsiGetPropertyW( handle, name.c_str(), buffer1, & length ) ) | 33 UINT x = MsiGetPropertyW( handle, name.c_str(), buffer1, & length ) ; |
| 34 switch ( x ) |
33 { | 35 { |
34 case ERROR_SUCCESS: | 36 case ERROR_SUCCESS: |
35 // This call might succeed, which means the return value was short enough to
fit into the buffer. | 37 // This call might succeed, which means the return value was short enough to
fit into the buffer. |
36 return std::wstring( buffer1, length ) ; | 38 return std::wstring( buffer1, length ) ; |
37 case ERROR_MORE_DATA: | 39 case ERROR_MORE_DATA: |
38 // Do nothing yet. | 40 // Do nothing yet. |
39 break ; | 41 break ; |
40 default: | 42 default: |
41 throw std::runtime_error( "Error getting property" ) ; | 43 throw windows_api_error( "MsiGetPropertyW", x, "fixed buffer" ) ; |
42 } | 44 } |
43 // Assert we received ERROR_MORE_DATA | 45 // Assert we received ERROR_MORE_DATA |
44 // unique_ptr handles deallocation transparently | 46 // unique_ptr handles deallocation transparently |
45 std::unique_ptr< WCHAR[] > buffer2( new WCHAR[ length ] ); | 47 std::unique_ptr< WCHAR[] > buffer2( new WCHAR[ length ] ) ; |
46 switch ( MsiGetPropertyW( handle, name.c_str(), buffer2.get(), & length ) ) | 48 x = MsiGetPropertyW( handle, name.c_str(), buffer2.get(), & length ) ; |
| 49 switch ( x ) |
47 { | 50 { |
48 case ERROR_SUCCESS: | 51 case ERROR_SUCCESS: |
49 return std::wstring( buffer2.get(), length ) ; | 52 return std::wstring( buffer2.get(), length ) ; |
50 default: | 53 default: |
51 throw std::runtime_error( "Error getting property" ) ; | 54 throw windows_api_error( "MsiGetPropertyW", x, "allocated buffer" ) ; |
52 } | 55 } |
53 } | 56 } |
54 | 57 |
55 /** | 58 /** |
56 * \par Implementation | 59 * \par Implementation |
57 * The center of the implementation is the <a href="http://msdn.microsoft.com/en
-us/library/windows/desktop/aa370391%28v=vs.85%29.aspx">MsiSetProperty function<
/a>. | 60 * The center of the implementation is the <a href="http://msdn.microsoft.com/en
-us/library/windows/desktop/aa370391%28v=vs.85%29.aspx">MsiSetProperty function<
/a>. |
58 */ | 61 */ |
59 void Property::operator=( const std::wstring & value ) | 62 void Property::operator=( const std::wstring & value ) |
60 { | 63 { |
61 if ( MsiSetProperty( handle, name.c_str(), value.c_str() ) != ERROR_SUCCESS ) | 64 UINT x = MsiSetPropertyW( handle, name.c_str(), value.c_str() ) ; |
| 65 if ( x != ERROR_SUCCESS ) |
62 { | 66 { |
63 throw std::runtime_error( "Error setting property" ) ; | 67 throw windows_api_error( "MsiSetPropertyW", x ) ; |
64 } | 68 } |
65 } | 69 } |
OLD | NEW |