com.deathbycaptcha.SocketClient Maven / Gradle / Ivy
The newest version!
package com.deathbycaptcha;
import org.base64.Base64;
import org.json.JSONObject;
import org.json.JSONException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.Date;
import java.util.Iterator;
import java.util.Random;
/**
* Death by Captcha socket API client.
*
*/
public class SocketClient extends Client
{
final static public String HOST = "api.dbcapi.me";
final static public int FIRST_PORT = 8123;
final static public int LAST_PORT = 8130;
final static public String TERMINATOR = "\r\n";
protected SocketChannel channel = null;
protected Object callLock = new Object();
protected String sendAndReceive(byte[] payload)
throws IOException
{
ByteBuffer sbuf = ByteBuffer.wrap(payload);
ByteBuffer rbuf = ByteBuffer.allocateDirect(256);
CharsetDecoder rbufDecoder = Charset.forName("UTF-8").newDecoder();
StringBuilder response = new StringBuilder();
int ops = SelectionKey.OP_WRITE | SelectionKey.OP_READ;
if (this.channel.isConnectionPending()) {
ops = ops | SelectionKey.OP_CONNECT;
}
Selector selector = Selector.open();
try {
this.channel.register(selector, ops);
while (true) {
if (0 < selector.select(Client.POLLS_INTERVAL * 1000)) {
Iterator keys = selector.selectedKeys().iterator();
while (keys.hasNext()) {
SelectionKey key = (SelectionKey)keys.next();
SocketChannel ch = (SocketChannel)key.channel();
if (key.isConnectable()) {
// Just connected
ch.finishConnect();
}
if (key.isReadable() && !sbuf.hasRemaining()) {
// Receiving the response
while (0 < ch.read(rbuf)) {
rbuf.flip();
response.append(rbufDecoder.decode(rbuf).toString());
}
if (2 <= response.length() && response.substring(response.length() - 2, response.length()).equals(SocketClient.TERMINATOR)) {
response.setLength(response.length() - 2);
return response.toString();
} else if (0 == response.length()) {
throw new IOException("Connection lost");
}
}
if (key.isWritable() && sbuf.hasRemaining()) {
// Sending the request
while (0 < ch.write(sbuf) && sbuf.hasRemaining()) {
//
}
}
keys.remove();
}
}
}
} catch (java.lang.Exception e) {
throw new IOException("API communication failed: " + e.toString());
} finally {
selector.close();
}
}
/**
* @see com.deathbycaptcha.Client#close
*/
public void close()
{
if (null != this.channel) {
this.log("CLOSE");
if (this.channel.isConnected() || this.channel.isConnectionPending()) {
try {
this.channel.socket().shutdownOutput();
this.channel.socket().shutdownInput();
} catch (java.lang.Exception e) {
//
} finally {
try {
this.channel.close();
} catch (java.lang.Exception e) {
//
}
}
}
try {
this.channel.socket().close();
} catch (java.lang.Exception e) {
//
}
this.channel = null;
}
}
/**
* @see com.deathbycaptcha.Client#connect
*/
public boolean connect()
throws IOException
{
if (null == this.channel) {
this.log("OPEN");
InetAddress host = null;
try {
host = InetAddress.getByName(SocketClient.HOST);
} catch (java.lang.Exception e) {
//System.out.println(e)
throw new IOException("API host not found");
}
SocketChannel channel = SocketChannel.open();
channel.configureBlocking(false);
try {
channel.connect(new InetSocketAddress(
host,
SocketClient.FIRST_PORT + new Random().nextInt(
SocketClient.LAST_PORT - SocketClient.FIRST_PORT + 1
)
));
} catch (IOException e) {
throw new IOException("API connection failed");
}
this.channel = channel;
}
return null != this.channel;
}
protected JSONObject call(String cmd, JSONObject args)
throws IOException, com.deathbycaptcha.Exception
{
try {
args.put("cmd", cmd).put("version", Client.API_VERSION);
} catch (JSONException e) {
//System.out.println(e);
return new JSONObject();
}
int attempts = 2;
byte[] payload = (args.toString() + SocketClient.TERMINATOR).getBytes();
JSONObject response = null;
while (0 < attempts && null == response) {
attempts--;
if (null == this.channel && !cmd.equals("login")) {
this.call("login", this.getCredentials());
}
synchronized (this.callLock) {
if (this.connect()) {
this.log("SEND", args.toString());
try {
response = new JSONObject(this.sendAndReceive(payload));
} catch (java.lang.Exception e) {
//System.out.println("SocketClient.call(): " + e.toString());
this.close();
}
}
}
}
if (null == response) {
throw new IOException("API connection lost or timed out");
}
this.log("RECV", response.toString());
String error = response.optString("error", null);
if (null != error) {
synchronized (this.callLock) {
this.close();
}
if (error.equals("not-logged-in")) {
throw new AccessDeniedException("Access denied, check your credentials");
} else if (error.equals("banned")) {
throw new AccessDeniedException("Access denied, account is suspended");
} else if (error.equals("insufficient-funds")) {
throw new AccessDeniedException("Access denied, balance is too low");
} else if (error.equals("invalid-captcha")) {
throw new InvalidCaptchaException("CAPTCHA was rejected by the service, check if it's a valid image");
} else if (error.equals("service-overload")) {
throw new ServiceOverloadException("CAPTCHA was rejected due to service overload, try again later");
} else {
throw new IOException("API server error occured: " + error);
}
} else {
return response;
}
}
{
}
protected JSONObject call(String cmd)
throws IOException, com.deathbycaptcha.Exception
{
return this.call(cmd, new JSONObject());
}
/**
* @see com.deathbycaptcha.Client#Client(String, String)
*/
public SocketClient(String username, String password)
{
super(username, password);
}
public void finalize()
{
this.close();
}
/**
* @see com.deathbycaptcha.Client#getUser
*/
public User getUser()
throws IOException, com.deathbycaptcha.Exception
{
return new User(this.call("user"));
}
/**
* @see com.deathbycaptcha.Client#upload
*/
public Captcha upload(byte[] img)
throws IOException, com.deathbycaptcha.Exception
{
JSONObject args = new JSONObject();
try {
args.put("captcha",
Base64.encodeBytes(img)).put("swid",
Client.SOFTWARE_VENDOR_ID);
} catch (JSONException e) {
//System.out.println(e);
}
Captcha c = new Captcha(this.call("upload", args));
return c.isUploaded() ? c : null;
}
/**
* @see com.deathbycaptcha.Client#getCaptcha
*/
public Captcha getCaptcha(int id)
throws IOException, com.deathbycaptcha.Exception
{
JSONObject args = new JSONObject();
try {
args.put("captcha", id);
} catch (JSONException e) {
//System.out.println(e);
}
return new Captcha(this.call("captcha", args));
}
/**
* @see com.deathbycaptcha.Client#report
*/
public boolean report(int id)
throws IOException, com.deathbycaptcha.Exception
{
JSONObject args = new JSONObject();
try {
args.put("captcha", id);
} catch (JSONException e) {
//System.out.println(e);
}
return !(new Captcha(this.call("report", args))).isCorrect();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy