예제 #1
0
static void
ngx_mysql_read_query_result(ngx_event_t *rev)
{
    ssize_t                    n, len;
    ngx_str_t                  msg;
    ngx_mysql_t               *m;
    ngx_connection_t          *c;
    ngx_mysql_error_pkt_t     *epkt;
    ngx_mysql_response_pkt_t  *pkt;
    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, rev->log, 0, "mysql read query result");
    c = rev->data;
    m = c->data;
    m->peer.log->action = "reading mysql read query result";
    n = ngx_recv(m->peer.connection, m->buf->pos, /* STUB */ 1024);
    if (n == NGX_AGAIN)
    {
        return;
    }
    if (n < 5)
    {
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    pkt = (ngx_mysql_response_pkt_t *) m->buf->pos;
    len = ngx_m24toh(pkt->pktlen);
    if (len > n - 4)
    {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "mysql server %V sent incomplete response packet",
                      m->peer.name);
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    if (pkt->fields != 0xff)
    {
        ngx_log_debug0(NGX_LOG_DEBUG_HTTP, rev->log, 0, "mysql query OK");
        m->state = NGX_OK;
        m->pktn = pkt->pktn;
        m->handler(m);
        return;
    }
    epkt = (ngx_mysql_error_pkt_t *) pkt;
    msg.len = (u_char *) epkt + 4 + len - epkt->message;
    msg.data = epkt->message;
    ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                  "mysql server %V sent error (%ud): \"%V\"",
                  m->peer.name, ngx_m16toh(epkt->code), &msg);
    ngx_mysql_close(m, NGX_ERROR);
}
예제 #2
0
파일: ngx_mysql.c 프로젝트: 0x7E/tengine
ngx_int_t
ngx_mysql_query(ngx_mysql_t *m)
{
    ssize_t                   n;
    ngx_mysql_command_pkt_t  *pkt;

    pkt = (ngx_mysql_command_pkt_t *) m->query.data;

    ngx_htom24(pkt->pktlen, m->query.len - 4);
    pkt->pktn = (u_char) m->pktn++;
    pkt->command = NGX_MYSQL_CMD_QUERY;

    n = ngx_send(m->peer.connection, m->query.data, m->query.len);

    if (n < (ssize_t) m->query.len) {
        ngx_log_error(NGX_LOG_ERR, m->peer.log, 0,
                      "the incomplete packet was sent to mysql server %V",
                      m->peer.name);

        ngx_mysql_close(m, NGX_ERROR);
        return NGX_OK;
    }

    m->peer.connection->read->handler = ngx_mysql_read_query_result;

    ngx_add_timer(m->peer.connection->read, /* STUB */ 5000);

    /* STUB handle event */

    return NGX_OK;
}
예제 #3
0
static void
ngx_mysql_read_server_greeting(ngx_event_t *rev)
{
    size_t                      len;
    u_char                     *p;
    ssize_t                     n;
    ngx_uint_t                  i, capacity;
    ngx_mysql_t                *m;
    ngx_connection_t           *c;
    ngx_mysql_greeting1_pkt_t  *gr1;
    ngx_mysql_greeting2_pkt_t  *gr2;
    ngx_mysql_auth_pkt_t       *auth;
    ngx_sha1_t                  sha;
    u_char                      hash1[20], hash2[20];
    c = rev->data;
    m = c->data;
    if (rev->timedout)
    {
        ngx_log_error(NGX_LOG_ERR, rev->log, NGX_ETIMEDOUT,
                      "mysql server %V timed out", m->peer.name);
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    if (m->buf == NULL)
    {
        m->peer.log->action = "reading mysql server greeting";
        m->buf = ngx_create_temp_buf(m->pool, /* STUB */ 1024);
        if (m->buf == NULL)
        {
            ngx_mysql_close(m, NGX_ERROR);
            return;
        }
    }
    n = ngx_recv(m->peer.connection, m->buf->pos, /* STUB */ 1024);
    if (n == NGX_AGAIN)
    {
        return;
    }
    if (n < 5)
    {
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    gr1 = (ngx_mysql_greeting1_pkt_t *) m->buf->pos;
    if (ngx_m24toh(gr1->pktlen) > n - 4)
    {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "mysql server %V sent incomplete greeting packet",
                      m->peer.name);
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    if (gr1->protocol < 10)
    {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "mysql server %V sent unsupported protocol version %ud",
                      m->peer.name, gr1->protocol);
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    gr2 = (ngx_mysql_greeting2_pkt_t *)
          (gr1->version + ngx_strlen(gr1->version) + 1);
    capacity = ngx_m16toh(gr2->capacity);
    ngx_log_debug8(NGX_LOG_DEBUG_MYSQL, rev->log, 0,
                   "mysql version: %ud, \"%s\", thread: %ud, salt: \"%s\", "
                   "capacity: %Xd, charset: %ud, status: %ud, salt rest \"%s\"",
                   gr1->protocol, gr1->version, ngx_m32toh(gr2->thread),
                   gr2->salt1, capacity, gr2->charset,
                   ngx_m16toh(gr2->status), &gr2->salt2);
    capacity = NGX_MYSQL_LONG_PASSWORD
               | NGX_MYSQL_CONNECT_WITH_DB
               | NGX_MYSQL_PROTOCOL_41
               | NGX_MYSQL_SECURE_CONNECTION;
    len = 4 + 4 + 4 + 1 + 23 + m->login->len + 1 + 1 + m->database->len + 1;
    if (m->passwd->len)
    {
        len += 20;
    }
    auth = ngx_pnalloc(m->pool, len);
    if (auth == NULL)
    {
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    ngx_htom24(auth->pktlen, len - 4);
    auth->pktn = (u_char)(gr1->pktn + 1);
    ngx_htom32(auth->capacity, capacity);
    ngx_htom32(auth->max_packet, 0x01000000);  /* max packet size 2^24 */
    ngx_memzero(auth->zero, 24);
    auth->charset = gr2->charset;
    p = ngx_copy(auth->login, m->login->data, m->login->len);
    *p++ = '\0';
    if (m->passwd->len)
    {
        *p++ = (u_char) 20;
        ngx_sha1_init(&sha);
        ngx_sha1_update(&sha, m->passwd->data, m->passwd->len);
        ngx_sha1_final(hash1, &sha);
        ngx_sha1_init(&sha);
        ngx_sha1_update(&sha, hash1, 20);
        ngx_sha1_final(hash2, &sha);
        ngx_sha1_init(&sha);
        ngx_sha1_update(&sha, gr2->salt1, 8);
        ngx_sha1_update(&sha, gr2->salt2, 12);
        ngx_sha1_update(&sha, hash2, 20);
        ngx_sha1_final(hash2, &sha);
        for (i = 0; i < 20; i++)
        {
            *p++ = (u_char)(hash1[i] ^ hash2[i]);
        }
    }
    else
    {
        *p++ = '\0';
    }
    p = ngx_copy(p, m->database->data, m->database->len);
    *p = '\0';
    n = ngx_send(m->peer.connection, (void *) auth, len);
    if (n < (ssize_t) len)
    {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "the incomplete packet was sent to mysql server %V",
                      m->peer.name);
        ngx_mysql_close(m, NGX_ERROR);
        return;
    }
    m->peer.connection->read->handler = ngx_mysql_read_auth_result;
    ngx_add_timer(m->peer.connection->read, /* STUB */ 5000);
}