mirror of
https://git.haproxy.org/git/haproxy.git/
synced 2025-08-06 15:17:01 +02:00
59b9479667
Commit e164e7a removed get_src/get_dst setting in the stream interfaces but forgot to set it in proto_tcp. Get the feature back because we need it for logging, transparent mode, ACLs etc... We now rely on the stream interface direction to know what syscall to use. One benefit of doing it this way is that we don't use getsockopt() anymore on outgoing stream interfaces nor on UNIX sockets.
154 lines
4.2 KiB
C
154 lines
4.2 KiB
C
/*
|
|
* include/proto/stream_interface.h
|
|
* This file contains stream_interface function prototypes
|
|
*
|
|
* Copyright (C) 2000-2011 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/stream_interface.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);
|
|
|
|
/* functions used when running a stream interface as a task */
|
|
void stream_int_update(struct stream_interface *si);
|
|
void stream_int_update_embedded(struct stream_interface *si);
|
|
void stream_int_shutr(struct stream_interface *si);
|
|
void stream_int_shutw(struct stream_interface *si);
|
|
void stream_int_chk_rcv(struct stream_interface *si);
|
|
void stream_int_chk_snd(struct stream_interface *si);
|
|
|
|
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 void clear_target(struct target *dest)
|
|
{
|
|
dest->type = TARG_TYPE_NONE;
|
|
dest->ptr.v = NULL;
|
|
}
|
|
|
|
static inline void set_target_client(struct target *dest)
|
|
{
|
|
dest->type = TARG_TYPE_CLIENT;
|
|
dest->ptr.v = NULL;
|
|
}
|
|
|
|
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 void stream_interface_prepare(struct stream_interface *si, const struct sock_ops *ops)
|
|
{
|
|
memcpy(&si->sock, ops, sizeof(si->sock));
|
|
}
|
|
|
|
|
|
/* 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->proto || !si->proto->get_src)
|
|
return;
|
|
|
|
if (si->proto->get_src(si->fd, (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->proto || !si->proto->get_dst)
|
|
return;
|
|
|
|
if (si->proto->get_dst(si->fd, (struct sockaddr *)&si->addr.to,
|
|
sizeof(si->addr.to),
|
|
si->target.type != TARG_TYPE_CLIENT) == -1)
|
|
return;
|
|
si->flags |= SI_FL_TO_SET;
|
|
}
|
|
|
|
|
|
#endif /* _PROTO_STREAM_INTERFACE_H */
|
|
|
|
/*
|
|
* Local variables:
|
|
* c-indent-level: 8
|
|
* c-basic-offset: 8
|
|
* End:
|
|
*/
|