clearlinux/uwsgi
2
0
Fork 0
mirror of https://github.com/clearlinux/uwsgi.git synced 2026-09-06 13:41:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

added base64 functions

Browse Source
This commit is contained in:
Roberto De Ioris
2013-01-05 17:19:23 +01:00
parent 912098c4c1
commit 08b2437146
3 changed files with 108 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/utils.c
+103
View File
@@ -4008,3 +4008,106 @@ void uwsgi_additional_header_add(struct wsgi_request *wsgi_req, char *hh, uint16
uwsgi_string_new_list(&wsgi_req->additional_headers, header);
}
// based on nginx implementation
static uint8_t b64_table64[] = {
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 62, 77, 77, 77, 63,
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 77, 77, 77, 77, 77, 77,
77, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 77, 77, 77, 77, 77,
77, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77
};
static char b64_table64_2[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
char *uwsgi_base64_decode(char *buf, size_t len, size_t *d_len) {
// find the real size and check for invalid values
size_t i;
for (i = 0; i < len; i++) {
if (buf[i] == '=')
break;
// check for invalid content
if (b64_table64[ (uint8_t) buf[i] ] == 77) {
return NULL;
}
}
// check for invalid size
if (i % 4 == 1)
return NULL;
// compute the new size
*d_len = (((len+3)/4) * 3);
char *dst = uwsgi_malloc(*d_len);
char *ptr = dst;
uint8_t *src = (uint8_t *) buf;
while(i > 3) {
*ptr++= (char) ( b64_table64[src[0]] << 2 | b64_table64[src[1]] >> 4);
*ptr++= (char) ( b64_table64[src[1]] << 4 | b64_table64[src[2]] >> 2);
*ptr++= (char) ( b64_table64[src[2]] << 6 | b64_table64[src[3]]);
src+=4;
i-=4;
}
if (i > 1) {
*ptr++= (char) ( b64_table64[src[0]] << 2 | b64_table64[src[1]] >> 4);
}
if (i > 2) {
*ptr++= (char) ( b64_table64[src[1]] << 4 | b64_table64[src[2]] >> 2);
}
*d_len = (ptr - dst);
return dst;
}
char *uwsgi_base64_encode(char *buf, size_t len, size_t *d_len) {
*d_len = ((len * 4)/3) + 5;
char *dst = uwsgi_malloc(*d_len);
char *ptr = dst;
while(len >= 3) {
*ptr++= b64_table64_2[ buf[0] >> 2];
*ptr++= b64_table64_2[((buf[0] << 4) & 0x30) | (buf[1] >> 4)];
*ptr++= b64_table64_2[((buf[1] << 2) & 0x3C) | (buf[2] >> 6)];
*ptr++= b64_table64_2[buf[2] & 0x3F];
buf += 3;
len -= 3;
}
if (len > 0) {
*ptr++= b64_table64_2[ buf[0] >> 2];
uint8_t tmp = (buf[0] << 4) & 0x30;
if (len > 1) tmp |= buf[1] >> 4;
*ptr++= b64_table64_2[tmp];
if (len < 2) {
*ptr++= '=';
}
else {
*ptr++= b64_table64_2[buf[2] & 0x3F];
}
*ptr++= '=';
}
*ptr = 0;
*d_len = (ptr - dst);
return dst;
}
plugins/router_basicauth/router_basicauth.c
+2 -67
View File
@@ -12,72 +12,6 @@ pthread_mutex_t ur_basicauth_crypt_mutex;
extern struct uwsgi_server uwsgi;
// this algo is based on nginx one (the fastest i have tested)
static char *http_basic_auth_get(char *authorization, uint16_t len) {
uint16_t i;
static uint8_t table64[] = {
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 62, 77, 77, 77, 63,
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 77, 77, 77, 77, 77, 77,
77, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 77, 77, 77, 77, 77,
77, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77,
77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77, 77
};
for (i = 0; i < len; i++) {
if (authorization[i] == '=')
break;
// check for invalid content
if (table64[ (uint8_t) authorization[i] ] == 77) {
return NULL;
}
}
// check for invalid length
if (i % 4 == 1)
return NULL;
uint16_t dst_len = (((len+3)/4) * 3) + 1;
char *dst = uwsgi_malloc(dst_len);
char *ptr = dst;
uint8_t *src = (uint8_t *) authorization;
while(i > 3) {
*ptr++= (char) ( table64[src[0]] << 2 | table64[src[1]] >> 4);
*ptr++= (char) ( table64[src[1]] << 4 | table64[src[2]] >> 2);
*ptr++= (char) ( table64[src[2]] << 6 | table64[src[3]]);
src+=4;
i-=4;
}
if (i > 1) {
*ptr++= (char) ( table64[src[0]] << 2 | table64[src[1]] >> 4);
}
if (i > 2) {
*ptr++= (char) ( table64[src[1]] << 4 | table64[src[2]] >> 2);
}
*ptr++= 0;
return dst;
}
static uint16_t htpasswd_check(char *filename, char *auth) {
char line[1024];
@@ -143,7 +77,8 @@ int uwsgi_routing_func_basicauth(struct wsgi_request *wsgi_req, struct uwsgi_rou
if (strncmp(wsgi_req->authorization, "Basic ", 6))
goto forbidden;
char *auth = http_basic_auth_get(wsgi_req->authorization+6, wsgi_req->authorization_len-6);
size_t auth_len = 0;
char *auth = uwsgi_base64_decode(wsgi_req->authorization+6, wsgi_req->authorization_len-6, &auth_len);
if (auth) {
if (!ur->custom) {
// check htpasswd-like file
uwsgi.h
+3
View File
@@ -3667,6 +3667,9 @@ void uwsgi_additional_header_add(struct wsgi_request *, char *, uint16_t);
void uwsgi_proto_hooks_setup(void);
char *uwsgi_base64_decode(char *, size_t, size_t *);
char *uwsgi_base64_encode(char *, size_t, size_t *);
void uwsgi_subscribe_all(uint8_t, int);
#define uwsgi_unsubscribe_all() uwsgi_subscribe_all(1, 1)