#include <windows.h>
#include <wincrypt.h>
#include <unicode.h>
#include "ui.h"
#include "instres.h"
#include "resource.h"

#include <malloc.h>
#include <assert.h>


//+-------------------------------------------------------------------------
//  Formats multi bytes into WCHAR hex. Includes a space after every 4 bytes.
//
//  Needs (cb * 2 + cb/4 + 1) characters in wsz
//--------------------------------------------------------------------------
static void FormatMsgBoxMultiBytes(DWORD cb, BYTE *pb, LPWSTR wsz)
{
    for (DWORD i = 0; i<cb; i++) {
        int b;
        if (i && 0 == (i & 1))
            *wsz++ = L' ';
        b = (*pb & 0xF0) >> 4;
        *wsz++ = (b <= 9) ? b + L'0' : (b - 10) + L'A';
        b = *pb & 0x0F;
        *wsz++ = (b <= 9) ? b + L'0' : (b - 10) + L'A';
        pb++;
    }
    *wsz++ = 0;
}


INT_PTR CALLBACK MoreInfoDialogProc(
  HWND hwndDlg,  // handle to dialog box
  UINT uMsg,     // message
  WPARAM wParam, // first message parameter
  LPARAM lParam  // second message parameter
) {

    PMIU                                pmiu            = NULL;
    FILETIME                            ftLocal;
    SYSTEMTIME                          stLocal;
    DWORD                               dwChar;
    LPWSTR                              wszName;
    BYTE                                rgbHash[MAX_HASH_LEN];
    DWORD                               cbHash = MAX_HASH_LEN;
    HWND                                hwnd;
    CRYPTUI_VIEWCERTIFICATE_STRUCTW     cryptUI;
    WCHAR                               wsz[128];

    switch(uMsg) {

        case WM_CLOSE:
            EndDialog(hwndDlg, 0);
            return(0);
            break;
     
        case WM_INITDIALOG:

            // remember my imput data
            SetWindowLongPtr(hwndDlg, DWLP_USER, lParam);
            pmiu = (PMIU) lParam;

            // hide the window if we don't have a cryptUI dll
            if(NULL == pmiu->pfnCryptUIDlgViewCertificateW  &&
               NULL != (hwnd = GetDlgItem(hwndDlg, IDC_CAINFO_VIEWCERT)) )
                   ShowWindow(hwnd, SW_HIDE);
 
            // put in the name
	    if(0 != (dwChar=CertNameToStrW(
		X509_ASN_ENCODING,
		&pmiu->pCertContext->pCertInfo->Subject,
        CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
                NULL,
                0
                ) )) {
                
                wszName = (LPWSTR) _alloca(sizeof(WCHAR) * dwChar); 
            
		if(dwChar == CertNameToStrW(
		    X509_ASN_ENCODING,
		    &pmiu->pCertContext->pCertInfo->Subject,
		    CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
		    wszName,
                    dwChar
                    ) ) {
                    
                    SendDlgItemMessageU( 
                        hwndDlg, 
                        IDC_CAINFO_NAME, 
                        WM_SETTEXT, 
                        0, 
                        (LPARAM) wszName);
                }
            }

            wsz[0] = 0;
            FileTimeToLocalFileTime(&pmiu->pCertContext->pCertInfo->NotAfter, &ftLocal);
            FileTimeToSystemTime(&ftLocal, &stLocal);
            GetDateFormatU(LOCALE_USER_DEFAULT, DATE_LONGDATE, &stLocal, NULL, wsz, 128);
            
            // put not after date
            SendDlgItemMessageU( 
                hwndDlg, 
                IDC_CAINFO_EXPIRATION_DATE,
                WM_SETTEXT, 
                0, 
                (LPARAM) wsz);
 
            // get the sha1 thumbprint
            if (CertGetCertificateContextProperty(
                    pmiu->pCertContext,
                    CERT_SHA1_HASH_PROP_ID,
                    rgbHash,
                    &cbHash)) {
                FormatMsgBoxMultiBytes(cbHash, rgbHash, wsz);
                SendDlgItemMessageU( 
                    hwndDlg, 
                    IDC_CAINFO_THUMBPRINT, 
                    WM_SETTEXT, 
                    0, 
                    (LPARAM) wsz);
            }

            // put in the thumbprint alg
            // no localization needed
            SendDlgItemMessageU( 
                hwndDlg, 
                IDC_CAINFO_THUMBPRINT_ALGORITHM, 
                WM_SETTEXT, 
                0, 
                (LPARAM) L"SHA1");

            return(TRUE);
            break;

        case WM_COMMAND:

            switch(HIWORD(wParam)) {

                case BN_CLICKED:

                    switch(LOWORD(wParam)) {
                        case IDOK:
                        case IDCANCEL:
                            EndDialog(hwndDlg, LOWORD(wParam));
                            return(TRUE);
                            break;

                        case IDC_CAINFO_VIEWCERT:

                        GetWindowLongPtr(hwndDlg, DWLP_USER);

                            if(NULL != (pmiu = (PMIU) GetWindowLongPtr(hwndDlg, DWLP_USER))     &&
                               NULL != pmiu->pfnCryptUIDlgViewCertificateW                      ) {
                            
                                memset(&cryptUI, 0, sizeof(CRYPTUI_VIEWCERTIFICATE_STRUCTW));
                                cryptUI.dwSize = sizeof(CRYPTUI_VIEWCERTIFICATE_STRUCTW);
                                cryptUI.pCertContext = pmiu->pCertContext;
                                cryptUI.hwndParent = hwndDlg;
                                cryptUI.dwFlags = 
                                    CRYPTUI_DISABLE_ADDTOSTORE | CRYPTUI_IGNORE_UNTRUSTED_ROOT;
                                pmiu->pfnCryptUIDlgViewCertificateW(&cryptUI, NULL);
                                return(TRUE);
                            }
                            break;
                    }
                    break;
            }
            break;
    }

    return(FALSE);
}

