1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package io.netty.handler.codec.socks;
17
18 import io.netty.buffer.ByteBuf;
19 import io.netty.channel.ChannelHandlerContext;
20 import io.netty.handler.codec.ReplayingDecoder;
21 import io.netty.handler.codec.socks.SocksInitResponseDecoder.State;
22
23 import java.util.List;
24
25
26
27
28
29 public class SocksInitResponseDecoder extends ReplayingDecoder<State> {
30 private static final String name = "SOCKS_INIT_RESPONSE_DECODER";
31
32
33
34
35 @Deprecated
36 public static String getName() {
37 return name;
38 }
39
40 public SocksInitResponseDecoder() {
41 super(State.CHECK_PROTOCOL_VERSION);
42 }
43
44 @Override
45 protected void decode(ChannelHandlerContext ctx, ByteBuf byteBuf, List<Object> out) throws Exception {
46 switch (state()) {
47 case CHECK_PROTOCOL_VERSION: {
48 if (byteBuf.readByte() != SocksProtocolVersion.SOCKS5.byteValue()) {
49 out.add(SocksCommonUtils.UNKNOWN_SOCKS_RESPONSE);
50 break;
51 }
52 checkpoint(State.READ_PREFERRED_AUTH_TYPE);
53 }
54 case READ_PREFERRED_AUTH_TYPE: {
55 SocksAuthScheme authScheme = SocksAuthScheme.valueOf(byteBuf.readByte());
56 out.add(new SocksInitResponse(authScheme));
57 break;
58 }
59 default: {
60 throw new Error();
61 }
62 }
63 ctx.pipeline().remove(this);
64 }
65
66 enum State {
67 CHECK_PROTOCOL_VERSION,
68 READ_PREFERRED_AUTH_TYPE
69 }
70 }