Hi Akihiro,
Thanks for working on that. I have a few comments to the patch.
On 17.04.2017 16:26, Akihiro Sagawa wrote:
> diff --git a/dlls/secur32/schannel.c b/dlls/secur32/schannel.c
> index 6914837..317686f 100644
> --- a/dlls/secur32/schannel.c
> +++ b/dlls/secur32/schannel.c
> @@ -54,11 +54,14 @@ struct schan_handle
> enum schan_handle_type type;
> };
>
> +#define SCHAN_KEY_NAME_MAX 8
> struct schan_context
> {
> schan_imp_session session;
> ULONG req_ctx_attr;
> const CERT_CONTEXT *cert;
> + SEC_WCHAR key_signature_name[SCHAN_KEY_NAME_MAX];
> + SEC_WCHAR key_encrypt_name[SCHAN_KEY_NAME_MAX];
Those should not be needed. You could for example store both WCHAR and
char strings inside alg_name_map.
> };
>
> static struct schan_handle *schan_handle_table;
> @@ -962,6 +965,53 @@ static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextA(
> return ret;
> }
>
> +struct schan_alg_id_name {
> + ALG_ID alg_id;
> + const char* name;
> +};
> +
> +static int comp_alg_id(const void *a, const void *b)
> +{
> + const struct schan_alg_id_name *lhs = a;
> + const struct schan_alg_id_name *rhs = b;
> + return (int)lhs->alg_id - (int)rhs->alg_id;
> +}
> +
> +static void fill_alg_name(ALG_ID id, void *buffer, BOOL wide)
If you change the patch as I suggested above, this could be something like:
static void *get_alg_name(ALF_ID id, BOOL wide)
> +{
> + static const struct schan_alg_id_name alg_name_map[] = {
> + { CALG_ECDSA, "ECDSA" },
> + { CALG_RSA_SIGN, "RSA" },
> + { CALG_DES, "DES" },
> + { CALG_RC2, "RC2" },
> + { CALG_3DES, "3DES" },
> + { CALG_AES_128, "AES" },
> + { CALG_AES_192, "AES" },
> + { CALG_AES_256, "AES" },
> + { CALG_RC4, "RC4" },
> + };
> + const struct schan_alg_id_name *res;
> + struct schan_alg_id_name key;
> + const char *name;
> +
> + key.alg_id = id;
> + res = bsearch(&key, alg_name_map,
> + sizeof(alg_name_map)/sizeof(alg_name_map[0]), sizeof(alg_name_map[0]),
> + comp_alg_id);
> + if (res)
> + name = res->name;
> + else
> + {
> + TRACE("Unknown ALG_ID %04x\n", id);
> + name = "???";
> + }
A FIXME would would be nice here. Also, I think you could just return
NULL in this case.
> +
> + if (wide)
> + MultiByteToWideChar(CP_ACP, 0, name, -1, (SEC_WCHAR*)buffer, SCHAN_KEY_NAME_MAX);
> + else
> + strcpy((SEC_CHAR*)buffer, name);
> +}
> +
> static SECURITY_STATUS ensure_remote_cert(struct schan_context *ctx)
> {
> HCERTSTORE cert_store;
> @@ -989,6 +1039,7 @@ static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesW(
>
> if (!context_handle) return SEC_E_INVALID_HANDLE;
> ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
> + if (!ctx) return SEC_E_INVALID_HANDLE;
This has an effect outside of scope of this patch. This should be be a
separated patch, with a test case.
> switch(attribute)
> {
> @@ -1016,6 +1067,19 @@ static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesW(
>
> return status;
> }
> + case SECPKG_ATTR_KEY_INFO:
> + {
> + SecPkgContext_KeyInfoW *info = buffer;
> + SECURITY_STATUS status = schan_imp_get_key_info(ctx->session, info);
> + if (status == SEC_E_OK)
> + {
> + info->sSignatureAlgorithmName = ctx->key_signature_name;
> + info->sEncryptAlgorithmName = ctx->key_encrypt_name;
> + fill_alg_name(info->SignatureAlgorithm, info->sSignatureAlgorithmName, TRUE);
> + fill_alg_name(info->EncryptAlgorithm, info->sEncryptAlgorithmName, TRUE);
> + }
> + return status;
> + }
> case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
> {
> PCCERT_CONTEXT *cert = buffer;
> @@ -1091,6 +1155,24 @@ static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesA(
> {
> case SECPKG_ATTR_STREAM_SIZES:
> return schan_QueryContextAttributesW(context_handle, attribute, buffer);
> + case SECPKG_ATTR_KEY_INFO:
> + {
> + SecPkgContext_KeyInfoA *info = buffer;
> + struct schan_context *ctx;
> + SECURITY_STATUS status;
> + if (!context_handle) return SEC_E_INVALID_HANDLE;
> + ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
> + if (!ctx) return SEC_E_INVALID_HANDLE;
> + status = schan_imp_get_key_info(ctx->session, (SecPkgContext_KeyInfoW*)info);
> + if (status == SEC_E_OK)
> + {
> + info->sSignatureAlgorithmName = (SEC_CHAR *)ctx->key_signature_name;
> + info->sEncryptAlgorithmName = (SEC_CHAR *)ctx->key_encrypt_name;
> + fill_alg_name(info->SignatureAlgorithm, info->sSignatureAlgorithmName, FALSE);
> + fill_alg_name(info->EncryptAlgorithm, info->sEncryptAlgorithmName, FALSE);
> + }
> + return status;
> + }
> case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
> return schan_QueryContextAttributesW(context_handle, attribute, buffer);
> case SECPKG_ATTR_CONNECTION_INFO:
> diff --git a/dlls/secur32/schannel_gnutls.c b/dlls/secur32/schannel_gnutls.c
> index 30640d3..ddf33a8 100644
> --- a/dlls/secur32/schannel_gnutls.c
> +++ b/dlls/secur32/schannel_gnutls.c
> @@ -363,6 +363,22 @@ static ALG_ID schannel_get_kx_algid(int kx)
> }
> }
>
> +static ALG_ID schannel_get_kx_signature_algid(gnutls_kx_algorithm_t kx)
> +{
> + switch (kx)
> + {
> + case GNUTLS_KX_UNKNOWN: return 0;
> + case GNUTLS_KX_RSA:
> + case GNUTLS_KX_RSA_EXPORT:
> + case GNUTLS_KX_DHE_RSA:
> + case GNUTLS_KX_ECDHE_RSA: return CALG_RSA_SIGN;
> + case GNUTLS_KX_ECDHE_ECDSA: return CALG_ECDSA;
> + default:
> + FIXME("unknown algorithm %d\n", kx);
> + return 0;
> + }
> +}
> +
> unsigned int schan_imp_get_session_cipher_block_size(schan_imp_session session)
> {
> gnutls_session_t s = (gnutls_session_t)session;
> @@ -394,6 +410,23 @@ SECURITY_STATUS schan_imp_get_connection_info(schan_imp_session session,
> return SEC_E_OK;
> }
>
> +SECURITY_STATUS schan_imp_get_key_info(schan_imp_session session, SecPkgContext_KeyInfoW *info)
> +{
> + gnutls_session_t s = (gnutls_session_t)session;
> + gnutls_cipher_algorithm_t alg = pgnutls_cipher_get(s);
> + gnutls_kx_algorithm_t kx = pgnutls_kx_get(s);
> +
> + TRACE("(%p,%p)\n", session, info);
> +
> + info->sSignatureAlgorithmName = NULL; /* filled later */
> + info->sEncryptAlgorithmName = NULL; /* filled later */
> + info->KeySize = pgnutls_cipher_get_key_size(alg) * 8;
> + info->SignatureAlgorithm = schannel_get_kx_signature_algid(kx);
> + info->EncryptAlgorithm = schannel_get_cipher_algid(alg);
> +
> + return SEC_E_OK;
> +}
I don't think such helper that fills output partially is the right
solution here. You could use schan_imp_get_connection_info to get all
info you need except signature algorithm. For signature algorithm, you
may introduce a new, simpler, function.
Thanks,
Jacek