src/php/ext/grpc/server.c

#include "call.h"

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "php.h"
#include "php_ini.h"
#include "ext/standard/info.h"
#include "ext/spl/spl_exceptions.h"
#include "php_grpc.h"

#include "zend_exceptions.h"

#include <stdbool.h>

#include "grpc/grpc.h"
#include "grpc/support/log.h"
#include "grpc/grpc_security.h"

#include "server.h"
#include "completion_queue.h"
#include "channel.h"
#include "server_credentials.h"

/* Frees and destroys an instance of wrapped_grpc_server */
void free_wrapped_grpc_server(void *object TSRMLS_DC){
  wrapped_grpc_server *server = (wrapped_grpc_server*)object;
  if(server->wrapped != NULL){
    grpc_server_shutdown(server->wrapped);
    grpc_server_destroy(server->wrapped);
  }
  efree(server);
}

/* Initializes an instance of wrapped_grpc_call to be associated with an object
 * of a class specified by class_type */
zend_object_value create_wrapped_grpc_server(
    zend_class_entry *class_type TSRMLS_DC){
  zend_object_value retval;
  wrapped_grpc_server *intern;

  intern = (wrapped_grpc_server*)emalloc(sizeof(wrapped_grpc_server));
  memset(intern, 0, sizeof(wrapped_grpc_server));

  zend_object_std_init(&intern->std, class_type TSRMLS_CC);
  object_properties_init(&intern->std, class_type);
  retval.handle = zend_objects_store_put(
      intern,
      (zend_objects_store_dtor_t) zend_objects_destroy_object,
      free_wrapped_grpc_server,
      NULL TSRMLS_CC);
  retval.handlers = zend_get_std_object_handlers();
  return retval;
}

/**
 * Constructs a new instance of the Server class
 * @param CompletionQueue $queue The completion queue to use with the server
 * @param array $args The arguments to pass to the server (optional)
 */
PHP_METHOD(Server, __construct){
  wrapped_grpc_server *server =
    (wrapped_grpc_server*)zend_object_store_get_object(getThis() TSRMLS_CC);
  zval *queue_obj;
  zval *args_array = NULL;
  grpc_channel_args args;
  HashTable *array_hash;
  zval **creds_obj = NULL;
  wrapped_grpc_server_credentials *creds = NULL;
  /* "O|a" == 1 Object, 1 optional array */
  if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC,
                           "O|a",
                           &queue_obj, grpc_ce_completion_queue,
                           &args_array) == FAILURE){
    zend_throw_exception(spl_ce_InvalidArgumentException,
                         "Server expects a CompletionQueue and an array",
                         1 TSRMLS_CC);
    return;
  }
  add_property_zval(getThis(), "completion_queue", queue_obj);
  wrapped_grpc_completion_queue *queue =
    (wrapped_grpc_completion_queue*)zend_object_store_get_object(
        queue_obj TSRMLS_CC);
  if (args_array == NULL) {
    server->wrapped = grpc_server_create(queue->wrapped, NULL);
  } else {
    array_hash = Z_ARRVAL_P(args_array);
    if(zend_hash_find(array_hash,
                      "credentials",
                      sizeof("credentials"),
                      (void**)&creds_obj) == SUCCESS) {
      if(zend_get_class_entry(*creds_obj TSRMLS_CC) !=
         grpc_ce_server_credentials) {
        zend_throw_exception(spl_ce_InvalidArgumentException,
                             "credentials must be a ServerCredentials object",
                             1 TSRMLS_CC);
        return;
      }
      creds = (wrapped_grpc_server_credentials*)zend_object_store_get_object(
          *creds_obj TSRMLS_CC);
      zend_hash_del(array_hash, "credentials", sizeof("credentials"));
    }
    php_grpc_read_args_array(args_array, &args);
    if (creds == NULL) {
      server->wrapped = grpc_server_create(queue->wrapped, &args);
    } else {
      gpr_log(GPR_DEBUG, "Initialized secure server");
      server->wrapped = grpc_secure_server_create(creds->wrapped,
                                                  queue->wrapped,
                                                  &args);
    }
    efree(args.args);
  }
}

/**
 * Request a call on a server. Creates a single GRPC_SERVER_RPC_NEW event.
 * @param long $tag_new The tag to associate with the new request
 * @param long $tag_cancel The tag to use if the call is cancelled
 * @return Void
 */
PHP_METHOD(Server, request_call){
  wrapped_grpc_server *server =
    (wrapped_grpc_server*)zend_object_store_get_object(getThis() TSRMLS_CC);
  long tag_new;
  /* "l" == 1 long */
  if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC,
                           "l",
                           &tag_new) == FAILURE){
    zend_throw_exception(spl_ce_InvalidArgumentException,
                         "request_call expects a long",
                         1 TSRMLS_CC);
    return;
  }
  grpc_server_request_call(server->wrapped, (void*)tag_new);
}

/**
 * Add a http2 over tcp listener.
 * @param string $addr The address to add
 * @return true on success, false on failure
 */
PHP_METHOD(Server, add_http2_port){
  wrapped_grpc_server *server =
    (wrapped_grpc_server*)zend_object_store_get_object(getThis() TSRMLS_CC);
  const char *addr;
  int addr_len;
  /* "s" == 1 string */
  if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC,
                           "s",
                           &addr, &addr_len) == FAILURE){
    zend_throw_exception(spl_ce_InvalidArgumentException,
                         "add_http2_port expects a string",
                         1 TSRMLS_CC);
    return;
  }
  RETURN_BOOL(grpc_server_add_http2_port(server->wrapped, addr));
}

PHP_METHOD(Server, add_secure_http2_port){
  wrapped_grpc_server *server =
    (wrapped_grpc_server*)zend_object_store_get_object(getThis() TSRMLS_CC);
  const char *addr;
  int addr_len;
  /* "s" == 1 string */
  if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC,
                           "s",
                           &addr, &addr_len) == FAILURE){
    zend_throw_exception(spl_ce_InvalidArgumentException,
                         "add_http2_port expects a string",
                         1 TSRMLS_CC);
    return;
  }
  RETURN_BOOL(grpc_server_add_secure_http2_port(server->wrapped, addr));
}

/**
 * Start a server - tells all listeners to start listening
 * @return Void
 */
PHP_METHOD(Server, start){
  wrapped_grpc_server *server =
    (wrapped_grpc_server*)zend_object_store_get_object(getThis() TSRMLS_CC);
  grpc_server_start(server->wrapped);
}

static zend_function_entry server_methods[] = {
  PHP_ME(Server, __construct, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
  PHP_ME(Server, request_call, NULL, ZEND_ACC_PUBLIC)
  PHP_ME(Server, add_http2_port, NULL, ZEND_ACC_PUBLIC)
  PHP_ME(Server, add_secure_http2_port, NULL, ZEND_ACC_PUBLIC)
  PHP_ME(Server, start, NULL, ZEND_ACC_PUBLIC)
  PHP_FE_END
};

void grpc_init_server(TSRMLS_D){
  zend_class_entry ce;
  INIT_CLASS_ENTRY(ce, "Grpc\\Server", server_methods);
  ce.create_object = create_wrapped_grpc_server;
  grpc_ce_server = zend_register_internal_class(&ce TSRMLS_CC);
}