validate: Implement a fault injection library.

+ And implement a corrupt-socket-recv action
+ Only compile this on Linux, LD_PRELOAD won't work on Windows.

For now the registering of the action is done through
a call to socket_interposer_init, this will get better
when we refactor the action logic.

https://bugzilla.gnome.org/show_bug.cgi?id=743871
This commit is contained in:
Mathieu Duponchelle 2015-01-30 18:52:57 +01:00
parent 89048ad862
commit 4ce91b98f0
8 changed files with 417 additions and 5 deletions

View file

@ -3,6 +3,7 @@ DISTCHECK_CONFIGURE_FLAGS=--enable-gtk-doc
SUBDIRS = \ SUBDIRS = \
common \ common \
data \ data \
fault_injection \
gst \ gst \
launcher \ launcher \
tools \ tools \

View file

@ -235,6 +235,10 @@ GST_CFLAGS="$GST_CFLAGS \$(GST_OPTION_CFLAGS)"
AC_SUBST(GST_CFLAGS) AC_SUBST(GST_CFLAGS)
AC_SUBST(GST_LIBS) AC_SUBST(GST_LIBS)
dnl Tiny library overriding calls such as socket recv / send
FAULTINJECTION_LIBS="-L\$(top_srcdir)/fault_injection/ -lfaultinjection-$GST_API_VERSION"
AC_SUBST([FAULTINJECTION_LIBS])
dnl GST_ALL_* dnl GST_ALL_*
dnl vars common to for all internal objects (core libs, elements, applications) dnl vars common to for all internal objects (core libs, elements, applications)
dnl CFLAGS: dnl CFLAGS:
@ -275,6 +279,7 @@ Makefile
common/Makefile common/Makefile
common/m4/Makefile common/m4/Makefile
data/Makefile data/Makefile
fault_injection/Makefile
gst/Makefile gst/Makefile
gst/validate/Makefile gst/validate/Makefile
gst/preload/Makefile gst/preload/Makefile

View file

@ -0,0 +1,13 @@
libfaultinjection_@GST_API_VERSION@_la_SOURCES = \
socket_interposer.c
libfaultinjection_@GST_API_VERSION@include_HEADERS = \
socket_interposer.h
lib_LTLIBRARIES = libfaultinjection-@GST_API_VERSION@.la
libfaultinjection_@GST_API_VERSION@_la_CFLAGS = $(GST_ALL_CFLAGS)
libfaultinjection_@GST_API_VERSION@_la_LIBADD = $(GST_ALL_LIBS)
libfaultinjection_@GST_API_VERSION@includedir = $(includedir)/gstreamer-@GST_API_VERSION@/faultinjection
CLEANFILES =

View file

