2 /* -----------------------------------------------------------------------
3 * <copyright file="PithosAccount.cs" company="GRNet">
5 * Copyright 2011-2012 GRNET S.A. All rights reserved.
7 * Redistribution and use in source and binary forms, with or
8 * without modification, are permitted provided that the following
11 * 1. Redistributions of source code must retain the above
12 * copyright notice, this list of conditions and the following
15 * 2. Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following
17 * disclaimer in the documentation and/or other materials
18 * provided with the distribution.
21 * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
22 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
23 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
25 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
28 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32 * POSSIBILITY OF SUCH DAMAGE.
34 * The views and conclusions contained in the software and
35 * documentation are those of the authors and should not be
36 * interpreted as representing official policies, either expressed
37 * or implied, of GRNET S.A.
39 * -----------------------------------------------------------------------
44 // -----------------------------------------------------------------------
47 using System.Reflection;
48 using Pithos.Client.WPF.Preferences;
49 using Pithos.Client.WPF.Properties;
53 namespace Pithos.Client.WPF
56 using System.Collections.Generic;
60 using System.Threading.Tasks;
61 using System.Diagnostics.Contracts;
62 using System.Diagnostics;
63 using System.Net.Sockets;
66 /// TODO: Update summary.
69 /// Retrieves an account name and token from PITHOS
71 public static class PithosAccount
73 private static readonly ILog Log = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
76 /// Asynchronously retrieves PITHOS credentials
78 /// <param name="loginUrl">URL to retrieve the account info from PITHOS. Must end with =</param>
79 /// <returns>The credentials wrapped in a Task</returns>
80 public static NetworkCredential RetrieveCredentials(string loginUrl,string accountName=null)
82 Contract.Requires(Uri.IsWellFormedUriString(loginUrl, UriKind.Absolute));
84 if (!Uri.IsWellFormedUriString(loginUrl, UriKind.Absolute))
85 throw new ArgumentException("The pithosSite parameter must be a valid absolute URL", "loginUrl");
87 //int port = GetFreePort();
89 //TODO:Force logout here to take care of existing cookies
92 //var listenerUrl = String.Format("http://127.0.0.1:{0}", port);
93 var listenerUrl = "pithos://127.0.0.1";
97 //var receiveCredentials = ListenForRedirectAsync(port);
99 //var logoutUrl = loginUrl.Replace("login", "im/logout");
100 //var logoutProcess=Process.Start(logoutUrl);
101 //TaskEx.Delay(2000).Wait();
102 var uriBuilder = new UriBuilder(loginUrl)
104 Query = String.Format("next={0}&force=", listenerUrl)
107 var retrieveUri = uriBuilder.Uri;
109 var logoutUrl= (from server in Settings.Default.Servers
110 where server.LoginUri == loginUrl
111 select new Uri(server.LogoutUri)).FirstOrDefault();
115 var browser = new LoginView(retrieveUri,logoutUrl,accountName);
116 if (true == browser.ShowDialog())
118 return new NetworkCredential(browser.Account, browser.Token);
124 Log.InfoFormat("[RETRIEVE] Open Browser at {0}", retrieveUri);
125 Process.Start(retrieveUri.ToString());
127 return receiveCredentials;
132 private static async Task<NetworkCredential> ListenHttpForRedirectAsync(string listenerUrl)
134 using (var listener = new HttpListener())
136 listener.Prefixes.Add(listenerUrl);
138 Log.InfoFormat("[RETRIEVE] Listening at {0}", listenerUrl);
145 var context = await listener.GetContextAsync()
146 .WithTimeout(TimeSpan.FromMinutes(5));
148 var request = context.Request;
149 Log.InfoFormat("[RETRIEVE] Got Connection {0}", request.RawUrl);
151 var query = request.QueryString;
152 var userName = query["user"];
153 var token = query["token"];
154 if (String.IsNullOrWhiteSpace(userName) ||
155 String.IsNullOrWhiteSpace(token))
157 Respond(context, "Failure", "The server did not return a username or token");
158 Log.ErrorFormat("[RETRIEVE] No credentials returned by server");
159 throw new Exception("The server did not return a username or token");
163 Log.InfoFormat("[RETRIEVE] Credentials retrieved user:{0} token:{1}", userName, token);
164 Respond(context, "Authenticated", "Got It");
166 return new NetworkCredential(userName, token);
169 catch (Exception exc)
171 Log.Error("[RETRIEVE][ERROR] Receive connection {0}", exc);
178 private static async Task<NetworkCredential> ListenForRedirectAsync(int port)
180 var listener = new TcpListener(IPAddress.Any, port);
181 Log.InfoFormat("[RETRIEVE] Listening at {0}", port);
187 using (var client = await listener.AcceptTcpClientAsync()
188 .WithTimeout(TimeSpan.FromMinutes(5)))
190 using (var stream = client.GetStream())
191 using (var reader=new StreamReader(stream))
193 var request = await reader.ReadLineAsync();
195 //TODO: Add proper warnings here if the content is empty, don't just throw an exception
196 //This may be a common occurence
198 if (String.IsNullOrWhiteSpace(request))
199 throw new PithosException("The server did send any information");
200 Log.InfoFormat("[RETRIEVE] Got Connection {0}", request);
203 var items = ParseResponse(request);
205 var userName = items["user"];
206 var token = items["token"];
207 if (String.IsNullOrWhiteSpace(userName) ||
208 String.IsNullOrWhiteSpace(token))
210 Respond(stream, "Failure", "The server did not return a username or token");
211 Log.ErrorFormat("[RETRIEVE] No credentials returned by server");
212 throw new PithosException("The server did not return a username or token");
216 Log.InfoFormat("[RETRIEVE] Credentials retrieved user:{0} token:{1}", userName, token);
217 Respond(stream, "Authenticated", "Got It");
219 return new NetworkCredential(userName, token);
223 catch (Exception exc)
225 Log.Error("[RETRIEVE][ERROR] Receive connection {0}", exc);
226 throw new PithosException("An error occured while retrieving credentials",exc);
230 private static Dictionary<string, string> ParseResponse(string request)
232 var parts = request.Split(' ');
233 var query = parts[1].TrimStart('/', '?');
235 var items = query.Split('&')
236 .Select(pair => pair.Split('='))
237 .ToDictionary(arr => arr[0].ToLower(), arr => Uri.UnescapeDataString(arr[1]));
242 private static void Respond(HttpListenerContext context,string title,string message)
244 var response = context.Response;
245 var html = String.Format("<html><head><title>{0}</title></head><body><h1>{1}</h1></body></html>", title, message);
246 var outBuffer = Encoding.UTF8.GetBytes(html);
247 response.ContentLength64 = outBuffer.Length;
248 using (var stream = response.OutputStream)
250 stream.Write(outBuffer, 0, outBuffer.Length);
253 Log.InfoFormat("[RETRIEVE] Responded");
256 private static void Respond(Stream stream,string title,string message)
259 var html = String.Format("<html><head><title>{0}</title></head><body><h1>{1}</h1></body></html>", title, message);
261 var response = new StringBuilder();
262 response.AppendLine("HTTP/1.1 200 OK");
263 response.AppendFormat("Content-Length: {0}\n", html.Length);
264 response.AppendLine("Server: Microsoft-HTTPAPI/2.0");
265 response.AppendFormat("Date: {0}\n\n", DateTime.UtcNow);
266 response.AppendLine(html);
268 var outBuffer=Encoding.UTF8.GetBytes(response.ToString());
270 stream.Write(outBuffer, 0, outBuffer.Length);
272 Log.InfoFormat("[RETRIEVE] Responded");
277 /// Locates a free local port
279 /// <returns>A free port</returns>
280 /// <remarks>The method starts and stops a TcpListener on port 0 to locate a free port.</remarks>
281 public static int GetFreePort()
283 //The TcpListener will locate a free port
284 var listener = new TcpListener(IPAddress.Any, 0);
286 var port = ((IPEndPoint)listener.LocalEndpoint).Port;