wrapper.cpp

//
// Created by tyy on 2017/1/9.
//
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>

#include "ipLocator.h"
#include "userAgent.h"
#include "wrapper.h"
#include "lrucache.h"


IPSearch *finder;
#ifdef LRU_CACHE
cache::lru_cache<uint32_t, string> cacheIP(100000);
#endif
void ipwrapper_init() {
    finder = IPSearch::instance();
    if (!finder) {
        printf("the IPSearch instance is null!");
        exit(1);
    }
}


static inline uint32_t strtoip(const char *ip) {
    uint32_t ipaddr[4] = {0};
    sscanf(ip, "%u.%u.%u.%u", &ipaddr[0], &ipaddr[1], &ipaddr[2], &ipaddr[3]);
    return (ipaddr[0] << 24) | (ipaddr[1] << 16) | (ipaddr[2] << 8) | ipaddr[3];
}


static inline void iptostr(char ipstr[], uint16_t len, uint32_t ip) {
    memset(ipstr, 0, len);
    sprintf(ipstr, "%u.%u.%u.%u",
            ip >> 24, (ip >> 16 & 0xff), (ip >> 8 & 0xff), (ip & 0xff));
}

#define MAX_JSON_STR 2048
static char jsonStr[MAX_JSON_STR];

#define MAX_IP_STR 256
static char ipStr[MAX_IP_STR];

void ipStrSplit(const char *query, int len, IPInfo &ipInfo) {
    char *p[16];
    char *token, *next;
    int in = 0;
    char *buf = ipStr;
    memcpy(buf, query, len);
    buf[len] = '|';

    while((token = strchr(buf, '|')))  {
        next = token + 1;
        *token = '\0';
        p[in++] = buf;
        buf = next;
    }

    if (in > 10 && strncmp(p[5], "qqzeng-ip", strlen("qqzeng-ip") != 0)) {
        ipInfo.city = p[3];
        ipInfo.isp = p[5];
        ipInfo.longitude = p[9];
        ipInfo.latitude = p[10];
    }
}


char *ip2JsonStr(const char *ip) {
    char ch[32] = {0,};
    const char *ipaddr = NULL;
    const char *raw = ip;

    const char *isdot = strstr(ip, ".");
    if (!isdot) {   // "1916214160" => 114.55.27.144
        iptostr(ch, 32, atoi(ip));
        ipaddr = ch;
    } else {    // "58.214.57.66" => 58.214.57.66
        int iplen = strlen(ip);
        if (ip[0] == '\"') {
            strncpy(ch, ip + 1, iplen);
            ch[iplen - 2] = '\0';
            ipaddr = ch;
        } else {
            ipaddr = ip;
        }
    }

    int intIP = strtoip(ipaddr);
#ifdef LRU_CACHE
    if (cacheIP.exists(intIP) == true) {
        string cache = cacheIP.get(intIP);
        memset(jsonStr, 0, MAX_JSON_STR);
        strncpy(jsonStr, cache.c_str(), cache.size());
        return jsonStr;
    }
#endif
    const char *query = finder->Query(ipaddr).c_str();
    IPInfo ipInfo;
    int qlen = strlen(query);
    if (qlen > 0) {
        ipStrSplit(query, qlen, ipInfo);
    }

    memset(jsonStr, 0, MAX_JSON_STR);

    char dec[32] = {0,};
    sprintf(dec, "%u", intIP);

    strcat(jsonStr, "{\"decimal\":");
    strcat(jsonStr, dec);

    strcat(jsonStr, ",\"raw\":");

    strcat(jsonStr, raw);

    strcat(jsonStr, ",\"dotted\":\"");
    strcat(jsonStr, ipaddr);

    strcat(jsonStr, "\",\"region\":\"");
    strcat(jsonStr, ipInfo.city.c_str());

    strcat(jsonStr, "\",\"isp\":\"");
    strcat(jsonStr, ipInfo.isp.c_str());

    strcat(jsonStr, "\",\"longtitude\":\"");
    strcat(jsonStr, ipInfo.longitude.c_str());

    strcat(jsonStr, "\",\"latitude\":\"");
    strcat(jsonStr, ipInfo.latitude.c_str());
    strcat(jsonStr, "\"}");
#ifdef LRU_CACHE
    cacheIP.put(intIP, jsonStr);
#endif
    return jsonStr;
}


char *ua2JsonStr(const char *ua, int ua_len) {

    UserAgent p;
    Parse(p, ua, ua_len);
//    echo_ua(p);

    memset(jsonStr, 0, MAX_JSON_STR);

    strcat(jsonStr, "{\"bot\":");
    if (p.bot) {
        strcat(jsonStr, "true");
    } else {
        strcat(jsonStr, "false");
    }

    strcat(jsonStr, ",\"browser\":\"");
    strcat(jsonStr, p.browser.Name.c_str());

    strcat(jsonStr, "\",\"browser_version\":\"");
    strcat(jsonStr, p.browser.Version.c_str());

    strcat(jsonStr, "\",\"engine\":\"");
    strcat(jsonStr, p.browser.Engine.c_str());

    strcat(jsonStr, "\",\"engine_version\":\"");
    strcat(jsonStr, p.browser.EngineVersion.c_str());

    strcat(jsonStr, "\",\"os\":\"");
    strcat(jsonStr, p.os.c_str());

    strcat(jsonStr, "\",\"platform\":\"");
    strcat(jsonStr, p.platform.c_str());

    strcat(jsonStr, "\",\"raw\":\"");
    strcat(jsonStr, ua);

    strcat(jsonStr, "\"}");

    return jsonStr;
}