5-10 C1
This commit is contained in:
parent
9347e07039
commit
351f141a6b
|
@ -29,74 +29,143 @@
|
|||
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#define WIN32_NO_STATUS
|
||||
#include <Windows.h>
|
||||
#undef WIN32_NO_STATUS
|
||||
#include <Ntsecapi.h>
|
||||
//#include <ntstatus.h>
|
||||
#include <ntstatus.h>
|
||||
#include "agent.h"
|
||||
#include "agent-request.h"
|
||||
|
||||
static void
|
||||
InitLsaString(LSA_STRING *lsa_string, const char *str)
|
||||
{
|
||||
if (str == NULL)
|
||||
memset(lsa_string, 0, sizeof(LSA_STRING));
|
||||
else {
|
||||
lsa_string->Buffer = str;
|
||||
lsa_string->Length = strlen(str);
|
||||
lsa_string->MaximumLength = lsa_string->Length + 1;
|
||||
}
|
||||
}
|
||||
|
||||
int process_authagent_request(struct sshbuf* request, struct sshbuf* response, struct agent_connection* con) {
|
||||
while (1)
|
||||
{
|
||||
HANDLE lsa_handle;
|
||||
PLSA_OPERATIONAL_MODE mode;
|
||||
static HANDLE
|
||||
generate_user_token(wchar_t* user) {
|
||||
HANDLE lsa_handle = 0, token = 0;;
|
||||
LSA_OPERATIONAL_MODE mode;
|
||||
ULONG auth_package_id;
|
||||
NTSTATUS ret;
|
||||
KERB_S4U_LOGON *s4u_logon;
|
||||
NTSTATUS ret, subStatus;
|
||||
KERB_S4U_LOGON *s4u_logon = NULL;
|
||||
size_t logon_info_size;
|
||||
LSA_STRING logon_process_name, auth_package_name, originName;
|
||||
TOKEN_SOURCE sourceContext;
|
||||
PKERB_INTERACTIVE_PROFILE pProfile = NULL;
|
||||
LUID logonId;
|
||||
QUOTA_LIMITS quotas;
|
||||
DWORD cbProfile;
|
||||
|
||||
InitLsaString(&logon_process_name, "ssh-agent");
|
||||
//InitLsaString(&auth_package_name, MICROSOFT_KERBEROS_NAME_A);
|
||||
InitLsaString(&auth_package_name, "Negotiate");
|
||||
InitLsaString(&auth_package_name, MICROSOFT_KERBEROS_NAME_A);
|
||||
//InitLsaString(&auth_package_name, "Negotiate");
|
||||
InitLsaString(&originName, "sshd");
|
||||
if (ret = LsaRegisterLogonProcess(&logon_process_name, &lsa_handle, &mode) != STATUS_SUCCESS)
|
||||
break;
|
||||
goto done;
|
||||
|
||||
if (ret = LsaLookupAuthenticationPackage(lsa_handle, &auth_package_name, &auth_package_id) != STATUS_SUCCESS)
|
||||
break;
|
||||
#define USER_NAME L"user@domain"
|
||||
goto done;
|
||||
|
||||
logon_info_size = sizeof(KERB_S4U_LOGON);
|
||||
logon_info_size += (wcslen(USER_NAME) * 2 + 2);
|
||||
logon_info_size += (wcslen(user) * 2 + 2);
|
||||
s4u_logon = malloc(logon_info_size);
|
||||
if (s4u_logon == NULL)
|
||||
goto done;
|
||||
|
||||
s4u_logon->MessageType = KerbS4ULogon;
|
||||
s4u_logon->Flags = 0;
|
||||
s4u_logon->ClientUpn.Length = wcslen(USER_NAME) * 2;
|
||||
s4u_logon->ClientUpn.Length = wcslen(user) * 2;
|
||||
s4u_logon->ClientUpn.MaximumLength = s4u_logon->ClientUpn.Length;
|
||||
s4u_logon->ClientUpn.Buffer = (WCHAR*)(s4u_logon + 1);
|
||||
memcpy(s4u_logon->ClientUpn.Buffer, USER_NAME, s4u_logon->ClientUpn.Length + 2);
|
||||
memcpy(s4u_logon->ClientUpn.Buffer, user, s4u_logon->ClientUpn.Length + 2);
|
||||
s4u_logon->ClientRealm.Length = 0;
|
||||
s4u_logon->ClientRealm.MaximumLength = 0;
|
||||
s4u_logon->ClientRealm.Buffer = 0;
|
||||
|
||||
TOKEN_SOURCE sourceContext;
|
||||
RtlCopyMemory(
|
||||
sourceContext.SourceName,
|
||||
".Jobs ",
|
||||
sizeof(sourceContext.SourceName)
|
||||
);
|
||||
memcpy(sourceContext.SourceName,".Jobs ", sizeof(sourceContext.SourceName));
|
||||
|
||||
if (AllocateLocallyUniqueId(&sourceContext.SourceIdentifier) != TRUE)
|
||||
break;
|
||||
goto done;
|
||||
|
||||
PKERB_INTERACTIVE_PROFILE pProfile = NULL;
|
||||
LUID logonId;
|
||||
QUOTA_LIMITS quotas;
|
||||
NTSTATUS subStatus;
|
||||
DWORD cbProfile;
|
||||
HANDLE hToken = INVALID_HANDLE_VALUE;
|
||||
if (ret = LsaLogonUser(lsa_handle, &originName, Network, auth_package_id, s4u_logon, logon_info_size, NULL, &sourceContext,
|
||||
if (ret = LsaLogonUser(lsa_handle,
|
||||
&originName,
|
||||
Network,
|
||||
auth_package_id,
|
||||
s4u_logon,
|
||||
logon_info_size,
|
||||
NULL,
|
||||
&sourceContext,
|
||||
(PVOID*)&pProfile,
|
||||
&cbProfile,
|
||||
&logonId,
|
||||
&hToken,
|
||||
&token,
|
||||
"as,
|
||||
&subStatus) != STATUS_SUCCESS)
|
||||
break;
|
||||
goto done;
|
||||
|
||||
CloseHandle(hToken);
|
||||
done:
|
||||
if (lsa_handle)
|
||||
LsaDeregisterLogonProcess(lsa_handle);
|
||||
break;
|
||||
}
|
||||
return -1;
|
||||
if (s4u_logon)
|
||||
free(s4u_logon);
|
||||
if (pProfile)
|
||||
LsaFreeReturnBuffer(pProfile);
|
||||
|
||||
return token;
|
||||
}
|
||||
|
||||
#define AUTH_REQUEST "keyauthenticate"
|
||||
#define MAX_USER_NAME_LEN 255 + 255
|
||||
|
||||
int process_authagent_request(struct sshbuf* request, struct sshbuf* response, struct agent_connection* con) {
|
||||
int r = 0;
|
||||
char* opn, key_blob, user, sig, blob;
|
||||
size_t opn_len, key_blob_len, user_len, sig_len, blob_len;
|
||||
struct sshkey *key = NULL;
|
||||
HANDLE token = NULL, dup_token = NULL;
|
||||
wchar_t wuser[MAX_USER_NAME_LEN];
|
||||
PWSTR wuser_home = NULL;
|
||||
|
||||
user = NULL;
|
||||
if ((r = sshbuf_get_string_direct(request, &opn, &opn_len)) != 0 ||
|
||||
(r = sshbuf_get_string_direct(request, &key_blob, &key_blob_len)) != 0 ||
|
||||
(r = sshbuf_get_cstring(request, &user, &user_len)) != 0 ||
|
||||
(r = sshbuf_get_string_direct(request, &sig, &sig_len)) != 0 ||
|
||||
(r = sshbuf_get_string_direct(request, &blob, &blob_len)) != 0 ||
|
||||
(r = sshkey_from_blob(key_blob, key_blob_len, &key)) != 0)
|
||||
goto done;
|
||||
|
||||
if ((opn_len != strlen(AUTH_REQUEST)) || (memcmp(opn, AUTH_REQUEST, opn_len) != 0)) {
|
||||
r = EINVAL;
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (0 == MultiByteToWideChar(CP_UTF8, 0, user, user_len + 1, wuser, MAX_USER_NAME_LEN) {
|
||||
r = GetLastError();
|
||||
goto done;
|
||||
}
|
||||
|
||||
if ((token = generate_user_token(wuser)) == 0) {
|
||||
r = EINVAL;
|
||||
goto done;
|
||||
}
|
||||
|
||||
done:
|
||||
if (user)
|
||||
free(user);
|
||||
if (key)
|
||||
sshkey_free(key);
|
||||
if (token)
|
||||
CloseHandle(token);
|
||||
if (wuser_home)
|
||||
CoTaskMemFree(wuser_home);
|
||||
return r;
|
||||
}
|
|
@ -73,9 +73,6 @@ convert_blob(struct agent_connection* con, const char *blob, DWORD blen, char **
|
|||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
*eblob = malloc(out.cbData);
|
||||
if (*eblob == NULL) {
|
||||
r = ERROR_OUTOFMEMORY;
|
||||
|
@ -133,7 +130,7 @@ process_add_identity(struct sshbuf* request, struct sshbuf* response, struct age
|
|||
|
||||
done:
|
||||
|
||||
/* TODO if r failed the delete reg entries*/
|
||||
/* TODO if r failed, delete reg entries*/
|
||||
|
||||
r1 = sshbuf_put_u8(response, (r==0) ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
|
||||
|
||||
|
|
Loading…
Reference in New Issue