@ -0,0 +1,359 @@
/* GStreamer
*
* Copyright (C) 2014 YouView TV Ltd
* Authors: Mariusz Buras <mariusz.buras@youview.com>
* Mathieu Duponchelle <mathieu.duponchelle@collabora.com>
*
* socket_interposer.c : overrides for standard socket functions
*
* 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; either
* version 2.1 of the License, or (at your option) any later version.
*
* 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., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#define _GNU_SOURCE
#include "socket_interposer.h"
#include <gst/validate/gst-validate-scenario.h>
#ifdef G_OS_UNIX
#include <sys/socket.h>
#include <netinet/ip.h>
#include <dlfcn.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/syscall.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <pthread.h>
#include <errno.h>
#define MAX_CALLBACKS (16)
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
/* Return 0 to remove the callback immediately */
typedef int (*socket_interposer_callback) (void *, const void *, size_t);
struct
{
socket_interposer_callback callback;
void *userdata;
struct sockaddr_in sockaddr;
int fd;
} callbacks[MAX_CALLBACKS];
static int
socket_interposer_remove_callback_unlocked (struct sockaddr_in *addrin,
socket_interposer_callback callback, void *userdata)
{
size_t i;
for (i = 0; i < MAX_CALLBACKS; i++) {
if (callbacks[i].callback == callback
&& callbacks[i].userdata == userdata
&& callbacks[i].sockaddr.sin_addr.s_addr == addrin->sin_addr.s_addr
&& callbacks[i].sockaddr.sin_port == addrin->sin_port) {
memset (&callbacks[i], 0, sizeof (callbacks[0]));
return 1;
}
}
return 0;
}
static void
socket_interposer_set_callback (struct sockaddr_in *addrin,
socket_interposer_callback callback, void *userdata)
{
size_t i;
pthread_mutex_lock (&mutex);
socket_interposer_remove_callback_unlocked (addrin, callback, userdata);
for (i = 0; i < MAX_CALLBACKS; i++) {
if (callbacks[i].callback == NULL) {
callbacks[i].callback = callback;
callbacks[i].userdata = userdata;
memcpy (&callbacks[i].sockaddr, addrin, sizeof (struct sockaddr_in));
callbacks[i].fd = -1;
break;
}
}
pthread_mutex_unlock (&mutex);
}
int
connect (int socket, const struct sockaddr_in *addrin, socklen_t address_len)
{
size_t i;
int override_errno = 0;
typedef ssize_t (*real_connect_fn) (int, const struct sockaddr_in *,
socklen_t);
static real_connect_fn real_connect = 0;
ssize_t ret = 0;
pthread_mutex_lock (&mutex);
for (i = 0; i < MAX_CALLBACKS; i++) {
if (callbacks[i].sockaddr.sin_addr.s_addr == addrin->sin_addr.s_addr
&& callbacks[i].sockaddr.sin_port == addrin->sin_port) {
callbacks[i].fd = socket;
if (callbacks[i].callback) {
int ret = callbacks[i].callback (callbacks[i].userdata, NULL,
0);
if (ret != 0)
override_errno = ret;
else /* Remove the callback */
memset (&callbacks[i], 0, sizeof (callbacks[0]));
}
break;
}
}
pthread_mutex_unlock (&mutex);
if (!real_connect) {
real_connect = (real_connect_fn) dlsym (RTLD_NEXT, "connect");
}
if (!override_errno) {
ret = real_connect (socket, addrin, address_len);
} else {
// override errno
errno = override_errno;
ret = -1;
}
return ret;
}
ssize_t
send (int socket, const void *buffer, size_t len, int flags)
{
size_t i;
int override_errno = 0;
typedef ssize_t (*real_send_fn) (int, const void *, size_t, int);
ssize_t ret;
static real_send_fn real_send = 0;
pthread_mutex_lock (&mutex);
for (i = 0; i < MAX_CALLBACKS; i++) {
if (callbacks[i].fd != 0 && callbacks[i].fd == socket) {
int ret = callbacks[i].callback (callbacks[i].userdata, buffer,
len);
if (ret != 0)
override_errno = ret;
else /* Remove the callback */
memset (&callbacks[i], 0, sizeof (callbacks[0]));
break;
}
}
pthread_mutex_unlock (&mutex);
if (!real_send) {
real_send = (real_send_fn) dlsym (RTLD_NEXT, "send");
}
ret = real_send (socket, buffer, len, flags);
// override errno
if (override_errno != 0) {
errno = override_errno;
ret = -1;
}
return ret;
}
ssize_t
recv (int socket, void *buffer, size_t length, int flags)
{
size_t i;
int old_errno;
typedef ssize_t (*real_recv_fn) (int, void *, size_t, int);
ssize_t ret;
static real_recv_fn real_recv = 0;
if (!real_recv) {
real_recv = (real_recv_fn) dlsym (RTLD_NEXT, "recv");
}
ret = real_recv (socket, buffer, length, flags);
old_errno = errno;
pthread_mutex_lock (&mutex);
for (i = 0; i < MAX_CALLBACKS; i++) {
if (callbacks[i].fd != 0 && callbacks[i].fd == socket) {
int newerrno = callbacks[i].callback (callbacks[i].userdata, buffer,
ret);
// override errno
if (newerrno != 0) {
old_errno = newerrno;
ret = -1;
} else { /* Remove the callback */
memset (&callbacks[i], 0, sizeof (callbacks[0]));
}
break;
}
}
pthread_mutex_unlock (&mutex);
errno = old_errno;
return ret;
}
struct errno_entry
{
const gchar *str;
int _errno;
};
static struct errno_entry errno_map[] = {
{"ECONNABORTED", ECONNABORTED},
{"ECONNRESET", ECONNRESET},
{"ENETRESET", ENETRESET},
{"ECONNREFUSED", ECONNREFUSED},
{"EHOSTUNREACH", EHOSTUNREACH},
{"EHOSTDOWN", EHOSTDOWN},
{NULL, 0},
};
static int
socket_callback_ (GstValidateAction * action, const void *buff, size_t len)
{
gint times;
gint real_errno;
gst_structure_get_int (action->structure, "times", &times);
gst_structure_get_int (action->structure, "real_errno", &real_errno);
times -= 1;
gst_structure_set (action->structure, "times", G_TYPE_INT, times, NULL);
if (times <= 0) {
gst_validate_action_set_done (action);
return 0;
}
return real_errno;
}
static gint
errno_string_to_int (const gchar * errno_str)
{
gint i;
for (i = 0; errno_map[i]._errno; i += 1) {
if (!g_ascii_strcasecmp (errno_map[i].str, errno_str))
return errno_map[i]._errno;
}
return 0;
}
static gboolean
_fault_injector_loaded (void)
{
const gchar *ld_preload = g_getenv ("LD_PRELOAD");
return (ld_preload && strstr (ld_preload, "libfaultinjection-1.0.so"));
}
static gboolean
_execute_corrupt_socket_recv (GstValidateScenario * scenario,
GstValidateAction * action)
{
struct sockaddr_in addr =
{ AF_INET, htons (42), {htonl (INADDR_LOOPBACK)}, {0} };
gint server_port, times;
const gchar *errno_str;
gint real_errno;
if (!_fault_injector_loaded ()) {
GST_ERROR
("The fault injector wasn't preloaded, can't execute socket recv corruption\n"
"You should set LD_PRELOAD to the path of libfaultinjection.so");
return FALSE;
}
if (!gst_structure_get_int (action->structure, "port", &server_port)) {
GST_ERROR ("could not get port to corrupt recv on.");
return FALSE;
}
if (!gst_structure_get_int (action->structure, "times", &times)) {
gst_structure_set (action->structure, "times", G_TYPE_INT, 1, NULL);
}
errno_str = gst_structure_get_string (action->structure, "errno");
if (!errno_str) {
GST_ERROR ("Could not get errno string");
return FALSE;
}
real_errno = errno_string_to_int (errno_str);
if (real_errno == 0) {
GST_ERROR ("unrecognized errno");
return FALSE;
}
gst_structure_set (action->structure, "real_errno", G_TYPE_INT, real_errno,
NULL);
addr.sin_port = htons (server_port);
socket_interposer_set_callback (&addr,
(socket_interposer_callback) socket_callback_, action);
return GST_VALIDATE_EXECUTE_ACTION_ASYNC;
}
void
socket_interposer_init (void)
{
gst_validate_register_action_type ("corrupt-socket-recv", "fault-injector",
_execute_corrupt_socket_recv, ((GstValidateActionParameter[]) {
{
.name = "port",.description =
"The port the socket to be corrupted listens on",.mandatory =
TRUE,.types = "int",.possible_variables = NULL,}, {
.name = "errno",.description =
"errno to set when failing",.mandatory = TRUE,.types =
"string",}, {
.name = "times",.description =
"Number of times to corrupt recv, default is one",.mandatory =
FALSE,.types = "int",.possible_variables = NULL,.def = "1",}, {
NULL}
}),
"corrupt the next socket receive", GST_VALIDATE_ACTION_TYPE_ASYNC);
}
#else /* No LD_PRELOAD tricks on Windows */
void
socket_interposer_init (void)
{
}
#endif

