Dr. Dobb's is part of the Informa Tech Division of Informa PLC

This site is operated by a business or businesses owned by Informa PLC and all copyright resides with them. Informa PLC's registered office is 5 Howick Place, London SW1P 1WG. Registered in England and Wales. Number 8860726.


Channels ▼
RSS

Win32 Security in Managed C++


Win32 Security in Managed C++

Listing 6 WinSTL's token_information helper template


/* /////////////////////////////////////////////////////////////
 * ...
 *
 * Extract from winstl_token_information.h
 *
 * Copyright (C) 2003, Synesis Software Pty Ltd.
 * (Licensed under the Synesis Software Standard Source License:
 *  http://www.synesis.com.au/licenses/ssssl.html)
 *
 * ...
 * ////////////////////////////////////////////////////////// */

namespace winstl
{
  template <TOKEN_INFORMATION_CLASS C>
  struct token_information_traits;


  template <>
  struct token_information_traits<TokenUser>
  {
    typedef TOKEN_USER  data_type;
  };

  ... // More specialisations of token_information_traits

  template< TOKEN_INFORMATION_CLASS C
          , ws_typename_param_k     X = stlsoft::null_exception
          , ws_typename_param_k     D = token_information_traits<C>::data_type
          , ws_typename_param_k     A = processheap_allocator<ss_byte_t>
          >
  class token_information
  {
  public:
    typedef token_information<C, X, D, A> class_type;
    typedef token_information_traits<C>   traits_type;
    typedef X                             exception_thrower_type;
    typedef D                             data_type;
    typedef A                             allocator_type;

  // Construction
  public:
    /// Constructs an instance from the given access token
    ws_explicit_k token_information(HANDLE hToken)
      : m_data(0)
    {
      DWORD cbRequired;
      DWORD dwError;

      ::GetTokenInformation(hToken, C, NULL, 0, &cbRequired);
      dwError = ::GetLastError();
      if(ERROR_INSUFFICIENT_BUFFER != dwError)
      {
        // Report error
        exception_thrower_type()(dwError);
      }
      else
      {
        data_type *data = (data_type*)allocator_type().allocate(cbRequired);

        if(NULL == data)
        {
          // Report error
          exception_thrower_type()(ERROR_NOT_ENOUGH_MEMORY);

          // Set the last error, in case the client code is not           // using exception reporting
          ::SetLastError(ERROR_NOT_ENOUGH_MEMORY);
        }
        else
        {
          if(!::GetTokenInformation(hToken, C, data, cbRequired, &cbRequired))
          {
            // Scope the last error, in case the client code is not             // using exception reporting
            dwError = ::GetLastError();

            allocator_type().deallocate((ss_byte_t*)data);

            // Report error
            ::SetLastError(dwError);
            exception_thrower_type()(dwError);
          }
          else
          {
            // Success
            m_data = data;

            ::SetLastError(ERROR_SUCCESS);
          }
        }
      }
    }
    ~token_information()
    {
      allocator_type().deallocate((ss_byte_t*)m_data);
    }

  // Conversion
  public:
    operator data_type *();
    operator data_type const *() const;

    data_type *operator ->();
    data_type const *operator ->() const;

    ws_bool_t operator !() const;

  /// Members
  private:
    data_type *m_data;

  /// Not to be implemented
  private:
    token_information(token_information const &);
    token_information &operator =(token_information const &);
  };

} // namespace winstl


Related Reading


More Insights






Currently we allow the following HTML tags in comments:

Single tags

These tags can be used alone and don't need an ending tag.

<br> Defines a single line break

<hr> Defines a horizontal line

Matching tags

These require an ending tag - e.g. <i>italic text</i>

<a> Defines an anchor

<b> Defines bold text

<big> Defines big text

<blockquote> Defines a long quotation

<caption> Defines a table caption

<cite> Defines a citation

<code> Defines computer code text

<em> Defines emphasized text

<fieldset> Defines a border around elements in a form

<h1> This is heading 1

<h2> This is heading 2

<h3> This is heading 3

<h4> This is heading 4

<h5> This is heading 5

<h6> This is heading 6

<i> Defines italic text

<p> Defines a paragraph

<pre> Defines preformatted text

<q> Defines a short quotation

<samp> Defines sample computer code text

<small> Defines small text

<span> Defines a section in a document

<s> Defines strikethrough text

<strike> Defines strikethrough text

<strong> Defines strong text

<sub> Defines subscripted text

<sup> Defines superscripted text

<u> Defines underlined text

Dr. Dobb's encourages readers to engage in spirited, healthy debate, including taking us to task. However, Dr. Dobb's moderates all comments posted to our site, and reserves the right to modify or remove any content that it determines to be derogatory, offensive, inflammatory, vulgar, irrelevant/off-topic, racist or obvious marketing or spam. Dr. Dobb's further reserves the right to disable the profile of any commenter participating in said activities.

 
Disqus Tips To upload an avatar photo, first complete your Disqus profile. | View the list of supported HTML tags you can use to style comments. | Please read our commenting policy.