windows-server-2003/admin/snapin/wsecmgr/cgroup.cpp

//+---------------------------------------------------------------------------
//
//  Microsoft Windows
//  Copyright (C) Microsoft Corporation 1996-2001.
//
//  File:       cgroup.cpp
//
//  Contents:   implementation of CConfigGroup
//
//----------------------------------------------------------------------------

#include "stdafx.h"
#include "wsecmgr.h"
#include "CGroup.h"
#include "Addgrp.h"
#include "GetUser.h"
#include "wrapper.h"
#include "util.h"
#include "snapmgr.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#define TAB_TITLE_LENGTH	40
/////////////////////////////////////////////////////////////////////////////
// CConfigGroup dialog


CConfigGroup::CConfigGroup(UINT nTemplateID)
: CAttribute (nTemplateID ? nTemplateID : IDD), 
    m_fDirty(false), 
    m_bNoMembers(FALSE), 
    m_bNoMemberOf(FALSE)

{
    //{{AFX_DATA_INIT(CConfigGroup)
        // NOTE: the ClassWizard will add member initialization here
    //}}AFX_DATA_INIT
   m_pHelpIDs = (DWORD_PTR)a107HelpIDs;
   m_uTemplateResID = IDD;
}


void CConfigGroup::DoDataExchange(CDataExchange* pDX)
{
    CAttribute::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CConfigGroup)
    DDX_Control(pDX, IDC_MEMBERS, m_lbMembers);
    DDX_Control(pDX, IDC_MEMBEROF, m_lbMemberOf);
    DDX_Control(pDX, IDC_NO_MEMBERS, m_eNoMembers);
    DDX_Control(pDX, IDC_NO_MEMBER_OF, m_eNoMemberOf);
    //}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CConfigGroup, CAttribute)
    //{{AFX_MSG_MAP(CConfigGroup)
    ON_BN_CLICKED(IDC_ADD_MEMBER, OnAddMember)
    ON_BN_CLICKED(IDC_ADD_MEMBEROF, OnAddMemberof)
    ON_BN_CLICKED(IDC_REMOVE_MEMBER, OnRemoveMember)
    ON_BN_CLICKED(IDC_REMOVE_MEMBEROF, OnRemoveMemberof)
    ON_LBN_SELCHANGE(IDC_MEMBERS, OnSelChangeMem)
    ON_LBN_SELCHANGE(IDC_MEMBEROF, OnSelChangeMemof)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CConfigGroup message handlers

void CConfigGroup::OnAddMember()
{
   CSCEAddGroup gu(this);
   gu.m_sTitle.LoadString( IDS_ADDMEMBER );
   gu.m_sDescription.LoadString(IDS_GROUP_MEMBERS_HEADER);

   gu.m_dwFlags = SCE_SHOW_USERS | SCE_SHOW_GLOBAL | SCE_SHOW_WELLKNOWN;
   gu.SetModeBits(m_pSnapin->GetModeBits());

   PSCE_NAME_LIST pName = 0;

   if( IDD_CONFIG_MEMBERSHIP == m_uTemplateResID ) //Raid #475075, Yang Gao
   {
       gu.m_fCheckName = FALSE;
   }

   CThemeContextActivator activator;
   if (gu.DoModal() == IDOK ) 
   {
      pName = gu.GetUsers();
      PWSTR pLongestName = NULL; //Raid #496574, yanggao
      int maxlength = 0;
      int curlength = 0;
      while(pName) 
      {
         if (LB_ERR == m_lbMembers.FindStringExact(-1,pName->Name)) 
         {
             if (LB_ERR == m_lbMemberOf.FindStringExact(-1,pName->Name)) 
             {
                if (m_bNoMembers) 
                {
                   m_bNoMembers = FALSE;
                   m_lbMembers.ShowWindow(SW_SHOW);
                   m_eNoMembers.ShowWindow(SW_HIDE);
                }
                m_lbMembers.AddString(pName->Name);
                m_fDirty = true;
                
                curlength = lstrlen(pName->Name);
                if( curlength > maxlength )
                {
                   maxlength = curlength;
                   pLongestName = pName->Name;
                }
             } else {
                 // already in the Members Of list, shouldn't add to members
             }
         }
         pName = pName->Next;
      }
      if( pLongestName )
      {
         CClientDC hDC(&m_lbMembers);
         HFONT hFontOld = (HFONT)hDC.SelectObject((HFONT)GetStockObject(DEFAULT_GUI_FONT));
         CSize cSize;
         GetTextExtentPoint32(hDC, pLongestName, lstrlen(pLongestName), &cSize);
         hDC.SelectObject(hFontOld);
         if( cSize.cx > m_lbMembers.GetHorizontalExtent() )
         {
            m_lbMembers.SetHorizontalExtent(cSize.cx+10);
         }
      }
      if( m_fDirty ) //Raid #497350, yanggao, 11/20/2001
      {
         SetModified(TRUE);
      }
      m_lbMembers.EnableWindow(TRUE); //Raid 472244, Yang Gao, 10/22/2001
   }
}

