nginx/src/mysql/ngx_mysql.c

/*
 * Copyright (C) Igor Sysoev
 */


#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_event.h>
#include <ngx_mysql.h>


/* the library supports the subset of the MySQL 4.1+ protocol (version 10) */


ngx_int_t
ngx_mysql_connect(ngx_mysql_t *m)
{
    ngx_int_t  rc;

#if 0
    if (cached) {
        return NGX_OK;
    }
#endif

    m->peer.log->action = "connecting to mysql server";

    rc = ngx_event_connect_peer(&m->peer);

    if (rc == NGX_ERROR || rc == NGX_BUSY || rc == NGX_DECLINED) {
        return rc;
    }

    m->peer.connection->read->handler = ngx_mysql_read_server_greeting;
    m->peer.connection->write->handler = ngx_mysql_emtpy_handler;

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

    return NGX_OK;
}


static void
ngx_mysql_read_server_greeting(ngx_event_t *rev)
{
    size_t             len;
    u_char            *p, *t;
    ngx_mysql_t       *m;
    ngx_connection_t  *c;

    c = rev->data;
    m = c->data;

    if (rev->timedout) {
        ngx_log_error(NGX_LOG_ERR, rev->log, NGX_ETIMEDOUT,
                      "mysql server %V timed out",
                      &ctx->peer.peers->peer[0].name);

        ngx_mysql_close(m, NGX_ERROR);
        return;
    }

    if (m->buf == NULL) {
        m->peer.log->action = "reading to mysql server greeting";

        m->buf = ngx_create_temp(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;
    }

    p = m->buf->pos;

    if (ngx_m24toh(p) > n - 4) {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "mysql server %V sent incomplete greeting packet",
                      &ctx->peer.peers->peer[0].name);

        ngx_mysql_close(m, NGX_ERROR);
        return;
    }

    if (p[4]) < 10) {
        ngx_log_error(NGX_LOG_ERR, rev->log, 0,
                      "mysql server %V sent unsupported protocol version %ud",
                      &ctx->peer.peers->peer[0].name, p[4]);

        ngx_mysql_close(m, NGX_ERROR);
        return;
    }

    len = ngx_strlen(&p[5]);
    t = p + 5 + len + 1;

    capacity = ngx_m16toh((&t[4 + 9]));

    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\"",
                   p[4], &p[5], ngx_m32toh(t), &t[4],
                   capacity, t[4 + 9 + 2],
                   ngx_m16toh((&t[4 + 9 + 2 + 1])),
                   t[4 + 9 + 2 + 1 + 2 + 13]);

    capacity &= NGX_MYSQL_LONG_PASSWORD
                | NGX_MYSQL_CONNECT_WITH_DB
                | NGX_MYSQL_PROTOCOL_41;

}


static void
ngx_mysql_close(ngx_mysql_t *m, ngx_int_t rc)
{
    if (rc == NGX_ERROR) {
        ngx_close_connection(m->peer.connection);
    }

    m->state = rc;

    m->handler(m);
}