321 lines
8.8 KiB
C
321 lines
8.8 KiB
C
#include <hio.h>
|
|
|
|
#if defined(HIO_ENABLE_MARIADB)
|
|
|
|
#include <hio-mar.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <signal.h>
|
|
|
|
#if 0
|
|
#include <mariadb/mysql.h>
|
|
#else
|
|
#include <mysql.h>
|
|
#endif
|
|
|
|
#if 0
|
|
static void mar_on_disconnect (hio_dev_mar_t* dev)
|
|
{
|
|
}
|
|
|
|
static void mar_on_connect (hio_dev_mar_t* dev)
|
|
{
|
|
printf ("CONNECTED...\n");
|
|
if (hio_dev_mar_querywithbchars(dev, "SHOW STATUS", 11) <= -1)
|
|
{
|
|
hio_dev_mar_halt (dev);
|
|
}
|
|
}
|
|
|
|
static void mar_on_query_started (hio_dev_mar_t* dev, int mar_ret, const hio_bch_t* mar_errmsg)
|
|
{
|
|
if (mar_ret != 0)
|
|
{
|
|
printf ("QUERY NOT SENT PROPERLY..%s\n", mysql_error(dev->hnd));
|
|
}
|
|
else
|
|
{
|
|
printf ("QUERY SENT..\n");
|
|
if (hio_dev_mar_fetchrows(dev) <= -1)
|
|
{
|
|
printf ("FETCH ROW FAILURE - %s\n", errmsg);
|
|
hio_dev_mar_halt (dev);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void mar_on_row_fetched (hio_dev_mar_t* dev, void* data)
|
|
{
|
|
MYSQL_ROW row = (MYSQL_ROW)data;
|
|
static int x = 0;
|
|
if (!row)
|
|
{
|
|
printf ("NO MORE ROW..\n");
|
|
if (x == 0 && hio_dev_mar_querywithbchars(dev, "SELECT * FROM pdns.records", 26) <= -1) hio_dev_mar_halt (dev);
|
|
x++;
|
|
}
|
|
else
|
|
{
|
|
if (x == 0)
|
|
printf ("%s %s\n", row[0], row[1]);
|
|
else if (x == 1)
|
|
printf ("%s %s %s %s %s\n", row[0], row[1], row[2], row[3], row[4]);
|
|
//printf ("GOT ROW\n");
|
|
}
|
|
}
|
|
|
|
int main (int argc, char* argv[])
|
|
{
|
|
|
|
hio_t* hio = HIO_NULL;
|
|
hio_dev_mar_t* mar;
|
|
hio_dev_mar_make_t mi;
|
|
hio_dev_mar_connect_t ci;
|
|
|
|
if (argc != 6)
|
|
{
|
|
fprintf (stderr, "Usage: %s ipaddr port username password dbname\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
hio = hio_open(HIO_NULL, 0, HIO_NULL, HIO_FEATURE_ALL, 512, HIO_NULL);
|
|
if (!hio)
|
|
{
|
|
printf ("Cannot open hio\n");
|
|
goto oops;
|
|
}
|
|
|
|
memset (&ci, 0, HIO_SIZEOF(ci));
|
|
ci.host = argv[1];
|
|
ci.port = 3306; /* TODO: argv[2]; */
|
|
ci.username = argv[3];
|
|
ci.password = argv[4];
|
|
ci.dbname = argv[5];
|
|
|
|
memset (&mi, 0, HIO_SIZEOF(mi));
|
|
/*mi.on_write = mar_on_write;
|
|
mi.on_read = mar_on_read;*/
|
|
mi.on_connect = mar_on_connect;
|
|
mi.on_disconnect = mar_on_disconnect;
|
|
mi.on_query_started = mar_on_query_started;
|
|
mi.on_row_fetched = mar_on_row_fetched;
|
|
|
|
mar = hio_dev_mar_make(hio, 0, &mi);
|
|
if (!mar)
|
|
{
|
|
printf ("Cannot make a mar db client device\n");
|
|
goto oops;
|
|
}
|
|
|
|
if (hio_dev_mar_connect(mar, &ci) <= -1)
|
|
{
|
|
printf ("Cannot connect to mar db server\n");
|
|
goto oops;
|
|
}
|
|
|
|
hio_loop (hio);
|
|
|
|
oops:
|
|
if (hio) hio_close (hio);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
static void on_result (hio_svc_marc_t* svc, hio_oow_t sid, hio_svc_marc_rcode_t rcode, void* data, void* qctx)
|
|
{
|
|
static int x = 0;
|
|
switch (rcode)
|
|
{
|
|
case HIO_SVC_MARC_RCODE_ROW:
|
|
{
|
|
MYSQL_ROW row = (MYSQL_ROW)data;
|
|
// if (x == 0)
|
|
printf ("[%lu] %s %s\n", sid, row[0], row[1]);
|
|
// else if (x == 1)
|
|
// printf ("%s %s %s %s %s\n", row[0], row[1], row[2], row[3], row[4]);
|
|
//printf ("GOT ROW\n");
|
|
#if 0
|
|
x++;
|
|
if (x == 1)
|
|
{
|
|
printf ("BLOCKING PACKET...........................\n");
|
|
system ("/sbin/iptables -I OUTPUT -p tcp --dport 3306 -j REJECT");
|
|
system ("/sbin/iptables -I INPUT -p tcp --sport 3306 -j REJECT");
|
|
}
|
|
#endif
|
|
break;
|
|
}
|
|
|
|
case HIO_SVC_MARC_RCODE_DONE:
|
|
printf ("[%lu] NO DATA..\n", sid);
|
|
break;
|
|
|
|
case HIO_SVC_MARC_RCODE_ERROR:
|
|
{
|
|
hio_svc_marc_dev_error_t* err = (hio_svc_marc_dev_error_t*)data;
|
|
printf ("QUERY ERROR - [%d] %s\n", err->mar_errcode, err->mar_errmsg);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static hio_t* g_hio = HIO_NULL;
|
|
|
|
static void handle_signal (int sig)
|
|
{
|
|
hio_stop (g_hio, HIO_STOPREQ_TERMINATION);
|
|
}
|
|
|
|
static void send_test_query (hio_t* hio, const hio_ntime_t* now, hio_tmrjob_t* job)
|
|
{
|
|
hio_svc_marc_t* marc = (hio_svc_marc_t*)job->ctx;
|
|
hio_bch_t buf[256];
|
|
hio_bch_t tmp[256];
|
|
int len;
|
|
|
|
HIO_INFO0 (hio, "ATTEMPTING TO SEND QUERY\n");
|
|
if (hio_svc_marc_querywithbchars(marc, 0, HIO_SVC_MARC_QTYPE_SELECT, "SHOW STATUS", 11, on_result, HIO_NULL) <= -1)
|
|
{
|
|
HIO_INFO1 (hio, "FAILED TO SEND QUERY - %js\n", hio_geterrmsg(hio));
|
|
}
|
|
|
|
hio_svc_marc_escapebchars (marc, "wild", 4, tmp);
|
|
len = snprintf(buf, HIO_COUNTOF(buf), "SELECT name, content FROM records WHERE name like '%%%s%%'", tmp);
|
|
if (hio_svc_marc_querywithbchars(marc, 1, HIO_SVC_MARC_QTYPE_SELECT, buf, len, on_result, HIO_NULL) <= -1)
|
|
{
|
|
HIO_INFO1 (hio, "FAILED TO SEND QUERY - %js\n", hio_geterrmsg(hio));
|
|
}
|
|
}
|
|
|
|
static int schedule_timer_job_after (hio_t* hio, const hio_ntime_t* fire_after, hio_tmrjob_handler_t handler, void* ctx)
|
|
{
|
|
hio_tmrjob_t tmrjob;
|
|
|
|
memset (&tmrjob, 0, HIO_SIZEOF(tmrjob));
|
|
tmrjob.ctx = ctx;
|
|
|
|
hio_gettime (hio, &tmrjob.when);
|
|
HIO_ADD_NTIME (&tmrjob.when, &tmrjob.when, fire_after);
|
|
|
|
tmrjob.handler = handler;
|
|
tmrjob.idxptr = HIO_NULL;
|
|
|
|
return hio_instmrjob(hio, &tmrjob);
|
|
}
|
|
|
|
|
|
int main (int argc, char* argv[])
|
|
{
|
|
|
|
hio_t* hio = HIO_NULL;
|
|
hio_svc_marc_t* marc;
|
|
hio_svc_marc_connect_t ci;
|
|
hio_svc_marc_tmout_t tmout;
|
|
|
|
if (argc != 6)
|
|
{
|
|
fprintf (stderr, "Usage: %s ipaddr port username password dbname\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
hio = hio_open(HIO_NULL, 0, HIO_NULL, HIO_FEATURE_ALL, 512, HIO_NULL);
|
|
if (!hio)
|
|
{
|
|
printf ("Cannot open hio\n");
|
|
goto oops;
|
|
}
|
|
|
|
memset (&ci, 0, HIO_SIZEOF(ci));
|
|
ci.host = argv[1];
|
|
ci.port = 3306; /* TODO: argv[2]; */
|
|
ci.username = argv[3];
|
|
ci.password = argv[4];
|
|
ci.dbname = argv[5];
|
|
|
|
memset (&tmout, 0, HIO_SIZEOF(tmout));
|
|
HIO_INIT_NTIME (&tmout.c, 4, 0);
|
|
HIO_INIT_NTIME (&tmout.r, -1, 0);
|
|
HIO_INIT_NTIME (&tmout.w, -1, 0);
|
|
marc = hio_svc_marc_start(hio, &ci, &tmout, HIO_NULL);
|
|
/*marc = hio_svc_marc_start(hio, &ci, HIO_NULL, HIO_NULL);*/
|
|
if (!marc)
|
|
{
|
|
printf ("Cannot start a mariadb client service\n");
|
|
goto oops;
|
|
}
|
|
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_SELECT, "SHOW STATUS", 11, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM", 11, on_result, HIO_NULL);
|
|
// hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_SELECT, "SHOW STATUS", 11, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM XXX", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM ZZZ", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM TTT", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM QQQ", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 4 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 10 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 10 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
hio_svc_marc_querywithbchars (marc, 10 | HIO_SVC_MARC_SID_FLAG_AUTO_BOUNDED, HIO_SVC_MARC_QTYPE_ACTION, "DELETE FROM RRR", 14, on_result, HIO_NULL);
|
|
|
|
#if 0
|
|
memset (&mi, 0, HIO_SIZEOF(mi));
|
|
/*mi.on_write = mar_on_write;
|
|
mi.on_read = mar_on_read;*/
|
|
mi.on_connect = mar_on_connect;
|
|
mi.on_disconnect = mar_on_disconnect;
|
|
mi.on_query_started = mar_on_query_started;
|
|
mi.on_row_fetched = mar_on_row_fetched;
|
|
|
|
mar = hio_dev_mar_make(hio, 0, &mi);
|
|
if (!mar)
|
|
{
|
|
printf ("Cannot make a mar db client device\n");
|
|
goto oops;
|
|
}
|
|
|
|
if (hio_dev_mar_connect(mar, &ci) <= -1)
|
|
{
|
|
printf ("Cannot connect to mar db server\n");
|
|
goto oops;
|
|
}
|
|
#endif
|
|
|
|
g_hio = hio;
|
|
signal (SIGINT, handle_signal);
|
|
|
|
/* ---------------------------------------- */
|
|
{
|
|
hio_ntime_t x;
|
|
HIO_INIT_NTIME (&x, 20, 0);
|
|
schedule_timer_job_after (hio, &x, send_test_query, marc);
|
|
hio_loop (hio);
|
|
}
|
|
/* ---------------------------------------- */
|
|
|
|
signal (SIGINT, SIG_IGN);
|
|
g_hio = HIO_NULL;
|
|
|
|
oops:
|
|
printf ("about to close hio...\n");
|
|
if (hio) hio_close (hio);
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
#else
|
|
|
|
#include <stdio.h>
|
|
int main (int argc, char* argv[])
|
|
{
|
|
printf ("mariadb not enabled\n");
|
|
return 0;
|
|
}
|
|
#endif
|