bool service::process_write(HTTP_CODE ret)
{
if(ret==FILE_REQUEST)
{
if(send_html())
return true;
else
return false;
}
return false;
}
int main(int argc, const char *argv[])
{
sqlite3 *db;
char *err = 0;
int ret = 0;
char buf[4096];
char value[4096];
char *p = NULL;
char *q = NULL;
int len = 0;
int reg = 0;
int log = 0;
int login_flag = 0;
char *confirm;
char ip[36] = {0};
printf("Content-Type: text/html\n\n");
while((ret = fread(buf, 1, sizeof(buf), stdin)) != 0)
{
//printf("read %d len\n", ret);
//printf("%s", buf);
}
if (get_local_ip(ip, NULL) != 0)
{
printf("Server is unavailable\n");
exit(1);
}
if ((p = strstr(buf, "=Register")) != NULL)
{
reg = 1;
p -= 4;
*p = '\0';
q = strstr(buf, "&confirm=");
*q = '\0';
confirm = q + 9;
}
if ((p = strstr(buf, "&log=Login")) != NULL)
{
log = 1;
*p = '\0';
}
//username=sd&password=sdd&confirm=dsssd®=Register
p = strchr(buf, '=');
p++;
q = strchr(buf, '&');
if (p == q)
{
send_html(0, 0, "Blank username!");
if (reg == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
}
if (log == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/index.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
return 0;
}
*q = '\0';
//username=ab&password=ab&confirm=ab®=Register
if (strchr(p, '+') != NULL)
{
send_html(0, 0, "Username can not have blank space !");
if (reg == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
}
if (log == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/admin_login.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
return 0;
}
if ((len = strlen(p)) > 20)
{
send_html(0, 0, "Username is overlength !");
if (reg == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
}
if (log == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/admin_login.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
return 0;
}
str.username = p;
q += 10;
if (*q == '\0' || *q == '&')
{
send_html(0, 0, "Blank password !");
if (reg == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
}
if (log == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/admin_login.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
return 0;
}
/*
if ((p = strstr(q, "®=Register")) != NULL)
{
reg = 1;
*p = '\0';
}
if ((p = strstr(q, "&log=Login")) != NULL)
{
log = 1;
*p = '\0';
}
*/
if ((len = strlen(q)) > 20)
{
printf("\n%s\n",q);
send_html(0, 0, "Password is overlength !");
if (reg == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
}
if (log == 1)
{
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/admin_login.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
return 0;
}
str.password = q;
#if 1
if (reg == 1)
{
if (strcmp(str.password, confirm) != 0)
{
send_html(0, 0, "Confirm password not match !");
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
return 0;
}
}
#endif
ret = sqlite3_open("./admin_login.db", &db);
if (ret != SQLITE_OK)
{
fputs(sqlite3_errmsg(db), stderr);
fputs("\n", stderr);
exit(1);
}
if (reg == 1)
{
//sprintf(value, "select username='******', username, password from register_info;", str.username);
sprintf(value, "select * from register_info where username='******';", str.username);
//ret = sqlite3_exec(db, "select * from register_info", rscallback_register, &login_flag, &err);
ret = sqlite3_exec(db, value, rscallback_register, &login_flag, &err);
if (login_flag == 1)
{
sqlite3_close(db);
send_html(0, 0, "Reduplicated username !");
send_html(0, 0, "Please input your username again !");
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/register.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/register.html\">",ip);
return 0;
}
sprintf(value, "insert into register_info(username, password) values('%s', '%s');", str.username, str.password);
ret = sqlite3_exec(db, value, NULL, NULL, &err);
}
if (log == 1)
{
//sprintf(value, "select username='******', username, password from register_info;", str.username);
sprintf(value, "select * from register_info where username='******' and password='******';", str.username, str.password);
ret = sqlite3_exec(db, value, rscallback_login, &login_flag, &err);
//ret = sqlite3_exec(db, "select * from register_info", rscallback_login, &login_flag, &err);
}
if (ret != SQLITE_OK)
{
fputs(err, stderr);
fputs("\n", stderr);
sqlite3_close(db);
if (reg == 1)
{
printf("\nRegister failure !\n");
}
if (log == 1)
{
printf("\nLogin failure !\n");
}
exit(1);
}
sqlite3_close(db);
if (reg == 1)
{
send_html(0, 0, "Register succesfully !");
send_html(0, 0, "Turning into administration system...");
//printf("<meta http-equiv=\"refresh\"content=\"1; url=http://192.168.11.252/administration.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/administration.html\">",ip);
}
if (log == 1)
{
if (login_flag == 1)
{
send_html(0, 0, "Login succesfully !");
send_html(0, 0, "Turning into administration system...");
//printf("<meta http-equiv=\"refresh\"content=\"1; url=http://192.168.11.252/administration.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/administration.html\">",ip);
}
else
{
send_html(0, 0, "Username or password wrong, please input again !");
//printf("<meta http-equiv=\"refresh\"content=\"2; url=http://192.168.11.252/index.html\">");
printf("<meta http-equiv=\"refresh\"content=\"2; url=http://%s/index.html\">",ip);
}
}
return 0;
}
