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.SocksAuthRequestDecoder.State;
22 import io.netty.util.internal.UnstableApi;
23
24 import java.util.List;
25
26
27
28
29
30 public class SocksAuthRequestDecoder extends ReplayingDecoder<State> {
31
32 private String username;
33
34 public SocksAuthRequestDecoder() {
35 super(State.CHECK_PROTOCOL_VERSION);
36 }
37
38 @Override
39 protected void decode(ChannelHandlerContext ctx, ByteBuf byteBuf, List<Object> out) throws Exception {
40 switch (state()) {
41 case CHECK_PROTOCOL_VERSION: {
42 if (byteBuf.readByte() != SocksSubnegotiationVersion.AUTH_PASSWORD.byteValue()) {
43 out.add(SocksCommonUtils.UNKNOWN_SOCKS_REQUEST);
44 break;
45 }
46 checkpoint(State.READ_USERNAME);
47 }
48 case READ_USERNAME: {
49 int fieldLength = byteBuf.readByte();
50 username = SocksCommonUtils.readUsAscii(byteBuf, fieldLength);
51 checkpoint(State.READ_PASSWORD);
52 }
53 case READ_PASSWORD: {
54 int fieldLength = byteBuf.readByte();
55 String password = SocksCommonUtils.readUsAscii(byteBuf, fieldLength);
56 out.add(new SocksAuthRequest(username, password));
57 break;
58 }
59 default: {
60 throw new Error();
61 }
62 }
63 ctx.pipeline().remove(this);
64 }
65
66 @UnstableApi
67 public enum State {
68 CHECK_PROTOCOL_VERSION,
69 READ_USERNAME,
70 READ_PASSWORD
71 }
72 }