int MoreInfoDlg(
    HWND            hDlgBox,
    int             idLB
) {
    PCCERT_CONTEXT  pCertContext;
    PMDI            pmdi    = (PMDI) GetWindowLongPtr(hDlgBox, DWLP_USER);
    INT_PTR         iItem;
    MIU             miu;

    // What is currently selected
    iItem = SendDlgItemMessageA( 
      hDlgBox,
      idLB, 
      LB_GETCURSEL, 
      0,
      0
      );

    if(iItem == LB_ERR)
      return(LB_ERR);


    // get the pCertContext
    pCertContext = (PCCERT_CONTEXT) SendDlgItemMessageA( 
      hDlgBox,
      idLB, 
      LB_GETITEMDATA, 
      (WPARAM) iItem,
      0
      );

    if(pCertContext == (PCCERT_CONTEXT) LB_ERR  ||  pCertContext == NULL)
      return(LB_ERR);

    // set up the parameters for the more info dialog
    miu.pCertContext                    = pCertContext;
    miu.hInstance                       = pmdi->hInstance;
    miu.pfnCryptUIDlgViewCertificateW   = pmdi->pfnCryptUIDlgViewCertificateW;

    // put the dialog up
    DialogBoxParam(
      pmdi->hInstance,  
      (LPSTR) MAKEINTRESOURCE(IDD_CAINFO),
      hDlgBox,      
      MoreInfoDialogProc,
      (LPARAM) &miu);

    return(0);
}

int AddCertNameToListBox(
    PCCERT_CONTEXT  pCertContext, 
    HWND            hDlgBox,
    int             idLB
) {

    int     itemIndex;
    DWORD   dwChar;
    LPWSTR  wszName;

    if(0 == (dwChar=CertNameToStrW(
	X509_ASN_ENCODING,
	&pCertContext->pCertInfo->Subject,
    CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
	NULL,
        0
        ) ))
        return(LB_ERR);

    wszName = (LPWSTR) _alloca(sizeof(WCHAR) * dwChar); // no error checking, will stack fault, not return NULL
    
    if(dwChar != CertNameToStrW(
	X509_ASN_ENCODING,
	&pCertContext->pCertInfo->Subject,
    CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
	wszName,
        dwChar
        ) )
         return(LB_ERR);

    itemIndex = (int) SendDlgItemMessageU( 
        hDlgBox, 
        idLB, 
        LB_ADDSTRING, 
        0, 
        (LPARAM) wszName) ;

    if(LB_ERR == itemIndex || LB_ERRSPACE == itemIndex)
        return(itemIndex);

    if(LB_ERR ==  SendDlgItemMessageA( 
      hDlgBox,
      idLB, 
      LB_SETITEMDATA, 
      (WPARAM) itemIndex,
      (LPARAM) CertDuplicateCertificateContext(pCertContext)
      ) )
      return(LB_ERR);
      
       
return(0);
}


