| 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 |