void CConfigGroup::OnAddMemberof()
{
   CSCEAddGroup gu;
   PSCE_NAME_LIST pName = 0;
   DWORD nFlag = 0;

    if ( IsDomainController() ) 
    {
        nFlag = SCE_SHOW_GROUPS | SCE_SHOW_ALIASES;
    } else {
        nFlag = SCE_SHOW_ALIASES | SCE_SHOW_BUILTIN | SCE_SHOW_LOCALGROUPS; //Raid #464603
    }

   gu.m_dwFlags = nFlag;
   gu.SetModeBits(m_pSnapin->GetModeBits());
   gu.m_sTitle.LoadString( IDS_GROUP_MEMBER_OF_HEADER );

   if( IDD_CONFIG_MEMBERSHIP == m_uTemplateResID ) //Raid #475075, Yang Gao
   {
       gu.m_fCheckName = FALSE;
   }

   CThemeContextActivator activator;
   if (gu.DoModal() == IDOK ) 
   {
      pName = gu.GetUsers();
      PWSTR pLongestName = NULL; //Raid #496574, yanggao
      int maxlength = 0;
      int curlength = 0;
      while(pName) 
      {
         if (LB_ERR == m_lbMemberOf.FindStringExact(-1,pName->Name)) 
         {
             if (LB_ERR == m_lbMembers.FindStringExact(-1,pName->Name)) 
             {
                if (m_bNoMemberOf) 
                {
                   m_bNoMemberOf = FALSE;
                   m_lbMemberOf.ShowWindow(SW_SHOW);
                   m_eNoMemberOf.ShowWindow(SW_HIDE);
                }
                m_lbMemberOf.AddString(pName->Name);
                m_fDirty = true;

                curlength = lstrlen(pName->Name);
                if( curlength > maxlength )
                {
                   maxlength = curlength;
                   pLongestName = pName->Name;
                }
             } 
             else 
             {
                 // already find in the members list, shouldn't add it
                 // otherwise, it will be in a circle
             }
         }
         pName = pName->Next;
      }
      if( pLongestName )
      {
         CClientDC hDC(&m_lbMemberOf);
         HFONT hFontOld = (HFONT)hDC.SelectObject((HFONT)GetStockObject(DEFAULT_GUI_FONT));
         CSize cSize;
         GetTextExtentPoint32(hDC, pLongestName, lstrlen(pLongestName), &cSize);
         hDC.SelectObject(hFontOld);
         if( cSize.cx > m_lbMemberOf.GetHorizontalExtent() )
         {
            m_lbMemberOf.SetHorizontalExtent(cSize.cx+10);
         }
      }
      if( m_fDirty ) //Raid #497350, yanggao, 11/20/2001
      {
         SetModified(TRUE);
      }
      m_lbMemberOf.EnableWindow(TRUE); //Raid 472244, Yang Gao, 10/22/2001
   }
}

