mirror of
https://github.com/wg/wrk
synced 2025-03-14 10:37:16 +08:00
65 lines
1.1 KiB
C
65 lines
1.1 KiB
C
#ifndef WRK_H
|
|
#define WRK_H
|
|
|
|
#include "config.h"
|
|
#include <pthread.h>
|
|
#include <inttypes.h>
|
|
#include <sys/types.h>
|
|
#include <netdb.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include <openssl/ssl.h>
|
|
#include <openssl/err.h>
|
|
#include <lua.h>
|
|
|
|
#include "stats.h"
|
|
#include "ae.h"
|
|
#include "http_parser.h"
|
|
|
|
#define VERSION "3.1.2"
|
|
#define RECVBUF 8192
|
|
|
|
#define MAX_THREAD_RATE_S 10000000
|
|
#define SOCKET_TIMEOUT_MS 2000
|
|
#define RECORD_INTERVAL_MS 100
|
|
|
|
typedef struct {
|
|
pthread_t thread;
|
|
aeEventLoop *loop;
|
|
struct addrinfo *addr;
|
|
uint64_t connections;
|
|
uint64_t complete;
|
|
uint64_t requests;
|
|
uint64_t bytes;
|
|
uint64_t start;
|
|
lua_State *L;
|
|
errors errors;
|
|
struct connection *cs;
|
|
} thread;
|
|
|
|
typedef struct {
|
|
char *buffer;
|
|
size_t length;
|
|
char *cursor;
|
|
} buffer;
|
|
|
|
typedef struct connection {
|
|
thread *thread;
|
|
http_parser parser;
|
|
enum {
|
|
FIELD, VALUE
|
|
} state;
|
|
int fd;
|
|
SSL *ssl;
|
|
uint64_t start;
|
|
char *request;
|
|
size_t length;
|
|
size_t written;
|
|
uint64_t pending;
|
|
buffer headers;
|
|
buffer body;
|
|
char buf[RECVBUF];
|
|
} connection;
|
|
|
|
#endif /* WRK_H */
|