| /* |
| * include/proto/stream_interface.h |
| * This file contains stream_interface function prototypes |
| * |
| * Copyright (C) 2000-2012 Willy Tarreau - w@1wt.eu |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation, version 2.1 |
| * exclusively. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| #ifndef _PROTO_STREAM_INTERFACE_H |
| #define _PROTO_STREAM_INTERFACE_H |
| |
| #include <stdlib.h> |
| |
| #include <common/config.h> |
| #include <types/session.h> |
| #include <types/stream_interface.h> |
| #include <proto/connection.h> |
| |
| |
| /* main event functions used to move data between sockets and buffers */ |
| int stream_int_check_timeouts(struct stream_interface *si); |
| void stream_int_report_error(struct stream_interface *si); |
| void stream_int_retnclose(struct stream_interface *si, const struct chunk *msg); |
| int conn_si_send_proxy(struct connection *conn, unsigned int flag); |
| void conn_notify_si(struct connection *conn); |
| void stream_int_update_conn(struct stream_interface *si); |
| int stream_int_shutr(struct stream_interface *si); |
| int stream_int_shutw(struct stream_interface *si); |
| void stream_int_chk_rcv_conn(struct stream_interface *si); |
| void stream_int_chk_snd_conn(struct stream_interface *si); |
| void si_conn_send_cb(struct connection *conn); |
| |
| extern struct sock_ops stream_int_embedded; |
| extern struct sock_ops stream_int_task; |
| |
| struct task *stream_int_register_handler(struct stream_interface *si, |
| struct si_applet *app); |
| struct task *stream_int_register_handler_task(struct stream_interface *si, |
| struct task *(*fct)(struct task *)); |
| void stream_int_unregister_handler(struct stream_interface *si); |
| |
| static inline const struct protocol *si_ctrl(struct stream_interface *si) |
| { |
| return si->conn.ctrl; |
| } |
| |
| static inline const struct sock_ops *si_data(struct stream_interface *si) |
| { |
| return si->conn.data; |
| } |
| |
| static inline int si_fd(struct stream_interface *si) |
| { |
| return si->conn.t.sock.fd; |
| } |
| |
| static inline void clear_target(struct target *dest) |
| { |
| dest->type = TARG_TYPE_NONE; |
| dest->ptr.v = NULL; |
| } |
| |
| static inline void set_target_client(struct target *dest, struct listener *l) |
| { |
| dest->type = TARG_TYPE_CLIENT; |
| dest->ptr.l = l; |
| } |
| |
| static inline void set_target_server(struct target *dest, struct server *s) |
| { |
| dest->type = TARG_TYPE_SERVER; |
| dest->ptr.s = s; |
| } |
| |
| static inline void set_target_proxy(struct target *dest, struct proxy *p) |
| { |
| dest->type = TARG_TYPE_PROXY; |
| dest->ptr.p = p; |
| } |
| |
| static inline void set_target_applet(struct target *dest, struct si_applet *a) |
| { |
| dest->type = TARG_TYPE_APPLET; |
| dest->ptr.a = a; |
| } |
| |
| static inline void set_target_task(struct target *dest, struct task *t) |
| { |
| dest->type = TARG_TYPE_TASK; |
| dest->ptr.t = t; |
| } |
| |
| static inline struct target *copy_target(struct target *dest, struct target *src) |
| { |
| *dest = *src; |
| return dest; |
| } |
| |
| static inline int target_match(struct target *a, struct target *b) |
| { |
| return a->type == b->type && a->ptr.v == b->ptr.v; |
| } |
| |
| static inline struct server *target_srv(struct target *t) |
| { |
| if (!t || t->type != TARG_TYPE_SERVER) |
| return NULL; |
| return t->ptr.s; |
| } |
| |
| static inline struct listener *target_client(struct target *t) |
| { |
| if (!t || t->type != TARG_TYPE_CLIENT) |
| return NULL; |
| return t->ptr.l; |
| } |
| |
| static inline void stream_interface_prepare(struct stream_interface *si, const struct sock_ops *ops) |
| { |
| si->conn.data = ops; |
| si->conn.data_st = 0; |
| si->conn.data_ctx = NULL; |
| } |
| |
| |
| /* Retrieves the source address for the stream interface. */ |
| static inline void si_get_from_addr(struct stream_interface *si) |
| { |
| if (si->flags & SI_FL_FROM_SET) |
| return; |
| |
| if (!si_ctrl(si) || !si_ctrl(si)->get_src) |
| return; |
| |
| if (si_ctrl(si)->get_src(si_fd(si), (struct sockaddr *)&si->addr.from, |
| sizeof(si->addr.from), |
| si->target.type != TARG_TYPE_CLIENT) == -1) |
| return; |
| si->flags |= SI_FL_FROM_SET; |
| } |
| |
| /* Retrieves the original destination address for the stream interface. */ |
| static inline void si_get_to_addr(struct stream_interface *si) |
| { |
| if (si->flags & SI_FL_TO_SET) |
| return; |
| |
| if (!si_ctrl(si) || !si_ctrl(si)->get_dst) |
| return; |
| |
| if (si_ctrl(si)->get_dst(si_fd(si), (struct sockaddr *)&si->addr.to, |
| sizeof(si->addr.to), |
| si->target.type != TARG_TYPE_CLIENT) == -1) |
| return; |
| si->flags |= SI_FL_TO_SET; |
| } |
| |
| /* Sends a shutr to the connection using the data layer */ |
| static inline void si_shutr(struct stream_interface *si) |
| { |
| if (stream_int_shutr(si)) |
| conn_data_stop_recv(&si->conn); |
| } |
| |
| /* Sends a shutw to the connection using the data layer */ |
| static inline void si_shutw(struct stream_interface *si) |
| { |
| if (stream_int_shutw(si)) |
| conn_data_stop_send(&si->conn); |
| } |
| |
| /* Calls the data state update on the stream interfaace */ |
| static inline void si_update(struct stream_interface *si) |
| { |
| si_data(si)->update(si); |
| } |
| |
| /* Calls chk_rcv on the connection using the data layer */ |
| static inline void si_chk_rcv(struct stream_interface *si) |
| { |
| si_data(si)->chk_rcv(si); |
| } |
| |
| /* Calls chk_snd on the connection using the data layer */ |
| static inline void si_chk_snd(struct stream_interface *si) |
| { |
| si_data(si)->chk_snd(si); |
| } |
| |
| /* Calls chk_snd on the connection using the ctrl layer */ |
| static inline int si_connect(struct stream_interface *si) |
| { |
| if (unlikely(!si_ctrl(si) || !si_ctrl(si)->connect)) |
| return SN_ERR_INTERNAL; |
| return si_ctrl(si)->connect(si); |
| } |
| |
| #endif /* _PROTO_STREAM_INTERFACE_H */ |
| |
| /* |
| * Local variables: |
| * c-indent-level: 8 |
| * c-basic-offset: 8 |
| * End: |
| */ |