INT_PTR CALLBACK MainDialogProc(
  HWND hwndDlg,  // handle to dialog box
  UINT uMsg,     // message
  WPARAM wParam, // first message parameter
  LPARAM lParam  // second message parameter
) {

    PMDI            pmdi            = NULL;
    PCCERT_CONTEXT  pCertContext    = NULL;
    WCHAR           wrgDisclaimer[4096];  // because legal stuff is long
    DWORD           dwChar;
    LPWSTR          wszName;

    switch(uMsg) {

        case WM_CLOSE:
            EndDialog(hwndDlg, IDNO);
            return(0);
            break;
     
        case WM_INITDIALOG:

            pmdi = (PMDI) lParam;
            SetWindowLongPtr(hwndDlg, DWLP_USER, lParam);

            // put in the signer name
	    if(0 != (dwChar=CertNameToStrW(
		X509_ASN_ENCODING,
		&pmdi->pCertSigner->pCertInfo->Subject,
        CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
		NULL,
                0
                ) )) {
                
                wszName = (LPWSTR) _alloca(sizeof(WCHAR) * dwChar); 
            
		if(dwChar == CertNameToStrW(
		    X509_ASN_ENCODING,
		    &pmdi->pCertSigner->pCertInfo->Subject,
		    CERT_SIMPLE_NAME_STR | CERT_NAME_STR_REVERSE_FLAG,
		    wszName,
                    dwChar
                    ) ) {
                    
                    SendDlgItemMessageU( 
                        hwndDlg, 
                        IDC_INSTALLCA_VERIFIER, 
                        WM_SETTEXT, 
                        0, 
                        (LPARAM) wszName);
                }
            }

            // set legal disclaimer
            LoadStringU(pmdi->hInstance, IDS_LEGALDISCLAIMER, wrgDisclaimer, sizeof(wrgDisclaimer)/sizeof(WCHAR));
            SendDlgItemMessageU( 
                hwndDlg, 
                IDC_INSTALLCA_LEGALDISCLAIMER, 
                WM_SETTEXT, 
                0, 
                (LPARAM) wrgDisclaimer) ;

            // add each cert to the list box
            while(NULL != (pCertContext = CertEnumCertificatesInStore(pmdi->hStore, pCertContext)))
                AddCertNameToListBox(pCertContext, hwndDlg, IDC_INSTALLCA_CALIST);

            // set the selection to the first item, don't worry about errors
            SendDlgItemMessageU( 
                hwndDlg, 
                IDC_INSTALLCA_CALIST, 
                LB_SETCURSEL, 
                0, 
                0);

            return(TRUE);
            break;

        case WM_COMMAND:

            switch(HIWORD(wParam)) {

                case BN_CLICKED:

                    switch(LOWORD(wParam)) {
                        case IDYES:
                        case IDNO:
                        case IDCANCEL:
                            EndDialog(hwndDlg, LOWORD(wParam));
                            return(TRUE);

                        case IDC_INSTALLCA_MOREINFO:
                            MoreInfoDlg(hwndDlg, IDC_INSTALLCA_CALIST);
                            return(TRUE);
                    }
                    break;

                case LBN_DBLCLK:

                    switch(LOWORD(wParam)) {
                        case IDC_INSTALLCA_CALIST:
                            MoreInfoDlg(hwndDlg, IDC_INSTALLCA_CALIST);
                            return(TRUE);
                    }
                    break;
            }
    }

    return(FALSE);
}


BOOL FIsTooManyCertsOK(DWORD cCerts, HINSTANCE hInstanceUI) {

    WCHAR           wszT[MAX_MSG_LEN];
    WCHAR           wszT1[MAX_MSG_LEN];

    // if too many, ask the user if you wan to continue
    if(cCerts > CACERTWARNINGLEVEL) {
        LoadStringU(hInstanceUI, IDS_INSTALLCA, wszT1, sizeof(wszT1)/sizeof(WCHAR));
        LoadStringU(hInstanceUI, IDS_TOO_MANY_CA_CERTS, wszT, sizeof(wszT)/sizeof(WCHAR));
        return(IDYES == MessageBoxU(NULL, wszT, wszT1, MB_YESNO));
    }
    
    return(TRUE);
}