technocloud-public/nginx-unit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

NJS: added http request prototype.

Browse Source
This commit is contained in:
Zhidao HONG
2022-11-22 10:13:18 +08:00
parent 4d6d146e92
commit e3bbf5b3b5
7 changed files with 359 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

273
src/nxt_http_js.c Normal file
View File

@@ -0,0 +1,273 @@
/*
* Copyright (C) NGINX, Inc.
*/
#include <nxt_router.h>
#include <nxt_http.h>
#include <njs.h>
static njs_int_t nxt_http_js_ext_uri(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval);
static njs_int_t nxt_http_js_ext_host(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval);
static njs_int_t nxt_http_js_ext_remote_addr(njs_vm_t *vm,
njs_object_prop_t *prop, njs_value_t *value, njs_value_t *setval,
njs_value_t *retval);
static njs_int_t nxt_http_js_ext_get_arg(njs_vm_t *vm,
njs_object_prop_t *prop, njs_value_t *value, njs_value_t *setval,
njs_value_t *retval);
static njs_int_t nxt_http_js_ext_get_header(njs_vm_t *vm,
njs_object_prop_t *prop, njs_value_t *value, njs_value_t *setval,
njs_value_t *retval);
static njs_int_t nxt_http_js_ext_get_cookie(njs_vm_t *vm,
njs_object_prop_t *prop, njs_value_t *value, njs_value_t *setval,
njs_value_t *retval);
static njs_external_t nxt_http_js_proto[] = {
{
.flags = NJS_EXTERN_PROPERTY,
.name.string = njs_str("uri"),
.enumerable = 1,
.u.property = {
.handler = nxt_http_js_ext_uri,
}
},
{
.flags = NJS_EXTERN_PROPERTY,
.name.string = njs_str("host"),
.enumerable = 1,
.u.property = {
.handler = nxt_http_js_ext_host,
}
},
{
.flags = NJS_EXTERN_PROPERTY,
.name.string = njs_str("remoteAddr"),
.enumerable = 1,
.u.property = {
.handler = nxt_http_js_ext_remote_addr,
}
},
{
.flags = NJS_EXTERN_OBJECT,
.name.string = njs_str("args"),
.enumerable = 1,
.u.object = {
.enumerable = 1,
.prop_handler = nxt_http_js_ext_get_arg,
}
},
{
.flags = NJS_EXTERN_OBJECT,
.name.string = njs_str("headers"),
.enumerable = 1,
.u.object = {
.enumerable = 1,
.prop_handler = nxt_http_js_ext_get_header,
}
},
{
.flags = NJS_EXTERN_OBJECT,
.name.string = njs_str("cookies"),
.enumerable = 1,
.u.object = {
.enumerable = 1,
.prop_handler = nxt_http_js_ext_get_cookie,
}
},
};
void
nxt_http_register_js_proto(nxt_js_conf_t *jcf)
{
nxt_js_set_proto(jcf, nxt_http_js_proto, njs_nitems(nxt_http_js_proto));
}
static njs_int_t
nxt_http_js_ext_uri(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
nxt_http_request_t *r;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
return njs_vm_value_string_set(vm, retval, r->path->start, r->path->length);
}
static njs_int_t
nxt_http_js_ext_host(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
nxt_http_request_t *r;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
return njs_vm_value_string_set(vm, retval, r->host.start, r->host.length);
}
static njs_int_t
nxt_http_js_ext_remote_addr(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
nxt_http_request_t *r;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
return njs_vm_value_string_set(vm, retval,
nxt_sockaddr_address(r->remote),
r->remote->address_length);
}
static njs_int_t
nxt_http_js_ext_get_arg(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
njs_int_t rc;
njs_str_t key;
nxt_array_t *args;
nxt_http_request_t *r;
nxt_http_name_value_t *nv, *start, *end;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
rc = njs_vm_prop_name(vm, prop, &key);
if (rc != NJS_OK) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
args = nxt_http_arguments_parse(r);
if (nxt_slow_path(args == NULL)) {
return NJS_ERROR;
}
start = args->elts;
end = start + args->nelts;
for (nv = start; nv < end; nv++) {
if (key.length == nv->name_length
&& memcmp(key.start, nv->name, nv->name_length) == 0)
{
return njs_vm_value_string_set(vm, retval, nv->value,
nv->value_length);
}
}
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
static njs_int_t
nxt_http_js_ext_get_header(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
njs_int_t rc;
njs_str_t key;
nxt_http_field_t *f;
nxt_http_request_t *r;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
rc = njs_vm_prop_name(vm, prop, &key);
if (rc != NJS_OK) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
nxt_list_each(f, r->fields) {
if (key.length == f->name_length
&& memcmp(key.start, f->name, f->name_length) == 0)
{
return njs_vm_value_string_set(vm, retval, f->value,
f->value_length);
}
} nxt_list_loop;
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
static njs_int_t
nxt_http_js_ext_get_cookie(njs_vm_t *vm, njs_object_prop_t *prop,
njs_value_t *value, njs_value_t *setval, njs_value_t *retval)
{
njs_int_t rc;
njs_str_t key;
nxt_array_t *cookies;
nxt_http_request_t *r;
nxt_http_name_value_t *nv, *start, *end;
r = njs_vm_external(vm, nxt_js_proto_id, value);
if (r == NULL) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
rc = njs_vm_prop_name(vm, prop, &key);
if (rc != NJS_OK) {
njs_value_undefined_set(retval);
return NJS_DECLINED;
}
cookies = nxt_http_cookies_parse(r);
if (nxt_slow_path(cookies == NULL)) {
return NJS_ERROR;
}
start = cookies->elts;
end = start + cookies->nelts;
for (nv = start; nv < end; nv++) {
if (key.length == nv->name_length
&& memcmp(key.start, nv->name, nv->name_length) == 0)
{
return njs_vm_value_string_set(vm, retval, nv->value,
nv->value_length);
}
}
njs_value_undefined_set(retval);
return NJS_DECLINED;
}