259 lines
8.7 KiB
C++
259 lines
8.7 KiB
C++
#include "item.h"
|
|
|
|
#ifndef CERT_NAME_STR_ENABLE_UTF8_UNICODE_FLAG
|
|
#define CERT_NAME_STR_ENABLE_UTF8_UNICODE_FLAG 0
|
|
#endif
|
|
|
|
BOOL FormatNames(DWORD cNames, PCERT_NAME_BLOB rgNames, HWND hwnd, DWORD idc)
|
|
{
|
|
DWORD cb;
|
|
BOOL f;
|
|
DWORD i;
|
|
DWORD i1;
|
|
DWORD i2;
|
|
PCERT_ALT_NAME_INFO pname = NULL;
|
|
WCHAR rgwch[4096];
|
|
|
|
rgwch[0] = 0;
|
|
|
|
for (i1=0; i1<cNames; i1++) {
|
|
f = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_ALTERNATE_NAME,
|
|
rgNames[i1].pbData, rgNames[i1].cbData,
|
|
CRYPT_ENCODE_ALLOC_FLAG, NULL, &pname, &cb);
|
|
if (!f) break;
|
|
|
|
for (i2=0; i2<pname->cAltEntry; i2++) {
|
|
if (i2 > 0) {
|
|
wcscat(rgwch, L" ");
|
|
}
|
|
switch( pname->rgAltEntry[i2].dwAltNameChoice ) {
|
|
case CERT_ALT_NAME_RFC822_NAME:
|
|
wcscat(rgwch, L"SMTP: ");
|
|
wcscat(rgwch, pname->rgAltEntry[i2].pwszRfc822Name);
|
|
wcscat(rgwch, L"\r\n");
|
|
break;
|
|
|
|
case CERT_ALT_NAME_DIRECTORY_NAME:
|
|
break;
|
|
}
|
|
}
|
|
free(pname);
|
|
pname = NULL;
|
|
}
|
|
|
|
if (f) SetDlgItemTextW(hwnd, idc, rgwch);
|
|
return f;
|
|
}
|
|
|
|
BOOL ParseNames(DWORD * pcNames, PCERT_NAME_BLOB * prgNames, HWND hwnd, DWORD idc)
|
|
{
|
|
DWORD cb;
|
|
DWORD cEntry = 0;
|
|
DWORD cNames = 0;
|
|
BOOL f;
|
|
DWORD i;
|
|
LPWSTR pwsz;
|
|
LPWSTR pwsz1;
|
|
CRYPT_DER_BLOB rgDer[50] = {0};
|
|
CERT_ALT_NAME_INFO rgNames[50] = {0};
|
|
CERT_ALT_NAME_ENTRY rgEntry[200] = {0};
|
|
WCHAR rgwch[4096];
|
|
|
|
GetDlgItemTextW(hwnd, idc, rgwch, sizeof(rgwch)/sizeof(WCHAR));
|
|
rgwch[4095] = 0;
|
|
|
|
pwsz = rgwch;
|
|
|
|
while (*pwsz != 0) {
|
|
if (cEntry == 200) {
|
|
MessageBox(hwnd, "Can't have more than 200 entries", "smimetst", MB_OK);
|
|
return FALSE;
|
|
}
|
|
|
|
if (*pwsz == ' ') {
|
|
while (*pwsz == ' ') pwsz++;
|
|
rgNames[cNames-1].cAltEntry += 1;
|
|
}
|
|
else {
|
|
if (cNames == 50) {
|
|
MessageBox(hwnd, "Can't have more than 50 names", "smimetst", MB_OK);
|
|
return FALSE;
|
|
}
|
|
cNames += 1;
|
|
rgNames[cNames-1].rgAltEntry = &rgEntry[cEntry];
|
|
rgNames[cNames-1].cAltEntry = 1;
|
|
}
|
|
|
|
if (_wcsnicmp(pwsz, L"SMTP:", 5) == 0) {
|
|
pwsz += 5;
|
|
while (*pwsz == ' ') pwsz++;
|
|
rgEntry[cEntry].dwAltNameChoice = CERT_ALT_NAME_RFC822_NAME;
|
|
rgEntry[cEntry].pwszRfc822Name = pwsz;
|
|
while ((*pwsz != 0) && (*pwsz != '\n') && (*pwsz != '\r')) pwsz++;
|
|
}
|
|
else if (_wcsnicmp(pwsz, L"X500:", 5) == 0) {
|
|
pwsz += 5;
|
|
while (*pwsz == ' ') pwsz++;
|
|
for (pwsz1 = pwsz; ((*pwsz != 0) && (*pwsz != '\n') &&
|
|
(*pwsz != '\r')); pwsz++);
|
|
if (*pwsz != 0) {
|
|
*pwsz = 0;
|
|
pwsz++;
|
|
}
|
|
|
|
rgEntry[cEntry].dwAltNameChoice = CERT_ALT_NAME_DIRECTORY_NAME;
|
|
f = CertStrToNameW(X509_ASN_ENCODING, pwsz1,
|
|
CERT_NAME_STR_ENABLE_UTF8_UNICODE_FLAG, NULL,
|
|
NULL, &cb, NULL);
|
|
if (!f) {
|
|
MessageBox(hwnd, "X500 name does not parse", "smimetst", MB_OK);
|
|
return FALSE;
|
|
}
|
|
|
|
rgEntry[cEntry].DirectoryName.pbData = (LPBYTE) malloc(cb);
|
|
f = CertStrToNameW(X509_ASN_ENCODING, pwsz1,
|
|
CERT_NAME_STR_ENABLE_UTF8_UNICODE_FLAG, NULL,
|
|
rgEntry[cEntry].DirectoryName.pbData, &cb,
|
|
NULL);
|
|
if (!f) return FALSE;
|
|
rgEntry[cEntry].DirectoryName.cbData = cb;
|
|
}
|
|
else {
|
|
MessageBox(hwnd, "unknown address type", "smimetst", MB_OK);
|
|
return FALSE;
|
|
}
|
|
|
|
if (*pwsz == '\r') {
|
|
*pwsz = 0;
|
|
pwsz++;
|
|
}
|
|
if (*pwsz == '\n') {
|
|
*pwsz = 0;
|
|
pwsz++;
|
|
}
|
|
cEntry += 1;
|
|
}
|
|
|
|
*prgNames = (PCERT_NAME_BLOB) malloc(sizeof(CERT_NAME_BLOB) * cNames);
|
|
if (*prgNames == NULL) return FALSE;
|
|
|
|
for (i=0; i<cNames; i++) {
|
|
f = CryptEncodeObjectEx(X509_ASN_ENCODING, X509_ALTERNATE_NAME,
|
|
&rgNames[i], CRYPT_ENCODE_ALLOC_FLAG, NULL,
|
|
&(*prgNames)[i].pbData, &(*prgNames)[i].cbData);
|
|
if (!f) return f;
|
|
}
|
|
*pcNames = cNames;
|
|
return f;
|
|
}
|
|
|
|
BOOL ReceiptCreateDlgProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
DWORD cb;
|
|
DWORD cbNames;
|
|
BOOL f;
|
|
LPBYTE pb;
|
|
PSMIME_RECEIPT_REQUEST pReceipt;
|
|
static CSignData * psd = NULL;
|
|
CHAR rgchContent[256];
|
|
SMIME_RECEIPT_REQUEST receipt;
|
|
|
|
switch (message) {
|
|
case WM_INITDIALOG:
|
|
pReceipt = NULL;
|
|
|
|
psd = (CSignData *) lParam;
|
|
psd->GetReceiptData(&pb, &cbNames);
|
|
f = CryptDecodeObjectEx(X509_ASN_ENCODING, szOID_SMIME_Receipt_Request,
|
|
pb, cbNames, CRYPT_ENCODE_ALLOC_FLAG, NULL,
|
|
&pReceipt, &cb);
|
|
|
|
if (pReceipt) {
|
|
// Set Content Identifier -- M00TODO
|
|
SetDlgItemText(hwnd, IDC_RC_CONTENT,
|
|
(LPSTR) pReceipt->ContentIdentifier.pbData);
|
|
|
|
// Set Top buttons -
|
|
// M00BUG
|
|
if (pReceipt->ReceiptsFrom.cNames > 0) {
|
|
FormatNames(pReceipt->ReceiptsFrom.cNames,
|
|
pReceipt->ReceiptsFrom.rgNames, hwnd, IDC_RC_FROM_TEXT);
|
|
}
|
|
|
|
// Set ReceiptsTo
|
|
FormatNames(pReceipt->cReceiptsTo, pReceipt->rgReceiptsTo, hwnd,
|
|
IDC_RC_TO_TEXT);
|
|
}
|
|
else {
|
|
FormatNames(CMyNames, RgMyNames, hwnd, IDC_RC_TO_TEXT);
|
|
}
|
|
break;
|
|
|
|
case WM_COMMAND:
|
|
switch(wParam) {
|
|
case MAKELONG(IDC_RC_FROM_ALL, BN_CLICKED):
|
|
case MAKELONG(IDC_RC_FROM_TOP, BN_CLICKED):
|
|
EnableWindow(GetDlgItem(hwnd, IDC_RC_FROM_TEXT), FALSE);
|
|
break;
|
|
|
|
case MAKELONG(IDC_RC_FROM_SOME, BN_CLICKED):
|
|
EnableWindow(GetDlgItem(hwnd, IDC_RC_FROM_TEXT), TRUE);
|
|
break;
|
|
|
|
case IDOK:
|
|
memset(&receipt, 0, sizeof(receipt));
|
|
|
|
GetDlgItemText(hwnd, IDC_RC_CONTENT,
|
|
rgchContent, sizeof(rgchContent));
|
|
receipt.ContentIdentifier.pbData = (LPBYTE) rgchContent;
|
|
receipt.ContentIdentifier.cbData = strlen(rgchContent);
|
|
|
|
if (SendMessage(GetDlgItem(hwnd, IDC_RC_FROM_ALL), BM_GETCHECK, 0, 0)) {
|
|
receipt.ReceiptsFrom.AllOrFirstTier = SMIME_RECEIPTS_FROM_ALL;
|
|
}
|
|
else if (SendMessage(GetDlgItem(hwnd, IDC_RC_FROM_TOP), BM_GETCHECK, 0, 0)) {
|
|
receipt.ReceiptsFrom.AllOrFirstTier = SMIME_RECEIPTS_FROM_FIRST_TIER;
|
|
}
|
|
else {
|
|
receipt.ReceiptsFrom.AllOrFirstTier = 0;
|
|
if (!ParseNames(&receipt.ReceiptsFrom.cNames,
|
|
&receipt.ReceiptsFrom.rgNames,
|
|
hwnd, IDC_RC_FROM_TEXT)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if (!ParseNames(&receipt.cReceiptsTo, &receipt.rgReceiptsTo,
|
|
hwnd, IDC_RC_TO_TEXT)) {
|
|
free(receipt.ReceiptsFrom.rgNames);
|
|
return FALSE;
|
|
}
|
|
|
|
f = CryptEncodeObjectEx(X509_ASN_ENCODING,
|
|
szOID_SMIME_Receipt_Request, &receipt,
|
|
CRYPT_ENCODE_ALLOC_FLAG, NULL, &pb, &cb);
|
|
|
|
|
|
free(receipt.ReceiptsFrom.rgNames);
|
|
free(receipt.rgReceiptsTo);
|
|
if (!f) return FALSE;
|
|
|
|
psd->SetReceiptData(pb, cb);
|
|
|
|
//
|
|
case IDCANCEL:
|
|
EndDialog(hwnd, wParam);
|
|
break;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|