udscs.h

   1 /*  udscs.h Unix Domain Socket Client Server framework header file
   2
   3     Copyright 2010 Red Hat, Inc.
   4
   5     Red Hat Authors:
   6     Hans de Goede <hdegoede@redhat.com>
   7
   8     This program is free software: you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation, either version 3 of the License, or
  11     (at your option) any later version.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #ifndef __UDSCS_H
  23 #define __UDSCS_H
  24
  25 #include <stdio.h>
  26 #include <stdint.h>
  27 #include <sys/select.h>
  28 #include <sys/socket.h>
  29
  30 struct udscs_connection;
  31 struct udscs_server;
  32 struct udscs_message_header {
  33     uint32_t type;
  34     uint32_t arg1;
  35     uint32_t arg2;
  36     uint32_t size;
  37 };
  38
  39 /* Callbacks with this type will be called when a new connection to a
  40    server is accepted. */
  41 typedef void (*udscs_connect_callback)(struct udscs_connection *conn);
  42 /* Callbacks with this type will be called when a complete message has been
  43    received. The callback may call udscs_destroy_connection, in which case
  44    *connp must be made NULL (which udscs_destroy_connection takes care of) */
  45 typedef void (*udscs_read_callback)(struct udscs_connection **connp,
  46     struct udscs_message_header *header, uint8_t *data);
  47 /* Callback type for udscs_server_for_all_clients. Clients can be disconnected
  48    from this callback just like with a read callback. */
  49 typedef int (*udscs_for_all_clients_callback)(struct udscs_connection **connp,
  50     void *priv);
  51 /* Callbacks with this type will be called when the connection is disconnected.
  52    Note:
  53    1) udscs will destroy the connection in question itself after
  54       this callback has completed!
  55    2) This callback is always called, even if the disconnect is initiated
  56       by the udscs user through returning -1 from a read callback, or
  57       by explictly calling udscs_destroy_connection */
  58 typedef void (*udscs_disconnect_callback)(struct udscs_connection *conn);
  59
  60 /* Create a unix domain socket named name and start listening on it. */
  61 struct udscs_server *udscs_create_server(const char *socketname,
  62     udscs_connect_callback connect_callback,
  63     udscs_read_callback read_callback,
  64     udscs_disconnect_callback disconnect_callback,
  65     const char * const type_to_string[], int no_types,
  66     FILE *logfile, FILE *errfile);
  67
  68 void udscs_destroy_server(struct udscs_server *server);
  69
  70 /* Connect to a unix domain socket named name. */
  71 struct udscs_connection *udscs_connect(const char *socketname,
  72     udscs_read_callback read_callback,
  73     udscs_disconnect_callback disconnect_callback,
  74     const char * const type_to_string[], int no_types,
  75     FILE *logfile, FILE *errfile);
  76
  77 /* The contents of connp will be made NULL */
  78 void udscs_destroy_connection(struct udscs_connection **connp);
  79
  80
  81 /* Given an udscs server or client fill the fd_sets pointed to by readfds and
  82    writefds for select() usage.
  83
  84    Return value: value of the highest fd + 1 */
  85 int udscs_server_fill_fds(struct udscs_server *server, fd_set *readfds,
  86         fd_set *writefds);
  87
  88 int udscs_client_fill_fds(struct udscs_connection *conn, fd_set *readfds,
  89         fd_set *writefds);
  90
  91 /* Handle any events flagged by select for the given udscs server or client. */
  92 void udscs_server_handle_fds(struct udscs_server *server, fd_set *readfds,
  93         fd_set *writefds);
  94
  95 /* Note the connection may be destroyed (when disconnected) by this call
  96    in this case the disconnect calllback will get called before the
  97    destruction and the contents of connp will be made NULL */
  98 void udscs_client_handle_fds(struct udscs_connection **connp, fd_set *readfds,
  99         fd_set *writefds);
 100
 101
 102 /* Queue a message for delivery to the client connected through conn.
 103
 104    Returns 0 on success -1 on error (only happens when malloc fails) */
 105 int udscs_write(struct udscs_connection *conn, uint32_t type, uint32_t arg1,
 106         uint32_t arg2, const uint8_t *data, uint32_t size);
 107
 108 /* Like udscs_write, but then send the message to all clients connected to
 109    the server */
 110 int udscs_server_write_all(struct udscs_server *server,
 111         uint32_t type, uint32_t arg1, uint32_t arg2,
 112         const uint8_t *data, uint32_t size);
 113 /* Call func for all clients connected to the server, passing through
 114    priv to all func calls. Returns the total of the return values from all
 115    calls to func */
 116 int udscs_server_for_all_clients(struct udscs_server *server,
 117         udscs_for_all_clients_callback func, void *priv);
 118
 119
 120 struct ucred udscs_get_peer_cred(struct udscs_connection *conn);
 121
 122 /* For server use, to associate per connection data with a connection */
 123 void udscs_set_user_data(struct udscs_connection *conn, void *data);
 124 void *udscs_get_user_data(struct udscs_connection *conn);
 125
 126 #endif