void CConfigGroup::OnRemoveMember()
{
   int cbItems = m_lbMembers.GetSelCount();
   if (cbItems > 0) 
   {
      int* pnItems = new int [cbItems];
      if ( pnItems ) 
      {
          m_lbMembers.GetSelItems(cbItems,pnItems);

          m_fDirty = true;
          SetModified(TRUE);

          while(cbItems--) 
          {
             m_lbMembers.DeleteString(pnItems[cbItems]);
          }

          delete[] pnItems;
      }
   }
   if (0 == m_lbMembers.GetCount()) 
   {
      m_bNoMembers = TRUE;
      m_lbMembers.ShowWindow(SW_HIDE);
      m_eNoMembers.ShowWindow(SW_SHOW);
      m_eNoMembers.EnableWindow(FALSE); 
   }

   CWnd* pCur = GetDlgItem(IDC_REMOVE_MEMBER); //Raid 472244, Yanggao, 10/23/2001
   if( pCur )
   {
      CWnd* pPrev = pCur->GetWindow(GW_HWNDPREV);
      if( pPrev )
      {
         this->GotoDlgCtrl(pPrev); 
         pCur->EnableWindow(FALSE); //Raid 467886, Yang Gao, 9/18/2001
      }
   }
}

void CConfigGroup::OnRemoveMemberof()
{
   int cbItems = m_lbMemberOf.GetSelCount();
   if (cbItems > 0) 
   {
      int* pnItems = new int [cbItems];
      if ( pnItems ) 
      {
          m_lbMemberOf.GetSelItems(cbItems,pnItems);

          m_fDirty = true;
          SetModified(TRUE);
          while(cbItems--) 
          {
             m_lbMemberOf.DeleteString(pnItems[cbItems]);
          }

          delete[] pnItems;
      }
   }
   if (0 == m_lbMemberOf.GetCount()) 
   {
      m_bNoMemberOf = TRUE;
      m_lbMemberOf.ShowWindow(SW_HIDE);
      m_eNoMemberOf.ShowWindow(SW_SHOW);
      m_eNoMemberOf.EnableWindow(FALSE); 
   }

   CWnd* pCur = GetDlgItem(IDC_REMOVE_MEMBEROF); //Raid 472244, Yanggao, 10/23/2001
   if( pCur )
   {
      CWnd* pPrev = pCur->GetWindow(GW_HWNDPREV);
      if( pPrev )
      {
         this->GotoDlgCtrl(pPrev); 
         pCur->EnableWindow(FALSE); //Raid 467886, Yang Gao, 9/18/2001
      }
   }
}
/*//////////////////////////////////////////////////////////////////////////////
    Method:     OnApply

    Synopsis:   Save all changes to for the group

    Arguments:  None

    Returns:    None

////////////////////////////////////////////////////////////////////////////////*/
BOOL CConfigGroup::OnApply()
{
   if ( !m_bReadOnly )
   {
      PSCE_GROUP_MEMBERSHIP pgm = 0;
      PSCE_NAME_LIST pNames = 0;
      CString strItem;
      int cItems = 0;
      int i = 0;

      if (m_fDirty) 
      {
         //
         // the group pointer is saved in the ID field
         //
         pgm = (PSCE_GROUP_MEMBERSHIP) (m_pData->GetID());

         //
         // should not free the old buffer first because
         // creation of the new buffer may fail
         //
         PSCE_NAME_LIST pTemp1=NULL, pTemp2=NULL;
         DWORD err=ERROR_SUCCESS;

         if (!m_bNoMembers) 
         {
            cItems = m_lbMembers.GetCount();
            for(i=0;i<cItems;i++) 
            {
               pNames = (PSCE_NAME_LIST) LocalAlloc(LPTR,sizeof(SCE_NAME_LIST));

               if ( pNames ) 
               {
                   m_lbMembers.GetText(i,strItem);
                   pNames->Name = (PWSTR) LocalAlloc(LPTR,(strItem.GetLength()+1)*sizeof(TCHAR));

                   if ( pNames->Name ) 
                   {
                       // This may not be a safe usage, using WCHAR instead of TCHAR. Consider fix
                       lstrcpy(pNames->Name,(LPCTSTR)strItem);

                       pNames->Next = pTemp1;
                       pTemp1 = pNames;
                   } 
                   else 
                   {
                       //
                       // no memory
                       //
                       LocalFree(pNames);
                       err = ERROR_NOT_ENOUGH_MEMORY;

                       break;
                   }
               } 
               else 
               {
                   err = ERROR_NOT_ENOUGH_MEMORY;
                   break;
               }
            }
         }

         if ( err == ERROR_SUCCESS ) 
         {
            if (!m_bNoMemberOf) 
            {
               cItems = m_lbMemberOf.GetCount();
               for (i=0;i<cItems;i++) 
               {
                  pNames = (PSCE_NAME_LIST) LocalAlloc(LPTR,sizeof(SCE_NAME_LIST));

                  if ( pNames ) 
                  {
                     m_lbMemberOf.GetText(i,strItem);
                     pNames->Name = (PWSTR) LocalAlloc(LPTR,(strItem.GetLength()+1)*sizeof(TCHAR));

                     if ( pNames->Name ) 
                     {
                        // This may not be a safe usage, using WCHAR instead of TCHAR. Consider fix.
                        lstrcpy(pNames->Name,(LPCTSTR)strItem);

                        pNames->Next = pTemp2;
                        pTemp2 = pNames;
                     } 
                     else 
                     {
                        //
                        // no memory
                        //
                        LocalFree(pNames);
                        err = ERROR_NOT_ENOUGH_MEMORY;

                        break;
                     }
                  } 
                  else 
                  {
                     err = ERROR_NOT_ENOUGH_MEMORY;
                     break;
                  }
               }
            }
         }

         if ( err == ERROR_SUCCESS ) 
         {
             SceFreeMemory(pgm->pMembers,SCE_STRUCT_NAME_LIST);
             pgm->pMembers = pTemp1;

             SceFreeMemory(pgm->pMemberOf,SCE_STRUCT_NAME_LIST);
             pgm->pMemberOf = pTemp2;

             m_pData->Update(m_pSnapin);
             m_fDirty = false;

         } 
         else 
         {
             //
             // error occured, can't save data
             // free the temp buffer and return to the dialog
             //
             SceFreeMemory(pTemp1,SCE_STRUCT_NAME_LIST);
             SceFreeMemory(pTemp2,SCE_STRUCT_NAME_LIST);

             return FALSE;
         }

      }
   }

   return CAttribute::OnApply();
}


