#define __CHARYBDIS_AUTHD_PROVIDER_H__
#include "stdinc.h"
+#include "authd.h"
+#include "rb_dictionary.h"
-/* Arbitrary limit */
-#define MAX_CLIENTS 1024
+#define MAX_PROVIDERS 32 /* This should be enough */
-/* Registered providers */
typedef enum
{
- PROVIDER_NULL = 0x0, /* Dummy value */
- PROVIDER_RDNS = 0x1,
- PROVIDER_IDENT = 0x2,
- PROVIDER_BLACKLIST = 0x4,
-} provider_t;
+ PROVIDER_STATUS_NOTRUN = 0,
+ PROVIDER_STATUS_RUNNING,
+ PROVIDER_STATUS_DONE,
+} provider_status_t;
+
+struct auth_client_data
+{
+ struct auth_provider *provider; /* Pointer back */
+ time_t timeout; /* Provider timeout */
+ void *data; /* Provider data */
+ provider_status_t status; /* Provider status */
+};
struct auth_client
{
- uint16_t cid; /* Client ID */
+ uint32_t cid; /* Client ID */
char l_ip[HOSTIPLEN + 1]; /* Listener IP address */
uint16_t l_port; /* Listener port */
+ struct rb_sockaddr_storage l_addr; /* Listener address/port */
char c_ip[HOSTIPLEN + 1]; /* Client IP address */
uint16_t c_port; /* Client port */
+ struct rb_sockaddr_storage c_addr; /* Client address/port */
- char hostname[IRCD_RES_HOSTLEN + 1]; /* Used for DNS lookup */
+ char hostname[HOSTLEN + 1]; /* Used for DNS lookup */
char username[USERLEN + 1]; /* Used for ident lookup */
- unsigned int providers; /* Providers at work,
- * none left when set to 0 */
+ bool providers_starting; /* Providers are still warming up */
+ bool providers_cancelled; /* Providers are being cancelled */
+ unsigned int providers_active; /* Number of active providers */
+ unsigned int refcount; /* Held references */
+
+ struct auth_client_data *data; /* Provider-specific data */
};
typedef bool (*provider_init_t)(void);
-typedef bool (*provider_perform_t)(struct auth_client *);
-typedef void (*provider_complete_t)(struct auth_client *, provider_t provider);
-typedef void (*provider_cancel_t)(struct auth_client *);
typedef void (*provider_destroy_t)(void);
+typedef bool (*provider_start_t)(struct auth_client *);
+typedef void (*provider_cancel_t)(struct auth_client *);
+typedef void (*uint32_timeout_t)(struct auth_client *);
+typedef void (*provider_complete_t)(struct auth_client *, uint32_t);
+
+struct auth_stats_handler
+{
+ const char letter;
+ authd_stat_handler handler;
+};
+
struct auth_provider
{
rb_dlink_node node;
- provider_t id;
+ uint32_t id; /* Provider ID */
+
+ const char *name; /* Name of the provider */
+ char letter; /* Letter used on reject, etc. */
provider_init_t init; /* Initalise the provider */
provider_destroy_t destroy; /* Terminate the provider */
- provider_perform_t start; /* Perform authentication */
+ provider_start_t start; /* Perform authentication */
provider_cancel_t cancel; /* Authentication cancelled */
+ uint32_timeout_t timeout; /* Timeout callback */
provider_complete_t completed; /* Callback for when other performers complete (think dependency chains) */
+
+ struct auth_stats_handler stats_handler;
+
+ struct auth_opts_handler *opt_handlers;
};
+extern struct auth_provider rdns_provider;
+extern struct auth_provider ident_provider;
+extern struct auth_provider blacklist_provider;
+extern struct auth_provider opm_provider;
+
+extern rb_dlink_list auth_providers;
+extern rb_dictionary *auth_clients;
+
+void load_provider(struct auth_provider *provider);
+void unload_provider(struct auth_provider *provider);
+
void init_providers(void);
void destroy_providers(void);
void cancel_providers(struct auth_client *auth);
-void provider_done(struct auth_client *auth, provider_t provider);
-void reject_client(struct auth_client *auth, provider_t provider, const char *reason);
-void accept_client(struct auth_client *auth, provider_t provider);
-
-void notice_client(struct auth_client *auth, const char *notice);
+void provider_done(struct auth_client *auth, uint32_t id);
+void accept_client(struct auth_client *auth);
+void reject_client(struct auth_client *auth, uint32_t id, const char *data, const char *fmt, ...);
-void start_auth(const char *cid, const char *l_ip, const char *l_port, const char *c_ip, const char *c_port);
void handle_new_connection(int parc, char *parv[]);
+void handle_cancel_connection(int parc, char *parv[]);
+void auth_client_free(struct auth_client *auth);
+
+static inline void
+auth_client_ref(struct auth_client *auth)
+{
+ auth->refcount++;
+}
-/* Provider is operating on this auth_client (set this if you have async work to do) */
-static inline void set_provider(struct auth_client *auth, provider_t provider)
+static inline void
+auth_client_unref(struct auth_client *auth)
{
- auth->providers |= provider;
+ auth->refcount--;
+ if (auth->refcount == 0)
+ auth_client_free(auth);
}
-/* Provider is no longer operating on this auth client (you should use provider_done) */
-static inline void unset_provider(struct auth_client *auth, provider_t provider)
+/* Get a provider by name */
+static inline struct auth_provider *
+find_provider(const char *name)
{
- auth->providers &= ~provider;
+ rb_dlink_node *ptr;
+
+ RB_DLINK_FOREACH(ptr, auth_providers.head)
+ {
+ struct auth_provider *provider = ptr->data;
+
+ if(strcasecmp(provider->name, name) == 0)
+ return provider;
+ }
+
+ return NULL;
+}
+
+/* Get a provider's id by name */
+static inline bool
+get_provider_id(const char *name, uint32_t *id)
+{
+ struct auth_provider *provider = find_provider(name);
+
+ if(provider != NULL)
+ {
+ *id = provider->id;
+ return true;
+ }
+ else
+ return false;
+}
+
+/* Get a provider's raw status */
+static inline provider_status_t
+get_provider_status(struct auth_client *auth, uint32_t provider)
+{
+ return auth->data[provider].status;
}
/* Check if provider is operating on this auth client */
-static inline bool is_provider(struct auth_client *auth, provider_t provider)
+static inline bool
+is_provider_running(struct auth_client *auth, uint32_t provider)
+{
+ return get_provider_status(auth, provider) == PROVIDER_STATUS_RUNNING;
+}
+
+/* Check if provider has finished on this client */
+static inline bool
+is_provider_done(struct auth_client *auth, uint32_t provider)
+{
+ return get_provider_status(auth, provider) == PROVIDER_STATUS_DONE;
+}
+
+/* Get provider auth client data */
+static inline void *
+get_provider_data(struct auth_client *auth, uint32_t id)
+{
+ return auth->data[id].data;
+}
+
+/* Set provider auth client data */
+static inline void
+set_provider_data(struct auth_client *auth, uint32_t id, void *data)
+{
+ auth->data[id].data = data;
+}
+
+/* Set timeout relative to current time on provider
+ * When the timeout lapses, the provider's timeout call will execute */
+static inline void
+set_provider_timeout_relative(struct auth_client *auth, uint32_t id, time_t timeout)
+{
+ auth->data[id].timeout = timeout + rb_current_time();
+}
+
+/* Set timeout value in absolute time (Unix timestamp)
+ * When the timeout lapses, the provider's timeout call will execute */
+static inline void
+set_provider_timeout_absolute(struct auth_client *auth, uint32_t id, time_t timeout)
+{
+ auth->data[id].timeout = timeout;
+}
+
+/* Get the timeout value for the provider */
+static inline time_t
+get_provider_timeout(struct auth_client *auth, uint32_t id)
{
- return auth->providers & provider;
+ return auth->data[id].timeout;
}
#endif /* __CHARYBDIS_AUTHD_PROVIDER_H__ */
projects / solanum.git / blobdiff