You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.1 KiB
137 lines
3.1 KiB
/*
|
|
* include/nuster/key.h
|
|
* This file defines everything related to nuster key.
|
|
*
|
|
* Copyright (C) Jiang Wenyuan, < koubunen AT gmail DOT com >
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation, version 2.1
|
|
* exclusively.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef _NUSTER_KEY_H
|
|
#define _NUSTER_KEY_H
|
|
|
|
#include <haproxy/chunk.h>
|
|
|
|
#include <nuster/common.h>
|
|
|
|
|
|
#define NST_KEY_UUID_LEN 20
|
|
|
|
enum {
|
|
NST_KEY_MEMORY_CHECKED = 0x0001,
|
|
NST_KEY_DISK_CHECKED = 0x0002,
|
|
};
|
|
|
|
typedef struct nst_key {
|
|
uint8_t flags;
|
|
uint32_t size;
|
|
char *data;
|
|
uint64_t hash;
|
|
unsigned char uuid[NST_KEY_UUID_LEN];
|
|
} nst_key_t;
|
|
|
|
|
|
static inline int
|
|
nst_key_memory_checked(nst_key_t *key) {
|
|
return key->flags & NST_KEY_MEMORY_CHECKED;
|
|
}
|
|
|
|
static inline void
|
|
nst_key_memory_set_checked(nst_key_t *key) {
|
|
key->flags |= NST_KEY_MEMORY_CHECKED;
|
|
}
|
|
|
|
static inline int
|
|
nst_key_disk_checked(nst_key_t *key) {
|
|
return key->flags & NST_KEY_DISK_CHECKED;
|
|
}
|
|
|
|
static inline void
|
|
nst_key_disk_set_checked(nst_key_t *key) {
|
|
key->flags |= NST_KEY_DISK_CHECKED;
|
|
}
|
|
|
|
static inline void
|
|
nst_key_reset_flag(nst_key_t *key) {
|
|
key->flags = 0;
|
|
}
|
|
|
|
static inline hpx_buffer_t *
|
|
nst_key_init() {
|
|
hpx_buffer_t *key = get_trash_chunk();
|
|
|
|
memset(key->area, 0, key->size);
|
|
|
|
return key;
|
|
}
|
|
|
|
static inline int
|
|
nst_key_cat(hpx_buffer_t *key, const char *ptr, int len) {
|
|
|
|
if(key->data + len > key->size) {
|
|
return NST_ERR;
|
|
}
|
|
|
|
memcpy(key->area + key->data, ptr, len);
|
|
key->data += len;
|
|
|
|
return NST_OK;
|
|
}
|
|
|
|
static inline int
|
|
nst_key_catist(hpx_buffer_t *key, hpx_ist_t v) {
|
|
/* additional one NULL delimiter */
|
|
if(key->data + v.len + 1 > key->size) {
|
|
return NST_ERR;
|
|
}
|
|
|
|
memcpy(key->area + key->data, v.ptr, v.len);
|
|
key->data += v.len + 1;
|
|
|
|
return NST_OK;
|
|
}
|
|
|
|
static inline int
|
|
nst_key_catdel(hpx_buffer_t *key) {
|
|
|
|
if(key->data + 1 > key->size) {
|
|
return NST_ERR;
|
|
}
|
|
|
|
key->data += 1;
|
|
|
|
return NST_OK;
|
|
}
|
|
|
|
/*
|
|
* It's the caller's reponsibility to allocate str and zero termination
|
|
*/
|
|
static inline void
|
|
nst_key_uuid_stringify(nst_key_t *key, char *str){
|
|
int i;
|
|
|
|
for(i = 0; i < NST_KEY_UUID_LEN; i++) {
|
|
sprintf((char*)&(str[i*2]), "%02x", key->uuid[i]);
|
|
}
|
|
}
|
|
|
|
void nst_key_hash(nst_key_t *key);
|
|
|
|
void nst_key_debug(hpx_stream_t *s, nst_key_t *key);
|
|
|
|
int nst_key_build(hpx_stream_t *s, hpx_http_msg_t *msg, nst_rule_t *rule, nst_http_txn_t *txn,
|
|
nst_key_t *key, hpx_http_meth_t method);
|
|
|
|
#endif /* _NUSTER_KEY_H */
|
|
|