Author: tridge Date: 2005-12-05 06:01:22 +0000 (Mon, 05 Dec 2005) New Revision: 12063
WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=12063 Log: fixed the krb5 client code to handle ICMP port unreachable errors, and error out immediatelly. This prevents a long timeout Modified: branches/SAMBA_4_0/source/auth/kerberos/krb5_init_context.c Changeset: Modified: branches/SAMBA_4_0/source/auth/kerberos/krb5_init_context.c =================================================================== --- branches/SAMBA_4_0/source/auth/kerberos/krb5_init_context.c 2005-12-05 04:10:13 UTC (rev 12062) +++ branches/SAMBA_4_0/source/auth/kerberos/krb5_init_context.c 2005-12-05 06:01:22 UTC (rev 12063) @@ -84,30 +84,41 @@ static void smb_krb5_socket_recv(struct smb_krb5_socket *smb_krb5) { TALLOC_CTX *tmp_ctx = talloc_new(smb_krb5); - NTSTATUS status; DATA_BLOB blob; size_t nread, dsize; switch (smb_krb5->hi->proto) { case KRB5_KRBHST_UDP: - status = socket_pending(smb_krb5->sock, &dsize); - if (!NT_STATUS_IS_OK(status)) { + smb_krb5->status = socket_pending(smb_krb5->sock, &dsize); + if (!NT_STATUS_IS_OK(smb_krb5->status)) { talloc_free(tmp_ctx); return; } + if (dsize == 0) { + smb_krb5->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR; + talloc_free(tmp_ctx); + return; + } blob = data_blob_talloc(tmp_ctx, NULL, dsize); if (blob.data == NULL) { + smb_krb5->status = NT_STATUS_NO_MEMORY; talloc_free(tmp_ctx); return; } - status = socket_recv(smb_krb5->sock, blob.data, blob.length, &nread, 0); - if (!NT_STATUS_IS_OK(status)) { + smb_krb5->status = socket_recv(smb_krb5->sock, blob.data, blob.length, &nread, 0); + if (!NT_STATUS_IS_OK(smb_krb5->status)) { talloc_free(tmp_ctx); return; } blob.length = nread; + + if (nread == 0) { + smb_krb5->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR; + talloc_free(tmp_ctx); + return; + } DEBUG(2,("Received smb_krb5 packet of length %d\n", (int)blob.length)); @@ -131,17 +142,14 @@ if (smb_krb5->partial_read < 4) { uint32_t packet_length; - status = socket_recv(smb_krb5->sock, + smb_krb5->status = socket_recv(smb_krb5->sock, smb_krb5->partial.data + smb_krb5->partial_read, 4 - smb_krb5->partial_read, &nread, 0); - if (NT_STATUS_IS_ERR(status)) { - smb_krb5->status = status; + /* todo: this should be converted to the packet_*() routines */ + if (!NT_STATUS_IS_OK(smb_krb5->status)) { return; } - if (!NT_STATUS_IS_OK(status)) { - return; - } smb_krb5->partial_read += nread; if (smb_krb5->partial_read != 4) { @@ -161,15 +169,11 @@ } /* read in the body */ - status = socket_recv(smb_krb5->sock, + smb_krb5->status = socket_recv(smb_krb5->sock, smb_krb5->partial.data + smb_krb5->partial_read, smb_krb5->partial.length - smb_krb5->partial_read, &nread, 0); - if (NT_STATUS_IS_ERR(status)) { - smb_krb5->status = status; - return; - } - if (!NT_STATUS_IS_OK(status)) return; + if (!NT_STATUS_IS_OK(smb_krb5->status)) return; smb_krb5->partial_read += nread;