source: TI05-delivery/trunk/src/python_ext/bbftpd.c @ 1100

/**
 * NDG python embedded bbftp daemon module.
 *
 * @author Stephen Pascoe
 *
 * Copyright (C) 2006 CCLRC & NERC
 *
 * This software may be distributed under the terms of the Q Public Licence, version 1.0 or later.
 *
 */

#include <Python.h>

extern char **environ;
/*
 * variables and prototypes from bbftpd source
 */

extern int be_daemon;
extern int newcontrolport;

int bbftpd_main(int argc, char **argv, char **envp);
int ndg_message_send(char *buffer, int buffertosendlength, char *logmessage) ;
int ndg_message_recv(char **buffer, int *length, char *logmessage) ;

/*
 * Static variables.
 */
static PyObject *authContext = NULL;
static PyObject *authzContext = NULL;

/*---------------------------------------------------------------------------------------------------
 * The following functions interact with bbftpd_private_user.  They are placed here to separate
 * python and bbftp APIs.
 *
 */

/**
 * Make a callback bbftpd->python to handle authentication.
 *
 * @param[out] logmessage filled with an error message on failure
 * @return 0 on success, -1 on failure
 */
int bbftpd_private_auth_callback(char *logmessage) {

  if (authzContext != NULL) {
    sprintf(logmessage, "bbftpd_private_auth_callback: authzContext already present");
    return -1;
  }
  
  if (authContext == NULL) {
    sprintf(logmessage, "bbftpd_private_auth_callback: no authContext set");
    return -1;
  }

  if ((authzContext = PyObject_CallMethod(authContext, "authorise", "")) == NULL) {
    //!TODO: should probably read the exception object to fill logmessage.
    sprintf(logmessage, "bbftpd_private_auth_callback: authContext.authorise() failed");
    return -1;
  }

  /* If authContext.authorise() returned None then authorisation failed.  */
  if (authzContext == Py_None) {
    sprintf(logmessage, "bbftpd_private_auth_callback: python authorisation failed");
    return -1;
  }

  return 0;
}

/**
 * retrieve the username from authzContext.
 *
 * @param[out] logmessage filled with an error message on failure
 * @return pointer to the username or NULL on failure
 */
char *bbftpd_private_auth_getusername(char *logmessage) {
  char *username;
  PyObject *str_obj;

  if (authzContext == NULL) {
    sprintf(logmessage, "bbftpd_private_auth_getusername: no authzContext set");
    return NULL;
  }

  if ((str_obj = PyObject_GetAttrString(authzContext, "username")) == NULL) {
    sprintf(logmessage, "bbftpd_private_auth_getusername: authzContext.username lookup failed");
    return NULL;
  }

  if ((username = PyString_AsString(str_obj)) == NULL) {
    Py_DECREF(str_obj);
    sprintf(logmessage, "bbftpd_private_auth_getusername: authzContext.username not a string");
    return NULL;
  }

  return username;
}
      
/*-------------------------------------------------------------------------------------------------
 * Functions exported to python
 *
 */

/* send(string) */
static PyObject *bbftpd_send(PyObject *self, PyObject *args) {
  char *buffer, logmessage[1024];
  int len;

  if (!PyArg_ParseTuple(args, "s#", &buffer, &len)) {
    return NULL;
  }

  if (authContext == NULL) {
    PyErr_SetString(PyExc_RuntimeError, "Must be called within a bbftpd callback");
    return NULL;
  }

  if (ndg_message_send(buffer, len, logmessage) == -1) {
    PyErr_SetString(PyExc_IOError, logmessage);
    return NULL;
  }

  Py_RETURN_NONE;
}

/* recv(length) -> string */
static PyObject *bbftpd_recv(PyObject *self, PyObject *args) {
  char *buffer, logmessage[1024];
  int len;
  PyObject *ret;

  if (!PyArg_ParseTuple(args, "")) {
    return NULL;
  }

  if (authContext == NULL) {
    PyErr_SetString(PyExc_RuntimeError, "Must be called within a bbftpd callback");
    return NULL;
  }

  if (ndg_message_recv(&buffer, &len, logmessage) == -1) {
    PyErr_SetString(PyExc_IOError, logmessage);
    free(buffer);
    return NULL;
  }

  ret = Py_BuildValue("s#", buffer, len);
  free(buffer);
  return ret;
}



/**
 * Main entry point for the python module.
 */
static PyObject *bbftpd_run(PyObject *self, PyObject *args) {
  int argc, i;
  char **argv, **arg_p;
  int pid;
  char preargs[2][20] = { "bbftpd_embedded", "-b" };
  PyObject *daemon_args, *item;

  if (!PyArg_ParseTuple(args, "OO", &authContext, &daemon_args)) {
    return NULL;
  }
  Py_INCREF(authContext);
  Py_INCREF(daemon_args);

  /*
   * Convert arguments into a standard argv sequence.
   */
  argc = PySequence_Size(daemon_args);
  if ((argv = (char**)malloc((argc + 2)*sizeof(char*))) == NULL) {
    PyErr_SetString(PyExc_MemoryError, "malloc failed");
    Py_DECREF(daemon_args);
    return NULL;
  }

  // Add fixed arguments to daemon's argv
  arg_p = argv;
  *arg_p = preargs[0]; arg_p++;
  *arg_p = preargs[1]; arg_p++;

  for (i=0; i<argc; i++) {
    if ((item = PySequence_GetItem(daemon_args, i)) == NULL) {
      Py_DECREF(authContext);
      Py_DECREF(daemon_args);
      return NULL;
    }
    if ((*arg_p = PyString_AsString(item)) == NULL) {
      free(argv);
      Py_DECREF(item);
      Py_DECREF(authContext);
      Py_DECREF(daemon_args);
      return NULL;
    }
    arg_p++;
    Py_DECREF(item);
  }

  /*
   * We must reset some global variables so that bbftpd_run can be executed multiple times.
   */
  be_daemon = 0;

  pid = bbftpd_main(argc+2, argv, environ);

  free(argv);
  Py_DECREF(authContext); authContext = NULL;
  Py_DECREF(daemon_args);

  return Py_BuildValue("i", pid);
}

static PyMethodDef BbftpdMethods[] = {
  {
    "run", bbftpd_run, METH_VARARGS, 
    "Execute the bbftpd server.\n"
    "\n"
    "Forks an embedded bbftpd process as if it were executed\n"
    "with the command line $ bbftpd -b <args>."
    "\n"
    "@param args: command line arguments to pass to the server\n"
    "@return: the PID of the server process\n"
  },
  {
    "send", bbftpd_send, METH_VARARGS,
    "Send an authentication message to the client.\n"
    "\n"
    "Equivilent to bbftpd_private_send() C call.\n"
    "\n"
    "@param buffer: A string containing the message\n"
    "@raise IOError: if bbftpd_private_send() call fails\n"
  },
  {
    "recv", bbftpd_recv, METH_VARARGS,
    "Receive an authentication message from the client.\n"
    "\n"
    "Equivilent to bbftpd_private_recv() C call.\n"
    "\n"
    "@param len: The length of message to expect\n"
    "@return: A string containing the message\n"
    "@raise IOError: if bbftpd_private_recv() call fails\n"
  },
  {NULL, NULL, 0, NULL}
};

PyMODINIT_FUNC initbbftpd(void) {
  (void) Py_InitModule("bbftpd", BbftpdMethods);
}