/*
* Copyright (C) NGINX, Inc.
*/
#include <nxt_auto_config.h>
#include <nxt_unit.h>
#include <nxt_unit_request.h>
#include <jni.h>
#include <stdio.h>
#include <stdlib.h>
#include "nxt_jni.h"
#include "nxt_jni_Request.h"
#include "nxt_jni_URLClassLoader.h"
#include "nxt_jni_HeadersEnumeration.h"
#include "nxt_jni_HeaderNamesEnumeration.h"
static jstring JNICALL nxt_java_Request_getHeader(JNIEnv *env, jclass cls,
jlong req_ptr, jstring name, jint name_len);
static jobject JNICALL nxt_java_Request_getHeaderNames(JNIEnv *env, jclass cls,
jlong req_ptr);
static jobject JNICALL nxt_java_Request_getHeaders(JNIEnv *env, jclass cls,
jlong req_ptr, jstring name, jint name_len);
static jint JNICALL nxt_java_Request_getIntHeader(JNIEnv *env, jclass cls,
jlong req_ptr, jstring name, jint name_len);
static jstring JNICALL nxt_java_Request_getMethod(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getQueryString(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getRequestURI(JNIEnv *env, jclass cls,
jlong req_ptr);
static jlong JNICALL nxt_java_Request_getContentLength(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getContentType(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getLocalAddr(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getLocalName(JNIEnv *env, jclass cls,
jlong req_ptr);
static jint JNICALL nxt_java_Request_getLocalPort(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getProtocol(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getRemoteAddr(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getRemoteHost(JNIEnv *env, jclass cls,
jlong req_ptr);
static jint JNICALL nxt_java_Request_getRemotePort(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getScheme(JNIEnv *env, jclass cls,
jlong req_ptr);
static jstring JNICALL nxt_java_Request_getServerName(JNIEnv *env, jclass cls,
jlong req_ptr);
static jint JNICALL nxt_java_Request_getServerPort(JNIEnv *env, jclass cls,
jlong req_ptr);
static jboolean JNICALL nxt_java_Request_isSecure(JNIEnv *env, jclass cls,
jlong req_ptr);
static void JNICALL nxt_java_Request_upgrade(JNIEnv *env, jclass cls,
jlong req_info_ptr);
static jboolean JNICALL nxt_java_Request_isUpgrade(JNIEnv *env, jclass cls,
jlong req_info_ptr);
static void JNICALL nxt_java_Request_log(JNIEnv *env, jclass cls,
jlong req_info_ptr, jstring msg, jint msg_len);
static void JNICALL nxt_java_Request_trace(JNIEnv *env, jclass cls,
jlong req_info_ptr, jstring msg, jint msg_len);
static jobject JNICALL nxt_java_Request_getResponse(JNIEnv *env, jclass cls,
jlong req_info_ptr);
static void JNICALL nxt_java_Request_sendWsFrameBuf(JNIEnv *env, jclass cls,
jlong req_info_ptr, jobject buf, jint pos, jint len, jbyte opCode, jboolean last);
static void JNICALL nxt_java_Request_sendWsFrameArr(JNIEnv *env, jclass cls,
jlong req_info_ptr, jarray arr, jint pos, jint len, jbyte opCode, jboolean last);
static void JNICALL nxt_java_Request_closeWs(JNIEnv *env, jclass cls,
jlong req_info_ptr);
static jclass nxt_java_Request_class;
static jmethodID nxt_java_Request_ctor;
static jmethodID nxt_java_Request_processWsFrame;
static jmethodID nxt_java_Request_closeWsSession;
int
nxt_java_initRequest(JNIEnv *env, jobject cl)
{
int res;
jclass cls;
cls = nxt_java_loadClass(env, cl, "nginx.unit.Request");
if (cls == NULL) {
return NXT_UNIT_ERROR;
}
nxt_java_Request_class = (*env)->NewGlobalRef(env, cls);
(*env)->DeleteLocalRef(env, cls);
cls = nxt_java_Request_class;
nxt_java_Request_ctor = (*env)->GetMethodID(env, cls, "<init>", "(Lnginx/unit/Context;JJ)V");
if (nxt_java_Request_ctor == NULL) {
(*env)->DeleteGlobalRef(env, cls);
return NXT_UNIT_ERROR;
}
nxt_java_Request_processWsFrame = (*env)->GetMethodID(env, cls, "processWsFrame", "(Ljava/nio/ByteBuffer;BZ)V");
if (nxt_java_Request_processWsFrame == NULL) {
(*env)->DeleteGlobalRef(env, cls);
return NXT_UNIT_ERROR;
}
nxt_java_Request_closeWsSession = (*env)->GetMethodID(env, cls, "closeWsSession", "()V");
if (nxt_java_Request_closeWsSession == NULL) {
(*env)->DeleteGlobalRef(env, cls);
return NXT_UNIT_ERROR;
}
JNINativeMethod request_methods[] = {
{ (char *) "getHeader",
(char *) "(JLjava/lang/String;I)Ljava/lang/String;",
nxt_java_Request_getHeader },
{ (char *) "getHeaderNames",
(char *) "(J)Ljava/util/Enumeration;",
nxt_java_Request_getHeaderNames },
{ (char *) "getHeaders",
(char *) "(JLjava/lang/String;I)Ljava/util/Enumeration;",
nxt_java_Request_getHeaders },
{ (char *) "getIntHeader",
(char *) "(JLjava/lang/String;I)I",
nxt_java_Request_getIntHeader },
{ (char *) "getMethod",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getMethod },
{ (char *) "getQueryString",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getQueryString },
{ (char *) "getRequestURI",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getRequestURI },
{ (char *) "getContentLength",
(char *) "(J)J",
nxt_java_Request_getContentLength },
{ (char *) "getContentType",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getContentType },
{ (char *) "getLocalAddr",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getLocalAddr },
{ (char *) "getLocalName",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getLocalName },
{ (char *) "getLocalPort",
(char *) "(J)I",
nxt_java_Request_getLocalPort },
{ (char *) "getProtocol",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getProtocol },
{ (char *) "getRemoteAddr",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getRemoteAddr },
{ (char *) "getRemoteHost",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getRemoteHost },
{ (char *) "getRemotePort",
(char *) "(J)I",
nxt_java_Request_getRemotePort },
{ (char *) "getScheme",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getScheme },
{ (char *) "getServerName",
(char *) "(J)Ljava/lang/String;",
nxt_java_Request_getServerName },
{ (char *) "getServerPort",
(char *) "(J)I",
nxt_java_Request_getServerPort },
{ (char *) "isSecure",
(char *) "(J)Z",
nxt_java_Request_isSecure },
{ (char *) "upgrade",
(char *) "(J)V",
nxt_java_Request_upgrade },
{ (char *) "isUpgrade",
(char *) "(J)Z",
nxt_java_Request_isUpgrade },
{ (char *) "log",
(char *) "(JLjava/lang/String;I)V",
nxt_java_Request_log },
{ (char *) "trace",
(char *) "(JLjava/lang/String;I)V",
nxt_java_Request_trace },
{ (char *) "getResponse",
(char *) "(J)Lnginx/unit/Response;",
nxt_java_Request_getResponse },
{ (char *) "sendWsFrame",
(char *) "(JLjava/nio/ByteBuffer;IIBZ)V",
nxt_java_Request_sendWsFrameBuf },
{ (char *) "sendWsFrame",
(char *) "(J[BIIBZ)V",
nxt_java_Request_sendWsFrameArr },
{ (char *) "closeWs",
(char *) "(J)V",
nxt_java_Request_closeWs },
};
res = (*env)->RegisterNatives(env, nxt_java_Request_class,
request_methods,
sizeof(request_methods) / sizeof(request_methods[0]));
nxt_unit_debug(NULL, "registered Request methods: %d", res);
if (res != 0) {
nxt_unit_warn(NULL, "registering natives for Request failed");
goto failed;
}
res = nxt_java_initHeadersEnumeration(env, cl);
if (res != NXT_UNIT_OK) {
goto failed;
}
res = nxt_java_initHeaderNamesEnumeration(env, cl);
if (res != NXT_UNIT_OK) {
goto failed;
}
return NXT_UNIT_OK;
failed:
(*env)->DeleteGlobalRef(env, cls);
return NXT_UNIT_ERROR;
}
jobject
nxt_java_newRequest(JNIEnv *env, jobject ctx, nxt_unit_request_info_t *req)
{
return (*env)->NewObject(env, nxt_java_Request_class,
nxt_java_Request_ctor, ctx, nxt_ptr2jlong(req),
nxt_ptr2jlong(req->request));
}
static jstring JNICALL
nxt_java_Request_getHeader(JNIEnv *env, jclass cls, jlong req_ptr,
jstring name, jint name_len)
{
const char *name_str;
nxt_unit_field_t *f;
nxt_unit_request_t *r;
name_str = (*env)->GetStringUTFChars(env, name, NULL);
if (name_str == NULL) {
return NULL;
}
r = nxt_jlong2ptr(req_ptr);
f = nxt_java_findHeader(r->fields, r->fields + r->fields_count,
name_str, name_len);
(*env)->ReleaseStringUTFChars(env, name, name_str);
if (f == NULL) {
return NULL;
}
return (*env)->NewStringUTF(env, nxt_unit_sptr_get(&f->value));
}
static jobject JNICALL
nxt_java_Request_getHeaderNames(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return nxt_java_newHeaderNamesEnumeration(env, r->fields, r->fields_count);
}
static jobject JNICALL
nxt_java_Request_getHeaders(JNIEnv *env, jclass cls, jlong req_ptr,
jstring name, jint name_len)
{
const char *name_str;
nxt_unit_field_t *f;
nxt_unit_request_t *r;
name_str = (*env)->GetStringUTFChars(env, name, NULL);
if (name_str == NULL) {
return NULL;
}
r = nxt_jlong2ptr(req_ptr);
f = nxt_java_findHeader(r->fields, r->fields + r->fields_count,
name_str, name_len);
(*env)->ReleaseStringUTFChars(env, name, name_str);
if (f == NULL) {
f = r->fields + r->fields_count;
}
return nxt_java_newHeadersEnumeration(env, r->fields, r->fields_count,
f - r->fields);
}
static jint JNICALL
nxt_java_Request_getIntHeader(JNIEnv *env, jclass cls, jlong req_ptr,
jstring name, jint name_len)
{
jint res;
char *value, *end;
const char *name_str;
nxt_unit_field_t *f;
nxt_unit_request_t *r;
res = -1;
name_str = (*env)->GetStringUTFChars(env, name, NULL);
if (name_str == NULL) {
return res;
}
r = nxt_jlong2ptr(req_ptr);
f = nxt_java_findHeader(r->fields, r->fields + r->fields_count,
name_str, name_len);
(*env)->ReleaseStringUTFChars(env, name, name_str);
if (f == NULL) {
return res;
}
value = nxt_unit_sptr_get(&f->value);
end = value + f->value_length;
res = strtol(value, &end, 10);
if (end < value + f->value_length) {
// TODO throw NumberFormatException.forInputString(value)
}
return res;
}
static jstring JNICALL
nxt_java_Request_getMethod(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return (*env)->NewStringUTF(env, nxt_unit_sptr_get(&r->method));
}
static jstring JNICALL
nxt_java_Request_getQueryString(JNIEnv *env, jclass cls, jlong req_ptr)
{
char *query;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
if (r->query.offset != 0) {
query = nxt_unit_sptr_get(&r->query);
return (*env)->NewStringUTF(env, query);
}
return NULL;
}
static jstring JNICALL
nxt_java_Request_getRequestURI(JNIEnv *env, jclass cls, jlong req_ptr)
{
char *target, *query;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
target = nxt_unit_sptr_get(&r->target);
if (r->query.offset != 0) {
query = nxt_unit_sptr_get(&r->query);
return nxt_java_newString(env, target, query - target - 1);
}
return (*env)->NewStringUTF(env, target);
}
static jlong JNICALL
nxt_java_Request_getContentLength(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return r->content_length;
}
static jstring JNICALL
nxt_java_Request_getContentType(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_field_t *f;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
if (r->content_type_field != NXT_UNIT_NONE_FIELD) {
f = r->fields + r->content_type_field;
return (*env)->NewStringUTF(env, nxt_unit_sptr_get(&f->value));
}
return NULL;
}
static jstring JNICALL
nxt_java_Request_getLocalAddr(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return nxt_java_newString(env, nxt_unit_sptr_get(&r->local_addr),
r->local_addr_length);
}
static jstring JNICALL
nxt_java_Request_getLocalName(JNIEnv *env, jclass cls, jlong req_ptr)
{
char *local, *colon;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
local = nxt_unit_sptr_get(&r->local_addr);
colon = memchr(local, ':', r->local_addr_length);
if (colon == NULL) {
colon = local + r->local_addr_length;
}
return nxt_java_newString(env, local, colon - local);
}
static jint JNICALL
nxt_java_Request_getLocalPort(JNIEnv *env, jclass cls, jlong req_ptr)
{
jint res;
char *local, *colon, tmp;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
local = nxt_unit_sptr_get(&r->local_addr);
colon = memchr(local, ':', r->local_addr_length);
if (colon == NULL) {
return 80;
}
tmp = local[r->local_addr_length];
local[r->local_addr_length] = '\0';
res = strtol(colon + 1, NULL, 10);
local[r->local_addr_length] = tmp;
return res;
}
static jstring JNICALL
nxt_java_Request_getProtocol(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return (*env)->NewStringUTF(env, nxt_unit_sptr_get(&r->version));
}
static jstring JNICALL
nxt_java_Request_getRemoteAddr(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return nxt_java_newString(env, nxt_unit_sptr_get(&r->remote),
r->remote_length);
}
static jstring JNICALL
nxt_java_Request_getRemoteHost(JNIEnv *env, jclass cls, jlong req_ptr)
{
char *remote, *colon;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
remote = nxt_unit_sptr_get(&r->remote);
colon = memchr(remote, ':', r->remote_length);
if (colon == NULL) {
colon = remote + r->remote_length;
}
return nxt_java_newString(env, remote, colon - remote);
}
static jint JNICALL
nxt_java_Request_getRemotePort(JNIEnv *env, jclass cls, jlong req_ptr)
{
jint res;
char *remote, *colon, tmp;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
remote = nxt_unit_sptr_get(&r->remote);
colon = memchr(remote, ':', r->remote_length);
if (colon == NULL) {
return 80;
}
tmp = remote[r->remote_length];
remote[r->remote_length] = '\0';
res = strtol(colon + 1, NULL, 10);
remote[r->remote_length] = tmp;
return res;
}
static jstring JNICALL
nxt_java_Request_getScheme(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return (*env)->NewStringUTF(env, r->tls ? "https" : "http");
}
static jstring JNICALL
nxt_java_Request_getServerName(JNIEnv *env, jclass cls, jlong req_ptr)
{
char *host, *colon;
nxt_unit_field_t *f;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
f = nxt_java_findHeader(r->fields, r->fields + r->fields_count,
"Host", 4);
if (f != NULL) {
host = nxt_unit_sptr_get(&f->value);
colon = memchr(host, ':', f->value_length);
if (colon == NULL) {
colon = host + f->value_length;
}
return nxt_java_newString(env, host, colon - host);
}
return nxt_java_Request_getLocalName(env, cls, req_ptr);
}
static jint JNICALL
nxt_java_Request_getServerPort(JNIEnv *env, jclass cls, jlong req_ptr)
{
jint res;
char *host, *colon, tmp;
nxt_unit_field_t *f;
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
f = nxt_java_findHeader(r->fields, r->fields + r->fields_count,
"Host", 4);
if (f != NULL) {
host = nxt_unit_sptr_get(&f->value);
colon = memchr(host, ':', f->value_length);
if (colon == NULL) {
return 80;
}
tmp = host[f->value_length];
host[f->value_length] = '\0';
res = strtol(colon + 1, NULL, 10);
host[f->value_length] = tmp;
return res;
}
return nxt_java_Request_getLocalPort(env, cls, req_ptr);
}
static jboolean JNICALL
nxt_java_Request_isSecure(JNIEnv *env, jclass cls, jlong req_ptr)
{
nxt_unit_request_t *r;
r = nxt_jlong2ptr(req_ptr);
return r->tls != 0;
}
static void JNICALL
nxt_java_Request_upgrade(JNIEnv *env, jclass cls, jlong req_info_ptr)
{
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
if (!nxt_unit_response_is_init(req)) {
nxt_unit_response_init(req, 101, 0, 0);
}
(void) nxt_unit_response_upgrade(req);
}
static jboolean JNICALL
nxt_java_Request_isUpgrade(JNIEnv *env, jclass cls, jlong req_info_ptr)
{
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
return nxt_unit_request_is_websocket_handshake(req);
}
static void JNICALL
nxt_java_Request_log(JNIEnv *env, jclass cls, jlong req_info_ptr, jstring msg,
jint msg_len)
{
const char *msg_str;
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
msg_str = (*env)->GetStringUTFChars(env, msg, NULL);
if (msg_str == NULL) {
return;
}
nxt_unit_req_log(req, NXT_UNIT_LOG_INFO, "%.*s", msg_len, msg_str);
(*env)->ReleaseStringUTFChars(env, msg, msg_str);
}
static void JNICALL
nxt_java_Request_trace(JNIEnv *env, jclass cls, jlong req_info_ptr, jstring msg,
jint msg_len)
{
#if (NXT_DEBUG)
const char *msg_str;
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
msg_str = (*env)->GetStringUTFChars(env, msg, NULL);
if (msg_str == NULL) {
return;
}
nxt_unit_req_debug(req, "%.*s", msg_len, msg_str);
(*env)->ReleaseStringUTFChars(env, msg, msg_str);
#endif
}
static jobject JNICALL
nxt_java_Request_getResponse(JNIEnv *env, jclass cls, jlong req_info_ptr)
{
nxt_unit_request_info_t *req;
nxt_java_request_data_t *data;
req = nxt_jlong2ptr(req_info_ptr);
data = req->data;
return data->jresp;
}
static void JNICALL
nxt_java_Request_sendWsFrameBuf(JNIEnv *env, jclass cls,
jlong req_info_ptr, jobject buf, jint pos, jint len, jbyte opCode, jboolean last)
{
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
uint8_t *b = (*env)->GetDirectBufferAddress(env, buf);
if (b != NULL) {
nxt_unit_websocket_send(req, opCode, last, b + pos, len);
} else {
nxt_unit_req_debug(req, "sendWsFrameBuf: b == NULL");
}
}
static void JNICALL
nxt_java_Request_sendWsFrameArr(JNIEnv *env, jclass cls,
jlong req_info_ptr, jarray arr, jint pos, jint len, jbyte opCode, jboolean last)
{
nxt_unit_request_info_t *req;
req = nxt_jlong2ptr(req_info_ptr);
uint8_t *b = (*env)->GetPrimitiveArrayCritical(env, arr, NULL);
if (b != NULL) {
if (!nxt_unit_response_is_sent(req)) {
nxt_unit_response_send(req);
}
nxt_unit_websocket_send(req, opCode, last, b + pos, len);
(*env)->ReleasePrimitiveArrayCritical(env, arr, b, 0);
} else {
nxt_unit_req_debug(req, "sendWsFrameArr: b == NULL");
}
}
static void JNICALL
nxt_java_Request_closeWs(JNIEnv *env, jclass cls, jlong req_info_ptr)
{
nxt_unit_request_info_t *req;
nxt_java_request_data_t *data;
req = nxt_jlong2ptr(req_info_ptr);
data = req->data;
(*env)->DeleteGlobalRef(env, data->jresp);
(*env)->DeleteGlobalRef(env, data->jreq);
nxt_unit_request_done(req, NXT_UNIT_OK);
}
void
nxt_java_Request_websocket(JNIEnv *env, jobject jreq, jobject jbuf,
uint8_t opcode, uint8_t fin)
{
(*env)->CallVoidMethod(env, jreq, nxt_java_Request_processWsFrame, jbuf, opcode, fin);
}
void
nxt_java_Request_close(JNIEnv *env, jobject jreq)
{
(*env)->CallVoidMethod(env, jreq, nxt_java_Request_closeWsSession);
}