inject.h

Go to the documentation of this file.

/*
 * Copyright (C) 2011-2012 IITiS PAN Gliwice <http://www.iitis.pl/>
 * Author: Paweł Foremski <pjf@iitis.pl>
 * Licensed under GNU GPL v. 3
 */

#ifndef _INJECT_H_
#define _INJECT_H_

#include <stdbool.h>
#include <stdint.h>
#include <sys/socket.h>
#include <linux/net.h>
#include <netinet/in.h>
#include <sys/user.h>
#include <libpjf/lib.h>

#include "tracedump.h"

void inject_escape_socketcall(struct tracedump *td, struct pid *sp);

void inject_restore_socketcall(struct tracedump *td, struct pid *sp);

enum arg_type {
        AT_LAST = 0,   
        AT_VALUE,      
        AT_MEM_IN,     
        AT_MEM_INOUT   
};

int32_t inject_socketcall(struct tracedump *td, struct pid *sp, uint32_t sc_code, ...);

static inline int inject_autobind(struct tracedump *td, struct pid *sp, int fd)
{
        struct sockaddr_in sa = {
                .sin_family = AF_INET,
                .sin_port   = 0,
                .sin_addr   = { INADDR_ANY }
        };

        return inject_socketcall(td, sp, SYS_BIND,
                AT_VALUE, fd,
                AT_MEM_IN, sizeof sa, &sa,
                AT_VALUE, sizeof sa,
                AT_LAST);
}

static inline int inject_getsockname_in(struct tracedump *td, struct pid *sp, int fd, struct sockaddr_in *sa)
{
        socklen_t size = sizeof *sa;
        int rc;

        rc = inject_socketcall(td, sp, SYS_GETSOCKNAME,
                AT_VALUE, fd,
                AT_MEM_INOUT, sizeof *sa, sa,
                AT_MEM_INOUT, sizeof size, &size,
                AT_LAST);

        if (size != sizeof *sa || sa->sin_family != AF_INET)
                return -2;
        else
                return rc;
}

static inline int inject_getsockopt(struct tracedump *td, struct pid *sp,
        int fd, int level, int optname,
        void *optval, socklen_t *optlen)
{
        return inject_socketcall(td, sp, SYS_GETSOCKOPT,
                AT_VALUE, fd,
                AT_VALUE, level,
                AT_VALUE, optname,
                AT_MEM_INOUT, *optlen, optval,
                AT_MEM_INOUT, sizeof(socklen_t), optlen,
                AT_LAST);
}

#endif