New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
deps: replace Jetty with HttpServer #433
Changes from 2 commits
cb0a581
48fb3f2
84c457e
2754068
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -16,15 +16,19 @@ | |
|
||
import com.google.api.client.extensions.java6.auth.oauth2.VerificationCodeReceiver; | ||
import com.google.api.client.util.Throwables; | ||
import com.sun.net.httpserver.*; | ||
|
||
import java.io.IOException; | ||
import java.io.OutputStream; | ||
import java.io.PrintWriter; | ||
import java.net.InetSocketAddress; | ||
import java.net.ServerSocket; | ||
import java.util.HashMap; | ||
import java.util.Map; | ||
import java.util.concurrent.Semaphore; | ||
import javax.servlet.http.HttpServletRequest; | ||
import javax.servlet.http.HttpServletResponse; | ||
import org.eclipse.jetty.server.Connector; | ||
import org.eclipse.jetty.server.Request; | ||
import org.eclipse.jetty.server.Server; | ||
import org.eclipse.jetty.server.handler.AbstractHandler; | ||
|
||
import static java.net.HttpURLConnection.HTTP_MOVED_TEMP; | ||
import static java.net.HttpURLConnection.HTTP_OK; | ||
|
||
/** | ||
* OAuth 2.0 verification code receiver that runs a Jetty server on a free port, waiting for a | ||
|
@@ -34,34 +38,48 @@ | |
* Implementation is thread-safe. | ||
* </p> | ||
* | ||
* @since 1.11 | ||
* @author Yaniv Inbar | ||
* @since 1.11 | ||
*/ | ||
public final class LocalServerReceiver implements VerificationCodeReceiver { | ||
|
||
private static final String LOCALHOST = "localhost"; | ||
|
||
private static final String CALLBACK_PATH = "/Callback"; | ||
|
||
/** Server or {@code null} before {@link #getRedirectUri()}. */ | ||
private Server server; | ||
/** | ||
* Server or {@code null} before {@link #getRedirectUri()}. | ||
*/ | ||
private HttpServer server; | ||
|
||
/** Verification code or {@code null} for none. */ | ||
/** | ||
* Verification code or {@code null} for none. | ||
*/ | ||
String code; | ||
|
||
/** Error code or {@code null} for none. */ | ||
/** | ||
* Error code or {@code null} for none. | ||
*/ | ||
String error; | ||
|
||
/** To block until receiving an authorization response or stop() is called. */ | ||
/** | ||
* To block until receiving an authorization response or stop() is called. | ||
*/ | ||
final Semaphore waitUnlessSignaled = new Semaphore(0 /* initially zero permit */); | ||
|
||
/** Port to use or {@code -1} to select an unused port in {@link #getRedirectUri()}. */ | ||
/** | ||
* Port to use or {@code -1} to select an unused port in {@link #getRedirectUri()}. | ||
*/ | ||
private int port; | ||
|
||
/** Host name to use. */ | ||
/** | ||
* Host name to use. | ||
*/ | ||
private final String host; | ||
|
||
/** Callback path of redirect_uri */ | ||
/** | ||
* Callback path of redirect_uri. | ||
*/ | ||
private final String callbackPath; | ||
|
||
/** | ||
|
@@ -115,28 +133,48 @@ public LocalServerReceiver() { | |
|
||
@Override | ||
public String getRedirectUri() throws IOException { | ||
|
||
server = HttpServer.create(new InetSocketAddress(port != -1 ? port : findOpenPort()), 0); | ||
HttpContext context = server.createContext(callbackPath, new CallbackHandler()); | ||
server.setExecutor(null); | ||
/* | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. why is this code commented out? |
||
server = new Server(port != -1 ? port : 0); | ||
Connector connector = server.getConnectors()[0]; | ||
ServerConnector connector = new ServerConnector(server); | ||
connector.setHost(host); | ||
server.setConnectors(new Connector[] { connector } ); | ||
server.setHandler(new CallbackHandler()); | ||
*/ | ||
try { | ||
server.start(); | ||
port = connector.getLocalPort(); | ||
port = server.getAddress().getPort(); | ||
} catch (Exception e) { | ||
Throwables.propagateIfPossible(e); | ||
throw new IOException(e); | ||
} | ||
return "http://" + connector.getHost() + ":" + port + callbackPath; | ||
return "http://" + this.getHost() + ":" + port + callbackPath; | ||
} | ||
|
||
/* | ||
*Copied from Jetty findFreePort() as referenced by: https://gist.github.com/vorburger/3429822 | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. It might be the Javadoc comment that checkstyle is complaining about; I'm not sure. Try deleting it. |
||
*/ | ||
|
||
private int findOpenPort() { | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. this could use try with resources |
||
try (ServerSocket socket = new ServerSocket(0)) { | ||
socket.setReuseAddress(true); | ||
return socket.getLocalPort(); | ||
} catch(IOException e) { | ||
throw new IllegalStateException("No free TCP/IP port to start embedded HTTP Server on"); | ||
} | ||
} | ||
|
||
/** | ||
* Blocks until the server receives a login result, or the server is stopped | ||
* by {@link #stop()}, to return an authorization code. | ||
* | ||
* @return authorization code if login succeeds; may return {@code null} if the server | ||
* is stopped by {@link #stop()} | ||
* is stopped by {@link #stop()} | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. four space indent |
||
* @throws IOException if the server receives an error code (through an HTTP request | ||
* parameter {@code error}) | ||
* parameter {@code error}) | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. only four space indent, per google style |
||
*/ | ||
@Override | ||
public String waitForCode() throws IOException { | ||
|
@@ -152,7 +190,7 @@ public void stop() throws IOException { | |
waitUnlessSignaled.release(); | ||
if (server != null) { | ||
try { | ||
server.stop(); | ||
server.stop(0); | ||
} catch (Exception e) { | ||
Throwables.propagateIfPossible(e); | ||
throw new IOException(e); | ||
|
@@ -161,7 +199,9 @@ public void stop() throws IOException { | |
} | ||
} | ||
|
||
/** Returns the host name to use. */ | ||
/** | ||
* Returns the host name to use. | ||
*/ | ||
public String getHost() { | ||
return host; | ||
} | ||
|
@@ -189,51 +229,69 @@ public String getCallbackPath() { | |
*/ | ||
public static final class Builder { | ||
|
||
/** Host name to use. */ | ||
/** | ||
* Host name to use. | ||
*/ | ||
private String host = LOCALHOST; | ||
|
||
/** Port to use or {@code -1} to select an unused port. */ | ||
/** | ||
* Port to use or {@code -1} to select an unused port. | ||
*/ | ||
private int port = -1; | ||
|
||
private String successLandingPageUrl; | ||
private String failureLandingPageUrl; | ||
|
||
private String callbackPath = CALLBACK_PATH; | ||
|
||
/** Builds the {@link LocalServerReceiver}. */ | ||
/** | ||
* Builds the {@link LocalServerReceiver}. | ||
*/ | ||
public LocalServerReceiver build() { | ||
return new LocalServerReceiver(host, port, callbackPath, | ||
successLandingPageUrl, failureLandingPageUrl); | ||
} | ||
|
||
/** Returns the host name to use. */ | ||
/** | ||
* Returns the host name to use. | ||
*/ | ||
public String getHost() { | ||
return host; | ||
} | ||
|
||
/** Sets the host name to use. */ | ||
/** | ||
* Sets the host name to use. | ||
*/ | ||
public Builder setHost(String host) { | ||
this.host = host; | ||
return this; | ||
} | ||
|
||
/** Returns the port to use or {@code -1} to select an unused port. */ | ||
/** | ||
* Returns the port to use or {@code -1} to select an unused port. | ||
*/ | ||
public int getPort() { | ||
return port; | ||
} | ||
|
||
/** Sets the port to use or {@code -1} to select an unused port. */ | ||
/** | ||
* Sets the port to use or {@code -1} to select an unused port. | ||
*/ | ||
public Builder setPort(int port) { | ||
this.port = port; | ||
return this; | ||
} | ||
|
||
/** Returns the callback path of redirect_uri */ | ||
/** | ||
* Returns the callback path of redirect_uri. | ||
*/ | ||
public String getCallbackPath() { | ||
return callbackPath; | ||
} | ||
|
||
/** Set the callback path of redirect_uri */ | ||
/** | ||
* Set the callback path of redirect_uri. | ||
*/ | ||
public Builder setCallbackPath(String callbackPath) { | ||
this.callbackPath = callbackPath; | ||
return this; | ||
|
@@ -247,51 +305,74 @@ public Builder setLandingPages(String successLandingPageUrl, String failureLandi | |
} | ||
|
||
/** | ||
* Jetty handler that takes the verifier token passed over from the OAuth provider and stashes it | ||
* HttpServer handler that takes the verifier token passed over | ||
* from the OAuth provider and stashes it | ||
* where {@link #waitForCode} will find it. | ||
*/ | ||
class CallbackHandler extends AbstractHandler { | ||
class CallbackHandler implements HttpHandler { | ||
|
||
@Override | ||
public void handle( | ||
String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response | ||
) | ||
throws IOException { | ||
if (!callbackPath.equals(target)) { | ||
public void handle(HttpExchange httpExchange) throws IOException { | ||
|
||
if (!callbackPath.equals(httpExchange.getRequestURI().getPath())) { | ||
return; | ||
} | ||
|
||
StringBuilder body = new StringBuilder(); | ||
|
||
try { | ||
((Request) request).setHandled(true); | ||
error = request.getParameter("error"); | ||
code = request.getParameter("code"); | ||
Map<String, String> parms = | ||
this.queryToMap(httpExchange.getRequestURI().getQuery()); | ||
error = parms.get("error"); | ||
code = parms.get("code"); | ||
|
||
Headers respHeaders = httpExchange.getResponseHeaders(); | ||
if (error == null && successLandingPageUrl != null) { | ||
response.sendRedirect(successLandingPageUrl); | ||
respHeaders.add("Location", successLandingPageUrl); | ||
httpExchange.sendResponseHeaders(HTTP_MOVED_TEMP, -1); | ||
} else if (error != null && failureLandingPageUrl != null) { | ||
response.sendRedirect(failureLandingPageUrl); | ||
respHeaders.add("Location", failureLandingPageUrl); | ||
httpExchange.sendResponseHeaders(HTTP_MOVED_TEMP, -1); | ||
} else { | ||
writeLandingHtml(response); | ||
writeLandingHtml(httpExchange, respHeaders); | ||
} | ||
response.flushBuffer(); | ||
} | ||
finally { | ||
httpExchange.close(); | ||
} finally { | ||
waitUnlessSignaled.release(); | ||
} | ||
} | ||
|
||
private void writeLandingHtml(HttpServletResponse response) throws IOException { | ||
response.setStatus(HttpServletResponse.SC_OK); | ||
response.setContentType("text/html"); | ||
private Map<String, String> queryToMap(String query) { | ||
Map<String, String> result = new HashMap<String, String>(); | ||
if (query != null) { | ||
for (String param : query.split("&")) { | ||
String pair[] = param.split("="); | ||
if (pair.length > 1) { | ||
result.put(pair[0], pair[1]); | ||
} else { | ||
result.put(pair[0], ""); | ||
} | ||
} | ||
} | ||
return result; | ||
} | ||
|
||
private void writeLandingHtml(HttpExchange exchange, Headers headers) throws IOException { | ||
OutputStream os = exchange.getResponseBody(); | ||
exchange.sendResponseHeaders(HTTP_OK, 0); | ||
headers.add("ContentType", "text/html"); | ||
|
||
PrintWriter doc = response.getWriter(); | ||
PrintWriter doc = new PrintWriter(os); | ||
doc.println("<html>"); | ||
doc.println("<head><title>OAuth 2.0 Authentication Token Received</title></head>"); | ||
doc.println("<body>"); | ||
doc.println("Received verification code. You may now close this window."); | ||
doc.println("</body>"); | ||
doc.println("</html>"); | ||
doc.flush(); | ||
os.close(); | ||
} | ||
|
||
} | ||
|
||
} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Google style is to never use wildcard imports.