ngx_stream_upstream_check_module.h

#ifndef _NGX_STREAM_UPSTREAM_CHECK_H_INCLUDED_
#define _NGX_STREAM_UPSTREAM_CHECK_H_INCLUDED_


#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_event.h>
#include <ngx_event_connect.h>
#include <ngx_event_pipe.h>
#include <ngx_stream.h>


typedef struct ngx_stream_check_peer_conf_s ngx_stream_check_peer_conf_t;
typedef struct ngx_stream_check_peers_conf_s ngx_stream_check_peers_conf_t;
typedef struct check_conf_s check_conf_t;


typedef struct {
    u_char                 major;
    u_char                 minor;
} ssl_protocol_version_t;

typedef struct {
    u_char                 msg_type;
    ssl_protocol_version_t version;
    uint16_t               length;

    u_char                 handshake_type;
    u_char                 handshake_length[3];
    ssl_protocol_version_t hello_version;

    time_t                 time;
    u_char                 random[28];

    u_char                 others[0];
} __attribute__((packed)) server_ssl_hello_t;

typedef struct {
    u_char                 packet_length[3];
    u_char                 packet_number;

    u_char                 protocol_version;
    u_char                 others[0];
} __attribute__((packed)) mysql_handshake_init_t;

typedef struct {
    ngx_buf_t              send;
    ngx_buf_t              recv;

    void                  *parser;
} ngx_stream_check_ctx;

/*state*/
#define NGX_STREAM_CHECK_CONNECT_DONE     0x0001
#define NGX_STREAM_CHECK_SEND_DONE        0x0002
#define NGX_STREAM_CHECK_RECV_DONE        0x0004
#define NGX_STREAM_CHECK_ALL_DONE         0x0008

typedef struct {
    ngx_pid_t              owner;

    ngx_msec_t             access_time;

    ngx_uint_t             fall_count;
    ngx_uint_t             rise_count;

    ngx_atomic_t           lock;
    ngx_atomic_t           down;

    ngx_uint_t             delete;

    ngx_uint_t             access_count;
} ngx_stream_check_peer_shm_t;

typedef struct {
    ngx_uint_t             generation;

    ngx_uint_t             state;
    ngx_atomic_t           lock;

    /*store the ngx_stream_check_status_peer_t*/
    ngx_stream_check_peer_shm_t peers[1];
} ngx_stream_check_peers_shm_t;

typedef ngx_int_t (*ngx_stream_check_packet_init_pt)
    (ngx_stream_check_peer_conf_t *peer_conf); 
typedef ngx_int_t (*ngx_stream_check_packet_parse_pt)
    (ngx_stream_check_peer_conf_t *peer_conf); 
typedef void (*ngx_stream_check_packet_clean_pt)
    (ngx_stream_check_peer_conf_t *peer_conf); 

#define NGX_STREAM_CHECK_TCP              0x0001
#define NGX_STREAM_CHECK_HTTP             0x0002
#define NGX_STREAM_CHECK_SSL_HELLO        0x0004
#define NGX_STREAM_CHECK_MYSQL            0x0010
#define NGX_STREAM_CHECK_POP3             0x0020
#define NGX_STREAM_CHECK_IMAP             0x0040

#define NGX_CHECK_HTTP_2XX             0x0002
#define NGX_CHECK_HTTP_3XX             0x0004
#define NGX_CHECK_HTTP_4XX             0x0008
#define NGX_CHECK_HTTP_5XX             0x0010
#define NGX_CHECK_HTTP_6XX             0x0020
#define NGX_CHECK_HTTP_ERR             0x8000


struct check_conf_s {
    ngx_uint_t                          type;

    char                               *name;

    ngx_str_t                           default_send;
    
    /* HTTP */
    ngx_uint_t                          default_status_alive;

    ngx_event_handler_pt                send_handler;
    ngx_event_handler_pt                recv_handler;

    ngx_stream_check_packet_init_pt     init;
    ngx_stream_check_packet_parse_pt    parse;
    ngx_stream_check_packet_clean_pt    reinit;

    unsigned need_pool;
};

typedef struct {
    ngx_uint_t                          fall_count;
    ngx_uint_t                          rise_count;
    ngx_msec_t                          check_interval;
    ngx_msec_t                          check_timeout;

    check_conf_t                       *check_type_conf;
    ngx_str_t                           send;

    union {
        ngx_uint_t                      return_code;
        ngx_uint_t                      status_alive;
    } code;
} ngx_stream_upstream_check_srv_conf_t;

struct ngx_stream_check_peer_conf_s {

    ngx_flag_t                             state;
    ngx_pool_t                            *pool;
    ngx_uint_t                             index;
    ngx_event_t                            check_ev;
    ngx_event_t                            check_timeout_ev;
    ngx_addr_t                            *peer;
    ngx_peer_connection_t                  pc;

    void *                                 check_data;
    ngx_event_handler_pt                   send_handler;
    ngx_event_handler_pt                   recv_handler;

    ngx_stream_check_packet_init_pt        init;
    ngx_stream_check_packet_parse_pt       parse;
    ngx_stream_check_packet_clean_pt       reinit;

    ngx_stream_upstream_check_srv_conf_t  *conf;
    ngx_stream_check_peer_shm_t           *shm;

    unsigned                               delete;
};

struct ngx_stream_check_peers_conf_s {
    ngx_str_t                           check_shm_name;
    ngx_array_t                         peers;

    ngx_stream_check_peers_shm_t       *peers_shm;
};

typedef struct {
    ngx_uint_t                          check_shm_size;
    ngx_stream_check_peers_conf_t      *peers_conf;
} ngx_stream_upstream_check_main_conf_t;


ngx_uint_t ngx_stream_check_add_peer(ngx_conf_t *cf,
    ngx_stream_upstream_srv_conf_t *uscf, ngx_addr_t *peer);

ngx_uint_t ngx_stream_check_peer_down(ngx_uint_t index);

void ngx_stream_check_get_peer(ngx_uint_t index);
void ngx_stream_check_free_peer(ngx_uint_t index);

ngx_uint_t ngx_stream_upstream_check_add_dynamic_peer(ngx_pool_t *pool,
    ngx_stream_upstream_srv_conf_t *us, ngx_addr_t *peer_addr);

void ngx_stream_upstream_check_delete_dynamic_peer(ngx_str_t *name,
    ngx_addr_t *peer_addr);


#endif //_NGX_STREAM_UPSTREAM_CHECK_H_INCLUDED_