BOOL CConfigGroup::OnInitDialog()
{
   CAttribute::OnInitDialog();

   CString str;

   UpdateData(TRUE);

   str.LoadString(IDS_NO_MEMBERS);
   m_eNoMembers.SetWindowText(str);
   str.LoadString(IDS_NO_MEMBER_OF);
   m_eNoMemberOf.SetWindowText(str);

   PSCE_GROUP_MEMBERSHIP pgm = (PSCE_GROUP_MEMBERSHIP) (m_pData->GetID());
   if ( pgm ) 
   {
       SCESTATUS rc;

       if( pgm->GroupName )
       {
          CString strCap; //Raid 467420, Yang Gao
          strCap.Format(IDS_GROUP_TITLE, pgm->GroupName);
          if( strCap.GetLength() > TAB_TITLE_LENGTH ) //Raid #495610, yanggao
          {
             strCap.Format(IDS_GROUP_TITLE_WRAP, pgm->GroupName);
          }

          CPropertySheet* pSheet = (CPropertySheet*)GetParent(); //Raid 469026
          if( pSheet )
          {
             CTabCtrl* pTab = pSheet->GetTabControl();
             if( pTab )
             {
                TC_ITEM ti;
                ti.mask = TCIF_TEXT;
                ti.pszText = strCap.GetBuffer(strCap.GetLength());
                int nPage = pSheet->GetActiveIndex(); 
                pTab->SetItem(nPage, &ti);
             }
          }
       }
       
       PSCE_NAME_LIST pNames = pgm->pMembers;
       if (!pNames) 
       {
          m_bNoMembers = TRUE;
          m_lbMembers.ShowWindow(SW_HIDE);
          m_eNoMembers.ShowWindow(SW_SHOW);
          m_eNoMembers.EnableWindow(FALSE); 
       } 
       else
       {
          PWSTR pLongestName = NULL; //Raid #496574, yanggao
          int maxlength = 0;
          int curlength = 0;
          while(pNames) 
          {
             if( pNames->Name )
             {
                curlength = lstrlen(pNames->Name);
                if( curlength )
                {
                   m_lbMembers.AddString(pNames->Name);
                   if( curlength > maxlength )
                   {
                      maxlength = curlength;
                      pLongestName = pNames->Name;
                   }
                }
             }
             pNames = pNames->Next;
          }
          if( pLongestName )
          {
             CClientDC hDC(&m_lbMembers);
             HFONT hFontOld = (HFONT)hDC.SelectObject((HFONT)GetStockObject(DEFAULT_GUI_FONT));
             CSize cSize;
             GetTextExtentPoint32(hDC, pLongestName, lstrlen(pLongestName), &cSize);
             hDC.SelectObject(hFontOld);
             m_lbMembers.SetHorizontalExtent(cSize.cx+10);
          }
       }
       if( m_lbMembers.GetCount() <= 0 ) //Raid 472244, Yang Gao, 9/18/2001
       {
          m_lbMembers.EnableWindow(false); 
       }
       GetDlgItem (IDC_REMOVE_MEMBER)->EnableWindow(FALSE);
       pNames = pgm->pMemberOf;
       if (!pNames) 
       {
          m_bNoMemberOf = TRUE;
          m_lbMemberOf.ShowWindow(SW_HIDE);
          m_eNoMemberOf.ShowWindow(SW_SHOW);
          m_eNoMemberOf.EnableWindow(FALSE); 
       } 
       else
       {
          PWSTR pLongestName = NULL; //Raid #496574, yanggao
          int maxlength = 0;
          int curlength = 0;
          while(pNames) 
          {
             if( pNames->Name )
             {
                curlength = lstrlen(pNames->Name);
                if( curlength )
                {
                   m_lbMemberOf.AddString(pNames->Name);
                   if( curlength > maxlength )
                   {
                      maxlength = curlength;
                      pLongestName = pNames->Name;
                   }
                }
             }
             pNames = pNames->Next;
          }
          if( pLongestName )
          {
             CClientDC hDC(&m_lbMemberOf);
             HFONT hFontOld = (HFONT)hDC.SelectObject((HFONT)GetStockObject(DEFAULT_GUI_FONT));
             CSize cSize;
             GetTextExtentPoint32(hDC, pLongestName, lstrlen(pLongestName), &cSize);
             hDC.SelectObject(hFontOld);
             m_lbMemberOf.SetHorizontalExtent(cSize.cx+10);
          }
       }
       if( m_lbMemberOf.GetCount() <= 0 ) //Raid 472244, Yang Gao, 10/22/2001
       {
          m_lbMemberOf.EnableWindow(false); 
       }
       GetDlgItem (IDC_REMOVE_MEMBEROF)->EnableWindow(FALSE);
   }

  // m_btnTitle.SetWindowText(m_pData->GetAttrPretty());
   if ( m_bReadOnly )
   {
      GetDlgItem (IDC_ADD_MEMBER)->EnableWindow (FALSE);
      GetDlgItem (IDC_REMOVE_MEMBER)->EnableWindow (FALSE);
      GetDlgItem (IDC_ADD_MEMBEROF)->EnableWindow (FALSE);
      GetDlgItem (IDC_REMOVE_MEMBEROF)->EnableWindow (FALSE);
   }

   return TRUE;  // return TRUE unless you set the focus to a control
                  // EXCEPTION: OCX Property Pages should return FALSE
}


void CConfigGroup::OnSelChangeMem() //Raid 467886, Yang Gao, 9/18/2001
{
   CWnd* pwnd = this->GetDlgItem(IDC_REMOVE_MEMBER);
   if( pwnd )
   {
       if( m_lbMembers.GetSelCount() == 0 ) //Raid #490559, Yanggao. For multi-select listbox
       {
           pwnd->EnableWindow(FALSE);
       }
       else
       {
           pwnd->EnableWindow(TRUE);
       }
   }
}

void CConfigGroup::OnSelChangeMemof()
{
   CWnd* pwnd = this->GetDlgItem(IDC_REMOVE_MEMBEROF);
   if( pwnd )
   {
       if( m_lbMemberOf.GetSelCount() == 0 ) //Raid #450559, Yanggao, For multi-select listbox
       {
           pwnd->EnableWindow(FALSE);
       }
       else
       {
           pwnd->EnableWindow(TRUE);
       }
   }
}