mirror of
https://github.com/github/codeql.git
synced 2026-05-20 06:07:07 +02:00
35 lines
995 B
Java
35 lines
995 B
Java
import java.sql.*;
|
|
import java.net.*;
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
class Test {
|
|
private String byteToString(byte[] data) {
|
|
return new String(data, StandardCharsets.UTF_8);
|
|
}
|
|
|
|
public void M1(Statement handle) throws Exception {
|
|
// Only a source if "remote" is a selected threat model
|
|
Socket sock = new Socket("localhost", 1234);
|
|
byte[] data = new byte[1024];
|
|
sock.getInputStream().read(data);
|
|
|
|
// Sink
|
|
sock.getOutputStream().write(data);
|
|
|
|
// Sink
|
|
handle.executeUpdate("INSERT INTO foo VALUES ('" + byteToString(data) + "')");
|
|
}
|
|
|
|
public void M2(Statement handle) throws Exception {
|
|
// Only a source if "sql" is a selected threat model
|
|
ResultSet rs = handle.executeQuery("SELECT * FROM foo");
|
|
|
|
// Sink
|
|
handle.executeUpdate("INSERT INTO foo VALUES ('" + rs.getString("name") + "')");
|
|
|
|
// Sink
|
|
Socket sock = new Socket("localhost", 1234);
|
|
sock.getOutputStream().write(rs.getString("name").getBytes());
|
|
}
|
|
}
|