Lanmikeman/Launcher
1
1
Fork 0
mirror of https://github.com/GravitLauncher/Launcher synced 2024-12-22 16:41:46 +03:00
Code Issues Projects Releases Packages Wiki Activity

Сохранение данных в сессии

Browse source
This commit is contained in:
Gravit 2018-10-01 16:42:42 +07:00
parent 6f16819cad
commit bf061bfaf2
6 changed files with 25 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/WebSocketFrameHandler.java
View file

@ -2,36 +2,42 @@
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketFrame;
import io.netty.util.AttributeKey;
import io.netty.util.AttributeMap;
import io.netty.util.concurrent.GlobalEventExecutor;
import ru.gravit.launchserver.LaunchServer;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.utils.helper.IOHelper;
import ru.gravit.utils.helper.LogHelper;
public class WebSocketFrameHandler extends SimpleChannelInboundHandler<WebSocketFrame> {
public static final ChannelGroup channels = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
public static LaunchServer server;
public static GsonBuilder builder = new GsonBuilder();
public static WebSocketService service = new WebSocketService(LaunchServer.server,builder);
private Client client;
static {
service.registerResponses();
}
@Override
public void channelActive(ChannelHandlerContext ctx) {
LogHelper.debug("New client %s", IOHelper.getIP(ctx.channel().remoteAddress()));
client = new Client(0);
channels.add(ctx.channel());
}
@Override
protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception {
// ping and pong frames already handled
if (frame instanceof TextWebSocketFrame) {
service.process(ctx, (TextWebSocketFrame) frame);
service.process(ctx, (TextWebSocketFrame) frame, client);
} else {
String message = "unsupported frame type: " + frame.getClass().getName();
throw new UnsupportedOperationException(message);

5
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/WebSocketService.java
View file

@ -5,6 +5,7 @@
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import ru.gravit.launchserver.LaunchServer;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.launchserver.socket.websocket.json.EchoResponse;
import ru.gravit.launchserver.socket.websocket.json.JsonResponseAdapter;
import ru.gravit.launchserver.socket.websocket.json.JsonResponseInterface;
@ -26,12 +27,12 @@ public WebSocketService(LaunchServer server, GsonBuilder gson) {
private final Gson gson;
private final GsonBuilder gsonBuiler;
void process(ChannelHandlerContext ctx, TextWebSocketFrame frame)
void process(ChannelHandlerContext ctx, TextWebSocketFrame frame, Client client)
{
String request = frame.text();
JsonResponseInterface response = gson.fromJson(request, JsonResponseInterface.class);
try {
response.execute(this,ctx);
response.execute(this,ctx,client);
} catch (Exception e)
{
LogHelper.error(e);

5
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/json/EchoResponse.java
View file

@ -1,6 +1,7 @@
package ru.gravit.launchserver.socket.websocket.json;
import io.netty.channel.ChannelHandlerContext;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.launchserver.socket.websocket.WebSocketService;
import ru.gravit.utils.helper.LogHelper;
@ -17,8 +18,8 @@ public String getType() {
}
@Override
public void execute(WebSocketService service,ChannelHandlerContext ctx) {
LogHelper.info("Echo: %s",echo);
public void execute(WebSocketService service,ChannelHandlerContext ctx, Client client) {
LogHelper.info("Echo: %s, isAuth %s",echo,client.isAuth ? "true" : "false");
service.sendObject(ctx,new Result(echo));
}
public class Result

3
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/json/JsonResponseInterface.java
View file

@ -2,9 +2,10 @@
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.WebSocketFrame;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.launchserver.socket.websocket.WebSocketService;
public interface JsonResponseInterface {
String getType();
void execute(WebSocketService service,ChannelHandlerContext ctx) throws Exception;
void execute(WebSocketService service,ChannelHandlerContext ctx, Client client) throws Exception;
}

3
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/json/SimpleResponse.java
View file

@ -1,6 +1,7 @@
package ru.gravit.launchserver.socket.websocket.json;
import io.netty.channel.ChannelHandlerContext;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.launchserver.socket.websocket.WebSocketService;
public class SimpleResponse implements JsonResponseInterface {
@ -10,7 +11,7 @@ public String getType() {
}
@Override
public void execute(WebSocketService service, ChannelHandlerContext ctx) throws Exception {
public void execute(WebSocketService service, ChannelHandlerContext ctx, Client client) throws Exception {
}
}

12
LaunchServer/src/main/java/ru/gravit/launchserver/socket/websocket/json/auth/AuthResponse.java
View file

@ -9,6 +9,7 @@
import ru.gravit.launchserver.auth.hwid.HWIDException;
import ru.gravit.launchserver.auth.provider.AuthProvider;
import ru.gravit.launchserver.auth.provider.AuthProviderResult;
import ru.gravit.launchserver.socket.Client;
import ru.gravit.launchserver.socket.websocket.WebSocketService;
import ru.gravit.launchserver.socket.websocket.json.JsonResponseInterface;
import ru.gravit.utils.helper.IOHelper;
@ -37,7 +38,7 @@ public String getType() {
}
@Override
public void execute(WebSocketService service, ChannelHandlerContext ctx) throws Exception {
public void execute(WebSocketService service, ChannelHandlerContext ctx, Client clientData) throws Exception {
try {
String ip = IOHelper.getIP(ctx.channel().remoteAddress());
if (LaunchServer.server.limiter.isLimit(ip)) {
@ -56,13 +57,14 @@ public void execute(WebSocketService service, ChannelHandlerContext ctx) throws
if (!p.object.isWhitelistContains(login)) {
throw new AuthException(LaunchServer.server.config.whitelistRejectString);
}
//clientData.profile = p.object;
clientData.profile = p.object;
}
}
//if(clientData.profile == null) {
// throw new AuthException("You profile not found");
//}
if(clientData.profile == null) {
throw new AuthException("You profile not found");
}
LaunchServer.server.config.hwidHandler.check(hwid, result.username);
clientData.isAuth = true;
service.sendObject(ctx,new WebSocketService.SuccessResult("auth"));
} catch (AuthException | HWIDException e)
{