View file

@ -0,0 +1,29 @@
/* GStreamer
*
* Copyright (C) 2014 YouView TV Ltd
* Author: Mariusz Buras <mariusz.buras@youview.com>
*
* socket_interposer.h : overrides for standard socket functions
*
* 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; either
* version 2.1 of the License, or (at your option) any later version.
*
* 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., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifndef _SOCKET_INTERPOSER_H_
#define _SOCKET_INTERPOSER_H_
extern void socket_interposer_init(void);
#endif /* _SOCKET_INTERPOSER_H_ */

View file

@ -44,15 +44,16 @@ noinst_HEADERS = \
gst-validate-i18n-lib.h \ gst-validate-i18n-lib.h \
gst-validate-internal.h gst-validate-internal.h
lib_LTLIBRARIES = libgstvalidate-@GST_API_VERSION@.la lib_LTLIBRARIES = libgstvalidate-@GST_API_VERSION@.la
libgstvalidate_@GST_API_VERSION@_la_CFLAGS = $(GST_ALL_CFLAGS) $(GIO_CFLAGS) $(GST_PBUTILS_CFLAGS) libgstvalidate_@GST_API_VERSION@_la_CFLAGS = $(GST_ALL_CFLAGS)\
$(GIO_CFLAGS) $(GST_PBUTILS_CFLAGS) -I$(top_srcdir)/fault_injection
libgstvalidate_@GST_API_VERSION@_la_LDFLAGS = $(GST_LIB_LDFLAGS) $(GST_ALL_LDFLAGS) \ libgstvalidate_@GST_API_VERSION@_la_LDFLAGS = $(GST_LIB_LDFLAGS) $(GST_ALL_LDFLAGS) \
$(GST_LT_LDFLAGS) $(GIO_LDFLAGS) $(GST_PBUTILS_LDFAGS) $(GST_LT_LDFLAGS) $(GIO_LDFLAGS) $(GST_PBUTILS_LDFAGS)
libgstvalidate_@GST_API_VERSION@_la_LIBADD = \ libgstvalidate_@GST_API_VERSION@_la_LIBADD = \
$(GST_PLUGINS_BASE_LIBS) $(GST_BASE_LIBS) \ $(GST_PLUGINS_BASE_LIBS) $(GST_BASE_LIBS) \
$(GST_ALL_LIBS) $(GIO_LIBS) $(GST_PBUTILS_LIBS) \ $(GST_ALL_LIBS) $(GIO_LIBS) $(GST_PBUTILS_LIBS) \
$(GLIB_LIBS) $(LIBM) $(GLIB_LIBS) $(LIBM)\
$(FAULTINJECTION_LIBS)
libgstvalidate_@GST_API_VERSION@includedir = $(includedir)/gstreamer-@GST_API_VERSION@/gst/validate libgstvalidate_@GST_API_VERSION@includedir = $(includedir)/gstreamer-@GST_API_VERSION@/gst/validate
@ -82,6 +83,7 @@ GstValidate-@GST_API_VERSION@.gir: $(INTROSPECTION_SCANNER) libgstvalidate-@GST_
--include=GModule-2.0 \ --include=GModule-2.0 \
--include=GLib-2.0 \ --include=GLib-2.0 \
--libtool="${LIBTOOL}" \ --libtool="${LIBTOOL}" \
$(FAULTINJECTION_LIBS) \
--pkg gstreamer-@GST_API_VERSION@ \ --pkg gstreamer-@GST_API_VERSION@ \
--pkg gstreamer-pbutils-@GST_API_VERSION@ \ --pkg gstreamer-pbutils-@GST_API_VERSION@ \
--pkg gstreamer-controller-@GST_API_VERSION@ \ --pkg gstreamer-controller-@GST_API_VERSION@ \
@ -114,6 +116,7 @@ typelibs_DATA = $(BUILT_GIRSOURCES:.gir=.typelib)
--includedir=`$(PKG_CONFIG) --variable=girdir gstreamer-base-@GST_API_VERSION@` \ --includedir=`$(PKG_CONFIG) --variable=girdir gstreamer-base-@GST_API_VERSION@` \
--includedir=`$(PKG_CONFIG) --variable=girdir gstreamer-controller-@GST_API_VERSION@` \ --includedir=`$(PKG_CONFIG) --variable=girdir gstreamer-controller-@GST_API_VERSION@` \
--includedir=`$(PKG_CONFIG) --variable=girdir gio-2.0` \ --includedir=`$(PKG_CONFIG) --variable=girdir gio-2.0` \
--shared-library=faultinjection-@GST_API_VERSION@ \
$(INTROSPECTION_COMPILER_OPTS) $< -o $(@F) $(INTROSPECTION_COMPILER_OPTS) $< -o $(@F)
endif endif

