diff --git a/src/main/java/com/corundumstudio/socketio/Configuration.java b/src/main/java/com/corundumstudio/socketio/Configuration.java index 601f2207..b68fc84d 100644 --- a/src/main/java/com/corundumstudio/socketio/Configuration.java +++ b/src/main/java/com/corundumstudio/socketio/Configuration.java @@ -15,10 +15,6 @@ */ package com.corundumstudio.socketio; -import java.io.InputStream; -import java.util.Arrays; -import java.util.List; - import com.corundumstudio.socketio.handler.SuccessAuthorizationListener; import com.corundumstudio.socketio.listener.DefaultExceptionListener; import com.corundumstudio.socketio.listener.ExceptionListener; @@ -27,6 +23,9 @@ import com.corundumstudio.socketio.store.StoreFactory; import javax.net.ssl.KeyManagerFactory; +import java.io.InputStream; +import java.util.Arrays; +import java.util.List; public class Configuration { @@ -84,6 +83,8 @@ public class Configuration { private String origin; + private boolean enableCors = true; + private boolean httpCompression = true; private boolean websocketCompression = true; @@ -154,6 +155,7 @@ public Configuration() { setAddVersionHeader(conf.isAddVersionHeader()); setOrigin(conf.getOrigin()); + setEnableCors(conf.isEnableCors()); setAllowHeaders(conf.getAllowHeaders()); setSSLProtocol(conf.getSSLProtocol()); @@ -517,13 +519,30 @@ public boolean isAddVersionHeader() { public void setOrigin(String origin) { this.origin = origin; } + public String getOrigin() { return origin; } + /** + * cors dispose + *

+ * Default is true + * + * @param enableCors enableCors + */ + public void setEnableCors(boolean enableCors) { + this.enableCors = enableCors; + } + + public boolean isEnableCors() { + return enableCors; + } + public boolean isUseLinuxNativeEpoll() { return useLinuxNativeEpoll; } + public void setUseLinuxNativeEpoll(boolean useLinuxNativeEpoll) { this.useLinuxNativeEpoll = useLinuxNativeEpoll; } diff --git a/src/main/java/com/corundumstudio/socketio/handler/EncoderHandler.java b/src/main/java/com/corundumstudio/socketio/handler/EncoderHandler.java index 53b9abbc..4d538a71 100644 --- a/src/main/java/com/corundumstudio/socketio/handler/EncoderHandler.java +++ b/src/main/java/com/corundumstudio/socketio/handler/EncoderHandler.java @@ -116,8 +116,8 @@ private void write(XHROptionsMessage msg, ChannelHandlerContext ctx, ChannelProm HttpResponse res = new DefaultHttpResponse(HTTP_1_1, HttpResponseStatus.OK); res.headers().add(HttpHeaderNames.SET_COOKIE, "io=" + msg.getSessionId()) - .add(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE) - .add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_HEADERS, HttpHeaderNames.CONTENT_TYPE); + .add(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE) + .add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_HEADERS, HttpHeaderNames.CONTENT_TYPE); String origin = ctx.channel().attr(ORIGIN).get(); addOriginHeaders(origin, res); @@ -136,7 +136,7 @@ private void sendMessage(HttpMessage msg, Channel channel, ByteBuf out, String t HttpResponse res = new DefaultHttpResponse(HTTP_1_1, status); res.headers().add(HttpHeaderNames.CONTENT_TYPE, type) - .add(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE); + .add(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE); if (msg.getSessionId() != null) { res.headers().add(HttpHeaderNames.SET_COOKIE, "io=" + msg.getSessionId()); } @@ -188,19 +188,21 @@ private void addOriginHeaders(String origin, HttpResponse res) { res.headers().add(HttpHeaderNames.SERVER, version); } - if (configuration.getOrigin() != null) { - res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, configuration.getOrigin()); - res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_CREDENTIALS, Boolean.TRUE); - } else { - if (origin != null) { - res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, origin); + if (configuration.isEnableCors()) { + if (configuration.getOrigin() != null) { + res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, configuration.getOrigin()); res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_CREDENTIALS, Boolean.TRUE); } else { - res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, "*"); + if (origin != null) { + res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, origin); + res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_CREDENTIALS, Boolean.TRUE); + } else { + res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_ORIGIN, "*"); + } + } + if (configuration.getAllowHeaders() != null) { + res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_HEADERS, configuration.getAllowHeaders()); } - } - if(configuration.getAllowHeaders() != null){ - res.headers().add(HttpHeaderNames.ACCESS_CONTROL_ALLOW_HEADERS, configuration.getAllowHeaders()); } }