ziti-nodejs.h

/*
Copyright NetFoundry Inc.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

#ifndef ZITI_ADD_ON_H
#define ZITI_ADD_ON_H

#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <tlsuv/tlsuv.h>
#include <tlsuv/http.h>
#include <tlsuv/websocket.h>
#include <tlsuv/tcp_src.h>

#include <node_version.h>
#define NAPI_EXPERIMENTAL
#include <node_api.h>

#include <ziti/ziti.h>
#include <ziti/ziti_log.h>
#include "utils.h"


#define NEWP(var, type) type *var = calloc(1, sizeof(type))


#ifdef _WIN32
#define _NO_CRT_STDIO_INLINE 1
  /* Windows - set up dll import/export decorators. */
# if defined(BUILDING_UV_SHARED)
    /* Building shared library. */
#   define UV_EXTERN __declspec(dllexport)
# elif defined(USING_UV_SHARED)
    /* Using shared library. */
#   define UV_EXTERN __declspec(dllimport)
# else
    /* Building static library. */
#   define UV_EXTERN /* nothing */
# endif


#define strcasecmp _stricmp
#define strncasecmp _strnicmp

#if !defined (strndup_DEFINED)
#define strndup_DEFINED
static char* strndup(char* p, size_t len) {
    char *s = malloc(len + 1);
    strncpy(s, p, len);
    s[len] = '\0';
    return s;
}
#endif // strndup_DEFINED


#elif __GNUC__ >= 4
# define UV_EXTERN __attribute__((visibility("default")))
#else
# define UV_EXTERN /* nothing */
#endif

#  ifdef NODE_MAJOR_VERSION
#    if NODE_MAJOR_VERSION == 11
UV_EXTERN int uv_gettimeofday(uv_timeval_t* tv);
#    else
UV_EXTERN int uv_gettimeofday(uv_timeval64_t* tv);
#    endif
#  endif


/**
 * 
 */
typedef struct {
  napi_async_work work;
  napi_threadsafe_function tsfn_on_enroll;
} EnrollAddonData;


/**
 * 
 */
typedef struct {
  bool isWebsocket;
  napi_async_work work;
  napi_threadsafe_function tsfn_on_connect;
  napi_threadsafe_function tsfn_on_data;
  napi_threadsafe_function tsfn_on_write;
  napi_threadsafe_function tsfn_on_service_available;
} ConnAddonData;

/**
 * 
 */
typedef struct {
  char *service_name;
  int64_t js_arb_data;
  ziti_connection server;
  napi_async_work work;
  napi_threadsafe_function tsfn_on_listen;
  napi_threadsafe_function tsfn_on_listen_client;
  napi_threadsafe_function tsfn_on_listen_client_connect;
  napi_threadsafe_function tsfn_on_listen_client_data;
} ListenAddonData;

/**
 * 
 */
typedef struct {
  napi_async_work work;
  napi_threadsafe_function tsfn_on_connect;
  napi_threadsafe_function tsfn_on_data;
  napi_threadsafe_function tsfn_on_write;
  tlsuv_src_t ziti_src;
  tlsuv_websocket_t ws;
  uv_connect_t req;
  uint32_t headers_array_length;
  char* header_name[100];
  char* header_value[100];
  char* service;
} WSAddonData;


// An item that will be passed into the JavaScript on_resp callback
typedef struct HttpsRespItem {
  tlsuv_http_req_t *req;
  int code;
  char* status;
  tlsuv_http_hdr *headers;
} HttpsRespItem;

// An item that will be passed into the JavaScript on_resp_body callback
typedef struct HttpsRespBodyItem {
  tlsuv_http_req_t *req;
  const void *body;
  ssize_t len;
} HttpsRespBodyItem;

// An item that will be passed into the JavaScript on_req_body callback
typedef struct HttpsReqBodyItem {
  tlsuv_http_req_t *req;
  const void *body;
  ssize_t status;
} HttpsReqBodyItem;


typedef struct HttpsAddonData HttpsAddonData;

typedef struct HttpsReq {
  tlsuv_http_req_t *req;
  bool on_resp_has_fired;
  int respCode;
  HttpsAddonData *addon_data;
} HttpsReq;

typedef struct {
  char* scheme_host_port;
  tlsuv_http_t client;
  tlsuv_src_t ziti_src;
  bool active;
  bool purge;
} HttpsClient;

struct HttpsAddonData {
  napi_env env;
  tlsuv_http_t client;
  tlsuv_http_req_t ziti_src;
  napi_threadsafe_function tsfn_on_req;
  napi_threadsafe_function tsfn_on_resp;
  napi_threadsafe_function tsfn_on_resp_body;
  napi_threadsafe_function tsfn_on_req_body;
  HttpsRespItem* item;
  HttpsReq* httpsReq;
  uv_work_t uv_req;
  bool haveURL;
  char* service;
  char* scheme_host_port;
  char* method;
  char* path;
  uint32_t headers_array_length;
  char* header_name[100];
  char* header_value[100];
  HttpsClient* httpsClient;
} ;


#ifdef __cplusplus
extern "C" {
#endif

extern ziti_context ztx;
extern uv_loop_t *thread_loop;

extern uv_mutex_t client_pool_lock;

// extern void set_signal_handler();

extern void expose_ziti_close(napi_env env, napi_value exports);
extern void expose_ziti_dial(napi_env env, napi_value exports);
extern void expose_ziti_enroll(napi_env env, napi_value exports);
extern void expose_ziti_sdk_version(napi_env env, napi_value exports);
extern void expose_ziti_init(napi_env env, napi_value exports);
extern void expose_ziti_listen(napi_env env, napi_value exports);
extern void expose_ziti_service_available(napi_env env, napi_value exports);
extern void expose_ziti_services_refresh(napi_env env, napi_value exports);
extern void expose_ziti_set_log_level(napi_env env, napi_value exports);
extern void expose_ziti_shutdown(napi_env env, napi_value exports);
extern void expose_ziti_write(napi_env env, napi_value exports);
extern void expose_ziti_https_request(napi_env env, napi_value exports);
extern void expose_ziti_https_request_data(napi_env env, napi_value exports);
extern void expose_ziti_https_request_end(napi_env env, napi_value exports);
extern void expose_ziti_websocket_connect(napi_env env, napi_value exports);
extern void expose_ziti_websocket_write(napi_env env, napi_value exports);

//
extern int tlsuv_websocket_init_with_src (uv_loop_t *loop, tlsuv_websocket_t *ws, tlsuv_src_t *src);

extern void track_service_to_hostname(const char* service_name, char* hostname, int port);

#ifdef __cplusplus
}
#endif

#endif /* ZITI_ADD_ON_H */