summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source4/smb_server/smb/negprot.c64
1 files changed, 34 insertions, 30 deletions
diff --git a/source4/smb_server/smb/negprot.c b/source4/smb_server/smb/negprot.c
index 6bac657e8c..b2f136f4f7 100644
--- a/source4/smb_server/smb/negprot.c
+++ b/source4/smb_server/smb/negprot.c
@@ -443,10 +443,10 @@ static const struct {
void reply_negprot(struct smbsrv_request *req)
{
- int Index=0;
- int choice = -1;
int protocol;
uint8_t *p;
+ uint32_t protos_count = 0;
+ char **protos = NULL;
if (req->smb_conn->negotiate.done_negprot) {
smbsrv_terminate_connection(req->smb_conn, "multiple negprot's are not permitted");
@@ -454,37 +454,41 @@ void reply_negprot(struct smbsrv_request *req)
}
req->smb_conn->negotiate.done_negprot = True;
- p = req->in.data + 1;
+ p = req->in.data;
+ while (True) {
+ size_t len;
- while (p < req->in.data + req->in.data_size) {
- Index++;
- DEBUG(3,("Requested protocol [%s]\n",(const char *)p));
- p += strlen((const char *)p) + 2;
+ protos = talloc_realloc(req, protos, char *, protos_count + 1);
+ if (!protos) {
+ smbsrv_terminate_connection(req->smb_conn, nt_errstr(NT_STATUS_NO_MEMORY));
+ return;
+ }
+ protos[protos_count] = NULL;
+ len = req_pull_ascii4(req, (const char **)&protos[protos_count], p, STR_ASCII|STR_TERMINATE);
+ p += len;
+ if (len == 0 || !protos[protos_count]) break;
+
+ DEBUG(3,("Requested protocol [%d][%s]\n", protos_count, protos[protos_count]));
+ protos_count++;
}
-
+
/* Check for protocols, most desirable first */
for (protocol = 0; supported_protocols[protocol].proto_name; protocol++) {
- p = req->in.data+1;
- Index = 0;
- if ((supported_protocols[protocol].protocol_level <= lp_maxprotocol()) &&
- (supported_protocols[protocol].protocol_level >= lp_minprotocol()))
- while (p < (req->in.data + req->in.data_size)) {
- if (strequal((const char *)p,supported_protocols[protocol].proto_name))
- choice = Index;
- Index++;
- p += strlen((const char *)p) + 2;
- }
- if(choice != -1)
- break;
- }
-
- if(choice != -1) {
- sub_set_remote_proto(supported_protocols[protocol].short_name);
- supported_protocols[protocol].proto_reply_fn(req, choice);
- DEBUG(3,("Selected protocol %s\n",supported_protocols[protocol].proto_name));
- } else {
- DEBUG(0,("No protocol supported !\n"));
+ int i;
+
+ if (supported_protocols[protocol].protocol_level > lp_maxprotocol()) continue;
+ if (supported_protocols[protocol].protocol_level < lp_minprotocol()) continue;
+
+ for (i = 0; i < protos_count; i++) {
+ if (strcmp(supported_protocols[protocol].proto_name, protos[i]) != 0) continue;
+
+ supported_protocols[protocol].proto_reply_fn(req, i);
+ sub_set_remote_proto(supported_protocols[protocol].short_name);
+ DEBUG(3,("Selected protocol [%d][%s]\n",
+ i, supported_protocols[protocol].proto_name));
+ return;
+ }
}
-
- DEBUG(5,("negprot index=%d\n", choice));
+
+ smbsrv_terminate_connection(req->smb_conn, "No protocol supported !");
}