package org.cacert.gigi.ping;
-import java.io.FileReader;
-import java.io.IOException;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
-import java.sql.SQLException;
+import java.security.KeyStore;
import java.util.HashMap;
-import java.util.Properties;
-import org.cacert.gigi.Domain;
-import org.cacert.gigi.User;
import org.cacert.gigi.database.DatabaseConnection;
+import org.cacert.gigi.database.GigiPreparedStatement;
+import org.cacert.gigi.database.GigiResultSet;
+import org.cacert.gigi.dbObjects.Domain;
+import org.cacert.gigi.dbObjects.User;
import org.cacert.gigi.util.RandomToken;
public class PingerDaemon extends Thread {
HashMap<String, DomainPinger> pingers = new HashMap<>();
- private PreparedStatement searchNeededPings;
+ private GigiPreparedStatement searchNeededPings;
- private PreparedStatement enterPingResult;
+ private GigiPreparedStatement enterPingResult;
+
+ private GigiPreparedStatement updatePingStatus;
+
+ private KeyStore truststore;
+
+ public PingerDaemon(KeyStore truststore) {
+ this.truststore = truststore;
+ }
@Override
public void run() {
- try {
- searchNeededPings = DatabaseConnection.getInstance().prepare("SELECT pingconfig.*, domains.domain, domains.memid FROM pingconfig LEFT JOIN domainPinglog ON domainPinglog.configId=pingconfig.id INNER JOIN domains ON domains.id=pingconfig.domainid WHERE domainPinglog.configId IS NULL ");
- enterPingResult = DatabaseConnection.getInstance().prepare("INSERT INTO domainPinglog SET configId=?, state=?, result=?, challenge=?");
- pingers.put("email", new EmailPinger());
- pingers.put("ssl", new SSLPinger());
- pingers.put("http", new HTTPFetch());
- pingers.put("dns", new DNSPinger());
- } catch (SQLException e) {
- e.printStackTrace();
- }
+ searchNeededPings = DatabaseConnection.getInstance().prepare("SELECT pingconfig.*, domains.domain, domains.memid FROM pingconfig LEFT JOIN domainPinglog ON domainPinglog.configId=pingconfig.id INNER JOIN domains ON domains.id=pingconfig.domainid WHERE ( pingconfig.reping='y' OR domainPinglog.configId IS NULL) AND domains.deleted IS NULL GROUP BY pingconfig.id");
+ enterPingResult = DatabaseConnection.getInstance().prepare("INSERT INTO domainPinglog SET configId=?, state=?, result=?, challenge=?");
+ updatePingStatus = DatabaseConnection.getInstance().prepare("UPDATE pingconfig SET reping='n' WHERE id=?");
+ pingers.put("email", new EmailPinger());
+ pingers.put("ssl", new SSLPinger(truststore));
+ pingers.put("http", new HTTPFetch());
+ pingers.put("dns", new DNSPinger());
+
while (true) {
- try {
- execute();
- } catch (SQLException e) {
- e.printStackTrace();
- }
+ execute();
try {
Thread.sleep(5000);
} catch (InterruptedException e) {
- e.printStackTrace();
}
}
}
- private void execute() throws SQLException {
+ private void execute() {
- ResultSet rs = searchNeededPings.executeQuery();
+ GigiResultSet rs = searchNeededPings.executeQuery();
while (rs.next()) {
String type = rs.getString("type");
String config = rs.getString("info");
token = RandomToken.generateToken(16);
config = config + ":" + token;
}
+ updatePingStatus.setInt(1, rs.getInt("id"));
+ updatePingStatus.execute();
enterPingResult.setInt(1, rs.getInt("id"));
String resp = dp.ping(Domain.getById(rs.getInt("domainid")), config, User.getById(rs.getInt("memid")));
enterPingResult.setString(2, DomainPinger.PING_STILL_PENDING == resp ? "open" : DomainPinger.PING_SUCCEDED.equals(resp) ? "success" : "failed");
}
}
- public static void main(String[] args) throws IOException {
- Properties conf = new Properties();
- conf.load(new FileReader("config/gigi.properties"));
- DatabaseConnection.init(conf);
- new PingerDaemon().run();
-
- }
}