Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
deps: replace Jetty with HttpServer (#433)
* feat: Replace Jetty with HttpServer.

Remove Jetty v8.2 and replace with com.sun.net.httpserver.HttpServer.  Jetty has had breaking changes since 8.2 (specifically with version 9.4).  This causes with conflicts with newer code using later versions of Jetty.  Removing dependency on Jetty resolves this issue.

Fixes #397.

* feat: Rewrite findOpenSocket using try with resources

* fix:  Remove commented out Jetty 9.4 code since we are using HttpServer now.

* fix:  Enforce Google coding style.
  • Loading branch information
ericraskin committed Feb 20, 2020
1 parent 303e1a1 commit bcabce2
Show file tree
Hide file tree
Showing 3 changed files with 144 additions and 73 deletions.
4 changes: 0 additions & 4 deletions google-oauth-client-jetty/pom.xml
Expand Up @@ -86,10 +86,6 @@
<groupId>com.google.oauth-client</groupId>
<artifactId>google-oauth-client-java6</artifactId>
</dependency>
<dependency>
<groupId>org.eclipse.jetty</groupId>
<artifactId>jetty-server</artifactId>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
Expand Down
Expand Up @@ -14,17 +14,24 @@

package com.google.api.client.extensions.jetty.auth.oauth2;

import static java.net.HttpURLConnection.HTTP_MOVED_TEMP;
import static java.net.HttpURLConnection.HTTP_OK;

import com.google.api.client.extensions.java6.auth.oauth2.VerificationCodeReceiver;
import com.google.api.client.util.Throwables;
import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpContext;
import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.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;

/**
* OAuth 2.0 verification code receiver that runs a Jetty server on a free port, waiting for a
Expand All @@ -34,34 +41,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;

/**
Expand All @@ -71,8 +92,8 @@ public final class LocalServerReceiver implements VerificationCodeReceiver {
private String successLandingPageUrl;

/**
* URL to an HTML page to be shown (via redirect) after failed login. If null, a canned
* default landing page will be shown (via direct response).
* URL to an HTML page to be shown (via redirect) after failed login. If null, a canned default
* landing page will be shown (via direct response).
*/
private String failureLandingPageUrl;

Expand All @@ -94,7 +115,7 @@ public LocalServerReceiver() {
* @param port Port to use or {@code -1} to select an unused port
*/
LocalServerReceiver(String host, int port,
String successLandingPageUrl, String failureLandingPageUrl) {
String successLandingPageUrl, String failureLandingPageUrl) {
this(host, port, CALLBACK_PATH, successLandingPageUrl, failureLandingPageUrl);
}

Expand All @@ -105,7 +126,7 @@ public LocalServerReceiver() {
* @param port Port to use or {@code -1} to select an unused port
*/
LocalServerReceiver(String host, int port, String callbackPath,
String successLandingPageUrl, String failureLandingPageUrl) {
String successLandingPageUrl, String failureLandingPageUrl) {
this.host = host;
this.port = port;
this.callbackPath = callbackPath;
Expand All @@ -115,28 +136,42 @@ public LocalServerReceiver() {

@Override
public String getRedirectUri() throws IOException {
server = new Server(port != -1 ? port : 0);
Connector connector = server.getConnectors()[0];
connector.setHost(host);
server.setHandler(new CallbackHandler());

server = HttpServer.create(new InetSocketAddress(port != -1 ? port : findOpenPort()), 0);
HttpContext context = server.createContext(callbackPath, new CallbackHandler());
server.setExecutor(null);

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
*/

private int findOpenPort() {
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.
* 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()}
* @throws IOException if the server receives an error code (through an HTTP request
* parameter {@code error})
* @return authorization code if login succeeds; may return {@code null} if the server is stopped
* by {@link #stop()}
* @throws IOException if the server receives an error code (through an HTTP request parameter
* {@code error})
*/
@Override
public String waitForCode() throws IOException {
Expand All @@ -152,7 +187,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);
Expand All @@ -161,7 +196,9 @@ public void stop() throws IOException {
}
}

/** Returns the host name to use. */
/**
* Returns the host name to use.
*/
public String getHost() {
return host;
}
Expand Down Expand Up @@ -189,51 +226,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);
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;
Expand All @@ -247,51 +302,73 @@ public Builder setLandingPages(String successLandingPageUrl, String failureLandi
}

/**
* Jetty handler that takes the verifier token passed over from the OAuth provider and stashes it
* where {@link #waitForCode} will find 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();
}

}

}

0 comments on commit bcabce2

Please sign in to comment.