View file

@ -45,6 +45,7 @@
#include "gst-validate-utils.h" #include "gst-validate-utils.h"
#include <gst/validate/gst-validate-override.h> #include <gst/validate/gst-validate-override.h>
#include <gst/validate/gst-validate-override-registry.h> #include <gst/validate/gst-validate-override-registry.h>
#include <socket_interposer.h>
#define GST_VALIDATE_SCENARIO_GET_PRIVATE(o) \ #define GST_VALIDATE_SCENARIO_GET_PRIVATE(o) \
(G_TYPE_INSTANCE_GET_PRIVATE ((o), GST_TYPE_VALIDATE_SCENARIO, GstValidateScenarioPrivate)) (G_TYPE_INSTANCE_GET_PRIVATE ((o), GST_TYPE_VALIDATE_SCENARIO, GstValidateScenarioPrivate))
@ -2411,6 +2412,7 @@ init_scenarios (void)
}), }),
"Emits a signal to an element in the pipeline", "Emits a signal to an element in the pipeline",
GST_VALIDATE_ACTION_TYPE_NONE); GST_VALIDATE_ACTION_TYPE_NONE);
/* *INDENT-ON* */
socket_interposer_init ();
/* *INDENT-ON* */
} }

View file

@ -7,7 +7,7 @@ bin_SCRIPTS = \
gst-validate-launcher gst-validate-launcher
AM_CFLAGS = $(GST_ALL_CFLAGS) $(GST_PBUTILS_CFLAGS) $(GST_VIDEO_CFLAGS) AM_CFLAGS = $(GST_ALL_CFLAGS) $(GST_PBUTILS_CFLAGS) $(GST_VIDEO_CFLAGS)
LDADD = $(top_builddir)/gst/validate/libgstvalidate-@GST_API_VERSION@.la $(GST_PBUTILS_LIBS) $(GST_LIBS) $(GST_VIDEO_LIBS) LDADD = $(top_builddir)/gst/validate/libgstvalidate-@GST_API_VERSION@.la $(GST_PBUTILS_LIBS) $(GST_LIBS) $(GST_VIDEO_LIBS) $(FAULTINJECTION_LIBS)
gst_validate_@GST_API_VERSION@_SOURCES = gst-validate.c gst_validate_@GST_API_VERSION@_SOURCES = gst-validate.c
gst_validate_@GST_API_VERSION@_CFLAGS = $(GIO_CFLAGS) $(AM_CFLAGS) gst_validate_@GST_API_VERSION@_CFLAGS = $(GIO_CFLAGS) $(AM_CFLAGS)