Merge pull request #498 from Unroll-Me/master
[mono.git] / mcs / class / System / Mono.Http / NtlmClient.cs
1 //
2 // Mono.Http.NtlmClient
3 //
4 // Authors:
5 //      Gonzalo Paniagua Javier (gonzalo@ximian.com)
6 //
7 // (c) 2003 Novell, Inc. (http://www.novell.com)
8 //
9
10 //
11 // Permission is hereby granted, free of charge, to any person obtaining
12 // a copy of this software and associated documentation files (the
13 // "Software"), to deal in the Software without restriction, including
14 // without limitation the rights to use, copy, modify, merge, publish,
15 // distribute, sublicense, and/or sell copies of the Software, and to
16 // permit persons to whom the Software is furnished to do so, subject to
17 // the following conditions:
18 // 
19 // The above copyright notice and this permission notice shall be
20 // included in all copies or substantial portions of the Software.
21 // 
22 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
23 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
24 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
25 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
26 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
27 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
28 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
29 //
30
31 #if SECURITY_DEP
32 using System;
33 using System.Collections;
34 using System.Net;
35 using Mono.Security.Protocol.Ntlm;
36
37 namespace Mono.Http
38 {
39         class NtlmSession
40         {
41                 MessageBase message;
42
43                 public NtlmSession () 
44                 {
45                 }
46
47                 public Authorization Authenticate (string challenge, WebRequest webRequest, ICredentials credentials) 
48                 {
49                         HttpWebRequest request = webRequest as HttpWebRequest;
50                         if (request == null)
51                                 return null;
52         
53                         NetworkCredential cred = credentials.GetCredential (request.RequestUri, "NTLM");
54                         if (cred == null)
55                                 return null;
56
57                         string userName = cred.UserName;
58                         string domain = cred.Domain;
59                         string password = cred.Password;
60                         if (userName == null || userName == "")
61                                 return null;
62
63                         if (String.IsNullOrEmpty (domain)) {
64                                 int idx = userName.IndexOf ('\\');
65                                 if (idx == -1) {
66                                         idx = userName.IndexOf ('/');
67                                 }
68                                 if (idx >= 0) {
69                                         domain = userName.Substring (0, idx);
70                                         userName = userName.Substring (idx + 1);
71                                 }
72                         }
73
74                         bool completed = false;
75                         if (message == null) {
76                                 Type1Message type1 = new Type1Message ();
77                                 type1.Domain = domain;
78                                 type1.Host = ""; // MS does not send it
79                                 type1.Flags |= NtlmFlags.NegotiateNtlm2Key;
80                                 message = type1;
81                         } else if (message.Type == 1) {
82                                 // Should I check the credentials?
83                                 if (challenge == null) {
84                                         message = null;
85                                         return null;
86                                 }
87
88                                 Type2Message type2 = new Type2Message (Convert.FromBase64String (challenge));
89                                 if (password == null)
90                                         password = "";
91
92                                 Type3Message type3 = new Type3Message (type2);
93                                 type3.Username = userName;
94                                 type3.Password = password;
95                                 message = type3;
96                                 completed = true;
97                         } else {
98                                 // Should I check the credentials?
99                                 // type must be 3 here
100                                 if (challenge == null || challenge == String.Empty) {
101                                         Type1Message type1 = new Type1Message ();
102                                         type1.Domain = domain;
103                                         type1.Host = ""; // MS does not send it
104                                         message = type1;
105                                 } else {
106                                         completed = true;
107                                 }
108                         }
109                         
110                         string token = "NTLM " + Convert.ToBase64String (message.GetBytes ());
111                         return new Authorization (token, completed);
112                 }
113         }
114
115         class NtlmClient : IAuthenticationModule
116         {
117                 static Hashtable cache;
118
119                 static NtlmClient () 
120                 {
121                         cache = new Hashtable ();
122                 }
123         
124                 public NtlmClient () {}
125         
126                 public Authorization Authenticate (string challenge, WebRequest webRequest, ICredentials credentials) 
127                 {
128                         if (credentials == null || challenge == null)
129                                 return null;
130         
131                         string header = challenge.Trim ();
132                         int idx = header.ToLower ().IndexOf ("ntlm");
133                         if (idx == -1)
134                                 return null;
135
136                         idx = header.IndexOfAny (new char [] {' ', '\t'});
137                         if (idx != -1) {
138                                 header = header.Substring (idx).Trim ();
139                         } else {
140                                 header = null;
141                         }
142
143                         HttpWebRequest request = webRequest as HttpWebRequest;
144                         if (request == null)
145                                 return null;
146
147                         lock (cache) {
148                                 NtlmSession ds = (NtlmSession) cache [request];
149                                 if (ds == null) {
150                                         ds = new NtlmSession ();
151                                         cache.Add (request, ds);
152                                 }
153
154                                 return ds.Authenticate (header, webRequest, credentials);
155                         }
156                 }
157
158                 public Authorization PreAuthenticate (WebRequest webRequest, ICredentials credentials) 
159                 {
160                         return null;
161                 }
162         
163                 public string AuthenticationType { 
164                         get { return "NTLM"; }
165                 }
166         
167                 public bool CanPreAuthenticate { 
168                         get { return false; }
169                 }
170         }
171 }
172 #endif