paxoscn / mysql-protocol Goto Github PK

public class YourSqlEngine implements SqlEngine {
}

@Override
public void query(ResultSetWriter resultSetWriter, String database, String userName, byte[] scramble411, byte[] authSeed, String sql) throws IOException {
    // 在结果集中放入一列，名为'col1'
    // 该调用必须在任意一次writeRow()之前
    resultSetWriter.writeColumns(List.of(new QueryResultColumn("col1", "varchar(255)")));

    // 在结果集中放入一行，值为'Hello World !'
    resultSetWriter.writeRow(List.of("Hello World !"));

    // 最后需要调用finish()方法结束响应
    resultSetWriter.finish();
}

@Override
public void authenticate(String database, String userName, byte[] scramble411, byte[] authSeed) throws IOException {
    // 只允许用户名'github'
    String validUser = "github";
    if (!userName.equals(validUser)) {
        throw new IOException(new IllegalAccessException("Authentication failed: User " + userName + " is not allowed to connect"));
    }
    
    // 只允许密码'123456'
    String validPassword = "123456";
    
    // 对密码进行SHA1及编码
    String validPasswordSha1 = SHAUtils.SHA(validPassword, SHAUtils.SHA_1);
    String validScramble411WithSeed20 = Utils.scramble411(validPasswordSha1, authSeed);
    
    // 比较密码编码结果
    if (!Utils.compareDigest(validScramble411WithSeed20, Base64.getEncoder().encodeToString(scramble411))) {
        throw new IOException(new IllegalAccessException("Authentication failed: Validation failed"));
    }
}

// 启动并监听端口3307
new MySqlListener(3307, 100, new YourSqlEngine());

// 为了有足够时间来进行客户端连接，可以在这里sleep一段时间.
Thread.sleep(1000L * 60 * 10);