mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-21 07:46:38 +00:00
995 lines
21 KiB
C
995 lines
21 KiB
C
/* GStreamer
|
|
* Copyright (C) <2009> Collabora Ltd
|
|
* @author: Olivier Crete <olivier.crete@collabora.co.uk
|
|
* Copyright (C) <2009> Nokia Inc
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#ifdef HAVE_OSX
|
|
#ifndef MSG_NOSIGNAL
|
|
#define MSG_NOSIGNAL SO_NOSIGPIPE
|
|
#endif
|
|
#endif
|
|
|
|
#include "shmpipe.h"
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
#include <sys/mman.h>
|
|
#include <assert.h>
|
|
|
|
#include "shmalloc.h"
|
|
|
|
/*
|
|
* The protocol over the pipe is in packets
|
|
*
|
|
* The defined types are:
|
|
* type 1: new shm area
|
|
* Area length
|
|
* Size of path (followed by path)
|
|
*
|
|
* type 2: Close shm area:
|
|
* No payload
|
|
*
|
|
* type 3: shm buffer
|
|
* offset
|
|
* bufsize
|
|
*
|
|
* type 4: ack buffer
|
|
* offset
|
|
*
|
|
* Type 4 goes from the client to the server
|
|
* The rest are from the server to the client
|
|
* The client should never write in the SHM
|
|
*/
|
|
|
|
|
|
#define LISTEN_BACKLOG 10
|
|
|
|
enum
|
|
{
|
|
COMMAND_NEW_SHM_AREA = 1,
|
|
COMMAND_CLOSE_SHM_AREA = 2,
|
|
COMMAND_NEW_BUFFER = 3,
|
|
COMMAND_ACK_BUFFER = 4
|
|
};
|
|
|
|
typedef struct _ShmArea ShmArea;
|
|
|
|
struct _ShmArea
|
|
{
|
|
int id;
|
|
|
|
int use_count;
|
|
int is_writer;
|
|
|
|
int shm_fd;
|
|
|
|
char *shm_area_buf;
|
|
size_t shm_area_len;
|
|
|
|
char *shm_area_name;
|
|
|
|
ShmAllocSpace *allocspace;
|
|
|
|
ShmArea *next;
|
|
};
|
|
|
|
struct _ShmBuffer
|
|
{
|
|
int use_count;
|
|
|
|
ShmArea *shm_area;
|
|
unsigned long offset;
|
|
size_t size;
|
|
|
|
ShmAllocBlock *ablock;
|
|
|
|
ShmBuffer *next;
|
|
|
|
void *tag;
|
|
|
|
int num_clients;
|
|
/* This must ALWAYS stay last in the struct */
|
|
int clients[0];
|
|
};
|
|
|
|
|
|
struct _ShmPipe
|
|
{
|
|
int main_socket;
|
|
char *socket_path;
|
|
int use_count;
|
|
void *data;
|
|
|
|
ShmArea *shm_area;
|
|
|
|
int next_area_id;
|
|
|
|
ShmBuffer *buffers;
|
|
|
|
int num_clients;
|
|
ShmClient *clients;
|
|
|
|
mode_t perms;
|
|
};
|
|
|
|
struct _ShmClient
|
|
{
|
|
int fd;
|
|
|
|
ShmClient *next;
|
|
};
|
|
|
|
struct _ShmBlock
|
|
{
|
|
ShmPipe *pipe;
|
|
ShmArea *area;
|
|
ShmAllocBlock *ablock;
|
|
};
|
|
|
|
struct CommandBuffer
|
|
{
|
|
unsigned int type;
|
|
int area_id;
|
|
|
|
union
|
|
{
|
|
struct
|
|
{
|
|
size_t size;
|
|
unsigned int path_size;
|
|
/* Followed by path */
|
|
} new_shm_area;
|
|
struct
|
|
{
|
|
unsigned long offset;
|
|
unsigned long size;
|
|
} buffer;
|
|
struct
|
|
{
|
|
unsigned long offset;
|
|
} ack_buffer;
|
|
} payload;
|
|
};
|
|
|
|
static ShmArea *sp_open_shm (char *path, int id, mode_t perms, size_t size);
|
|
static void sp_close_shm (ShmArea * area);
|
|
static int sp_shmbuf_dec (ShmPipe * self, ShmBuffer * buf,
|
|
ShmBuffer * prev_buf, ShmClient * client, void **tag);
|
|
static void sp_shm_area_dec (ShmPipe * self, ShmArea * area);
|
|
|
|
|
|
|
|
#define RETURN_ERROR(format, ...) do { \
|
|
fprintf (stderr, format, __VA_ARGS__); \
|
|
sp_writer_close (self, NULL, NULL); \
|
|
return NULL; \
|
|
} while (0)
|
|
|
|
ShmPipe *
|
|
sp_writer_create (const char *path, size_t size, mode_t perms)
|
|
{
|
|
ShmPipe *self = spalloc_new (ShmPipe);
|
|
int flags;
|
|
struct sockaddr_un sock_un;
|
|
int i = 0;
|
|
|
|
memset (self, 0, sizeof (ShmPipe));
|
|
|
|
self->main_socket = socket (PF_UNIX, SOCK_STREAM, 0);
|
|
self->use_count = 1;
|
|
|
|
if (self->main_socket < 0)
|
|
RETURN_ERROR ("Could not create socket (%d): %s\n", errno,
|
|
strerror (errno));
|
|
|
|
flags = fcntl (self->main_socket, F_GETFL, 0);
|
|
if (flags < 0)
|
|
RETURN_ERROR ("fcntl(F_GETFL) failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
if (fcntl (self->main_socket, F_SETFL, flags | O_NONBLOCK | FD_CLOEXEC) < 0)
|
|
RETURN_ERROR ("fcntl(F_SETFL) failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
sock_un.sun_family = AF_UNIX;
|
|
strncpy (sock_un.sun_path, path, sizeof (sock_un.sun_path) - 1);
|
|
|
|
while (bind (self->main_socket, (struct sockaddr *) &sock_un,
|
|
sizeof (struct sockaddr_un)) < 0) {
|
|
if (errno != EADDRINUSE)
|
|
RETURN_ERROR ("bind() failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
if (i > 256)
|
|
RETURN_ERROR ("Could not find a free socket name for %s", path);
|
|
|
|
snprintf (sock_un.sun_path, sizeof (sock_un.sun_path), "%s.%d", path, i);
|
|
i++;
|
|
}
|
|
|
|
self->socket_path = strdup (sock_un.sun_path);
|
|
|
|
if (chmod (self->socket_path, perms) < 0)
|
|
RETURN_ERROR ("failed to set socket permissions (%d): %s\n", errno,
|
|
strerror (errno));
|
|
|
|
if (listen (self->main_socket, LISTEN_BACKLOG) < 0)
|
|
RETURN_ERROR ("listen() failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
self->shm_area = sp_open_shm (NULL, ++self->next_area_id, perms, size);
|
|
|
|
self->perms = perms;
|
|
|
|
if (!self->shm_area)
|
|
RETURN_ERROR ("Could not open shm area (%d): %s", errno, strerror (errno));
|
|
|
|
return self;
|
|
}
|
|
|
|
#undef RETURN_ERROR
|
|
|
|
#define RETURN_ERROR(format, ...) do { \
|
|
fprintf (stderr, format, __VA_ARGS__); \
|
|
area->use_count--; \
|
|
sp_close_shm (area); \
|
|
return NULL; \
|
|
} while (0)
|
|
|
|
/* sp_open_shm:
|
|
* @path: Path of the shm area for a reader,
|
|
* NULL if this is a writer (then it will allocate its own path)
|
|
*
|
|
* Opens a ShmArea
|
|
*/
|
|
|
|
static ShmArea *
|
|
sp_open_shm (char *path, int id, mode_t perms, size_t size)
|
|
{
|
|
ShmArea *area = spalloc_new (ShmArea);
|
|
char tmppath[32];
|
|
int flags;
|
|
int prot;
|
|
int i = 0;
|
|
|
|
memset (area, 0, sizeof (ShmArea));
|
|
|
|
area->shm_area_buf = MAP_FAILED;
|
|
area->use_count = 1;
|
|
|
|
area->shm_area_len = size;
|
|
|
|
area->is_writer = (path == NULL);
|
|
|
|
|
|
if (path)
|
|
flags = O_RDONLY;
|
|
else
|
|
#ifdef HAVE_OSX
|
|
flags = O_RDWR | O_CREAT | O_EXCL;
|
|
#else
|
|
flags = O_RDWR | O_CREAT | O_TRUNC | O_EXCL;
|
|
#endif
|
|
|
|
area->shm_fd = -1;
|
|
|
|
if (path) {
|
|
area->shm_fd = shm_open (path, flags, perms);
|
|
} else {
|
|
do {
|
|
snprintf (tmppath, sizeof (tmppath), "/shmpipe.%5d.%5d", getpid (), i++);
|
|
area->shm_fd = shm_open (tmppath, flags, perms);
|
|
} while (area->shm_fd < 0 && errno == EEXIST);
|
|
}
|
|
|
|
if (area->shm_fd < 0)
|
|
RETURN_ERROR ("shm_open failed on %s (%d): %s\n",
|
|
path ? path : tmppath, errno, strerror (errno));
|
|
|
|
if (!path) {
|
|
area->shm_area_name = strdup (tmppath);
|
|
|
|
if (ftruncate (area->shm_fd, size))
|
|
RETURN_ERROR ("Could not resize memory area to header size,"
|
|
" ftruncate failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
prot = PROT_READ | PROT_WRITE;
|
|
} else {
|
|
area->shm_area_name = strdup (path);
|
|
prot = PROT_READ;
|
|
}
|
|
|
|
area->shm_area_buf = mmap (NULL, size, prot, MAP_SHARED, area->shm_fd, 0);
|
|
|
|
if (area->shm_area_buf == MAP_FAILED)
|
|
RETURN_ERROR ("mmap failed (%d): %s\n", errno, strerror (errno));
|
|
|
|
area->id = id;
|
|
|
|
if (!path)
|
|
area->allocspace = shm_alloc_space_new (area->shm_area_len);
|
|
|
|
return area;
|
|
}
|
|
|
|
#undef RETURN_ERROR
|
|
|
|
static void
|
|
sp_close_shm (ShmArea * area)
|
|
{
|
|
assert (area->use_count == 0);
|
|
|
|
if (area->allocspace)
|
|
shm_alloc_space_free (area->allocspace);
|
|
|
|
if (area->shm_area_buf != MAP_FAILED)
|
|
munmap (area->shm_area_buf, area->shm_area_len);
|
|
|
|
if (area->shm_fd >= 0)
|
|
close (area->shm_fd);
|
|
|
|
if (area->shm_area_name) {
|
|
if (area->is_writer)
|
|
shm_unlink (area->shm_area_name);
|
|
free (area->shm_area_name);
|
|
}
|
|
|
|
spalloc_free (ShmArea, area);
|
|
}
|
|
|
|
static void
|
|
sp_shm_area_inc (ShmArea * area)
|
|
{
|
|
area->use_count++;
|
|
}
|
|
|
|
static void
|
|
sp_shm_area_dec (ShmPipe * self, ShmArea * area)
|
|
{
|
|
assert (area->use_count > 0);
|
|
area->use_count--;
|
|
|
|
if (area->use_count == 0) {
|
|
ShmArea *item = NULL;
|
|
ShmArea *prev_item = NULL;
|
|
|
|
for (item = self->shm_area; item; item = item->next) {
|
|
if (item == area) {
|
|
if (prev_item)
|
|
prev_item->next = item->next;
|
|
else
|
|
self->shm_area = item->next;
|
|
break;
|
|
}
|
|
prev_item = item;
|
|
}
|
|
assert (item);
|
|
|
|
sp_close_shm (area);
|
|
}
|
|
}
|
|
|
|
void *
|
|
sp_get_data (ShmPipe * self)
|
|
{
|
|
return self->data;
|
|
}
|
|
|
|
void
|
|
sp_set_data (ShmPipe * self, void *data)
|
|
{
|
|
self->data = data;
|
|
}
|
|
|
|
static void
|
|
sp_inc (ShmPipe * self)
|
|
{
|
|
self->use_count++;
|
|
}
|
|
|
|
static void
|
|
sp_dec (ShmPipe * self)
|
|
{
|
|
self->use_count--;
|
|
|
|
if (self->use_count > 0)
|
|
return;
|
|
|
|
while (self->shm_area)
|
|
sp_shm_area_dec (self, self->shm_area);
|
|
|
|
spalloc_free (ShmPipe, self);
|
|
}
|
|
|
|
void
|
|
sp_writer_close (ShmPipe * self, sp_buffer_free_callback callback,
|
|
void *user_data)
|
|
{
|
|
if (self->main_socket >= 0) {
|
|
shutdown (self->main_socket, SHUT_RDWR);
|
|
close (self->main_socket);
|
|
}
|
|
|
|
if (self->socket_path) {
|
|
unlink (self->socket_path);
|
|
free (self->socket_path);
|
|
}
|
|
|
|
while (self->clients)
|
|
sp_writer_close_client (self, self->clients, callback, user_data);
|
|
|
|
sp_dec (self);
|
|
}
|
|
|
|
void
|
|
sp_client_close (ShmPipe * self)
|
|
{
|
|
sp_writer_close (self, NULL, NULL);
|
|
}
|
|
|
|
|
|
int
|
|
sp_writer_setperms_shm (ShmPipe * self, mode_t perms)
|
|
{
|
|
int ret = 0;
|
|
ShmArea *area;
|
|
|
|
self->perms = perms;
|
|
for (area = self->shm_area; area; area = area->next)
|
|
ret |= fchmod (area->shm_fd, perms);
|
|
|
|
ret |= chmod (self->socket_path, perms);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static int
|
|
send_command (int fd, struct CommandBuffer *cb, unsigned short int type,
|
|
int area_id)
|
|
{
|
|
cb->type = type;
|
|
cb->area_id = area_id;
|
|
|
|
if (send (fd, cb, sizeof (struct CommandBuffer), MSG_NOSIGNAL) !=
|
|
sizeof (struct CommandBuffer))
|
|
return 0;
|
|
|
|
return 1;
|
|
}
|
|
|
|
int
|
|
sp_writer_resize (ShmPipe * self, size_t size)
|
|
{
|
|
ShmArea *newarea;
|
|
ShmArea *old_current;
|
|
ShmClient *client;
|
|
int c = 0;
|
|
int pathlen;
|
|
|
|
if (self->shm_area->shm_area_len == size)
|
|
return 0;
|
|
|
|
newarea = sp_open_shm (NULL, ++self->next_area_id, self->perms, size);
|
|
|
|
if (!newarea)
|
|
return -1;
|
|
|
|
old_current = self->shm_area;
|
|
newarea->next = self->shm_area;
|
|
self->shm_area = newarea;
|
|
|
|
pathlen = strlen (newarea->shm_area_name) + 1;
|
|
|
|
for (client = self->clients; client; client = client->next) {
|
|
struct CommandBuffer cb = { 0 };
|
|
|
|
if (!send_command (client->fd, &cb, COMMAND_CLOSE_SHM_AREA,
|
|
old_current->id))
|
|
continue;
|
|
|
|
cb.payload.new_shm_area.size = newarea->shm_area_len;
|
|
cb.payload.new_shm_area.path_size = pathlen;
|
|
if (!send_command (client->fd, &cb, COMMAND_NEW_SHM_AREA, newarea->id))
|
|
continue;
|
|
|
|
if (send (client->fd, newarea->shm_area_name, pathlen, MSG_NOSIGNAL) !=
|
|
pathlen)
|
|
continue;
|
|
c++;
|
|
}
|
|
|
|
sp_shm_area_dec (self, old_current);
|
|
|
|
|
|
return c;
|
|
}
|
|
|
|
ShmBlock *
|
|
sp_writer_alloc_block (ShmPipe * self, size_t size)
|
|
{
|
|
ShmBlock *block;
|
|
ShmAllocBlock *ablock =
|
|
shm_alloc_space_alloc_block (self->shm_area->allocspace, size);
|
|
|
|
if (!ablock)
|
|
return NULL;
|
|
|
|
block = spalloc_new (ShmBlock);
|
|
sp_shm_area_inc (self->shm_area);
|
|
block->pipe = self;
|
|
block->area = self->shm_area;
|
|
block->ablock = ablock;
|
|
sp_inc (self);
|
|
return block;
|
|
}
|
|
|
|
char *
|
|
sp_writer_block_get_buf (ShmBlock * block)
|
|
{
|
|
return block->area->shm_area_buf +
|
|
shm_alloc_space_alloc_block_get_offset (block->ablock);
|
|
}
|
|
|
|
ShmPipe *
|
|
sp_writer_block_get_pipe (ShmBlock * block)
|
|
{
|
|
return block->pipe;
|
|
}
|
|
|
|
void
|
|
sp_writer_free_block (ShmBlock * block)
|
|
{
|
|
shm_alloc_space_block_dec (block->ablock);
|
|
sp_shm_area_dec (block->pipe, block->area);
|
|
sp_dec (block->pipe);
|
|
spalloc_free (ShmBlock, block);
|
|
}
|
|
|
|
/* Returns the number of client this has successfully been sent to */
|
|
|
|
int
|
|
sp_writer_send_buf (ShmPipe * self, char *buf, size_t size, void *tag)
|
|
{
|
|
ShmArea *area = NULL;
|
|
unsigned long offset = 0;
|
|
unsigned long bsize = size;
|
|
ShmBuffer *sb;
|
|
ShmClient *client = NULL;
|
|
ShmAllocBlock *ablock = NULL;
|
|
int i = 0;
|
|
int c = 0;
|
|
|
|
if (self->num_clients == 0)
|
|
return 0;
|
|
|
|
for (area = self->shm_area; area; area = area->next) {
|
|
if (buf >= area->shm_area_buf &&
|
|
buf < (area->shm_area_buf + area->shm_area_len)) {
|
|
offset = buf - area->shm_area_buf;
|
|
ablock = shm_alloc_space_block_get (area->allocspace, offset);
|
|
assert (ablock);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!ablock)
|
|
return -1;
|
|
|
|
sb = spalloc_alloc (sizeof (ShmBuffer) + sizeof (int) * self->num_clients);
|
|
memset (sb, 0, sizeof (ShmBuffer));
|
|
memset (sb->clients, -1, sizeof (int) * self->num_clients);
|
|
sb->shm_area = area;
|
|
sb->offset = offset;
|
|
sb->size = size;
|
|
sb->num_clients = self->num_clients;
|
|
sb->ablock = ablock;
|
|
sb->tag = tag;
|
|
|
|
for (client = self->clients; client; client = client->next) {
|
|
struct CommandBuffer cb = { 0 };
|
|
cb.payload.buffer.offset = offset;
|
|
cb.payload.buffer.size = bsize;
|
|
if (!send_command (client->fd, &cb, COMMAND_NEW_BUFFER, self->shm_area->id))
|
|
continue;
|
|
sb->clients[i++] = client->fd;
|
|
c++;
|
|
}
|
|
|
|
if (c == 0) {
|
|
spalloc_free1 (sizeof (ShmBuffer) + sizeof (int) * sb->num_clients, sb);
|
|
return 0;
|
|
}
|
|
|
|
sp_shm_area_inc (area);
|
|
shm_alloc_space_block_inc (ablock);
|
|
|
|
sb->use_count = c;
|
|
|
|
sb->next = self->buffers;
|
|
self->buffers = sb;
|
|
|
|
return c;
|
|
}
|
|
|
|
static int
|
|
recv_command (int fd, struct CommandBuffer *cb)
|
|
{
|
|
int retval;
|
|
|
|
retval = recv (fd, cb, sizeof (struct CommandBuffer), MSG_DONTWAIT);
|
|
if (retval == sizeof (struct CommandBuffer)) {
|
|
return 1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
long int
|
|
sp_client_recv (ShmPipe * self, char **buf)
|
|
{
|
|
char *area_name = NULL;
|
|
ShmArea *newarea;
|
|
ShmArea *area;
|
|
struct CommandBuffer cb;
|
|
int retval;
|
|
|
|
if (!recv_command (self->main_socket, &cb))
|
|
return -1;
|
|
|
|
switch (cb.type) {
|
|
case COMMAND_NEW_SHM_AREA:
|
|
assert (cb.payload.new_shm_area.path_size > 0);
|
|
assert (cb.payload.new_shm_area.size > 0);
|
|
|
|
area_name = malloc (cb.payload.new_shm_area.path_size + 1);
|
|
retval = recv (self->main_socket, area_name,
|
|
cb.payload.new_shm_area.path_size, 0);
|
|
if (retval != cb.payload.new_shm_area.path_size) {
|
|
free (area_name);
|
|
return -3;
|
|
}
|
|
/* Ensure area_name is NULL terminated */
|
|
area_name[retval] = 0;
|
|
|
|
newarea = sp_open_shm (area_name, cb.area_id, 0,
|
|
cb.payload.new_shm_area.size);
|
|
free (area_name);
|
|
if (!newarea)
|
|
return -4;
|
|
|
|
newarea->next = self->shm_area;
|
|
self->shm_area = newarea;
|
|
break;
|
|
|
|
case COMMAND_CLOSE_SHM_AREA:
|
|
for (area = self->shm_area; area; area = area->next) {
|
|
if (area->id == cb.area_id) {
|
|
sp_shm_area_dec (self, area);
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case COMMAND_NEW_BUFFER:
|
|
assert (buf);
|
|
for (area = self->shm_area; area; area = area->next) {
|
|
if (area->id == cb.area_id) {
|
|
*buf = area->shm_area_buf + cb.payload.buffer.offset;
|
|
sp_shm_area_inc (area);
|
|
return cb.payload.buffer.size;
|
|
}
|
|
}
|
|
return -23;
|
|
|
|
default:
|
|
return -99;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
sp_writer_recv (ShmPipe * self, ShmClient * client, void **tag)
|
|
{
|
|
ShmBuffer *buf = NULL, *prev_buf = NULL;
|
|
struct CommandBuffer cb;
|
|
|
|
if (!recv_command (client->fd, &cb))
|
|
return -1;
|
|
|
|
switch (cb.type) {
|
|
case COMMAND_ACK_BUFFER:
|
|
|
|
for (buf = self->buffers; buf; buf = buf->next) {
|
|
if (buf->shm_area->id == cb.area_id &&
|
|
buf->offset == cb.payload.ack_buffer.offset) {
|
|
return sp_shmbuf_dec (self, buf, prev_buf, client, tag);
|
|
}
|
|
prev_buf = buf;
|
|
}
|
|
|
|
return -2;
|
|
default:
|
|
return -99;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
sp_client_recv_finish (ShmPipe * self, char *buf)
|
|
{
|
|
ShmArea *shm_area = NULL;
|
|
unsigned long offset;
|
|
struct CommandBuffer cb = { 0 };
|
|
|
|
for (shm_area = self->shm_area; shm_area; shm_area = shm_area->next) {
|
|
if (buf >= shm_area->shm_area_buf &&
|
|
buf < shm_area->shm_area_buf + shm_area->shm_area_len)
|
|
break;
|
|
}
|
|
|
|
assert (shm_area);
|
|
|
|
offset = buf - shm_area->shm_area_buf;
|
|
|
|
sp_shm_area_dec (self, shm_area);
|
|
|
|
cb.payload.ack_buffer.offset = offset;
|
|
return send_command (self->main_socket, &cb, COMMAND_ACK_BUFFER,
|
|
self->shm_area->id);
|
|
}
|
|
|
|
ShmPipe *
|
|
sp_client_open (const char *path)
|
|
{
|
|
ShmPipe *self = spalloc_new (ShmPipe);
|
|
struct sockaddr_un sock_un;
|
|
int flags;
|
|
|
|
memset (self, 0, sizeof (ShmPipe));
|
|
|
|
self->main_socket = socket (PF_UNIX, SOCK_STREAM, 0);
|
|
self->use_count = 1;
|
|
|
|
if (self->main_socket < 0)
|
|
goto error;
|
|
|
|
flags = fcntl (self->main_socket, F_GETFL, 0);
|
|
if (flags < 0)
|
|
goto error;
|
|
|
|
if (fcntl (self->main_socket, F_SETFL, flags | FD_CLOEXEC) < 0)
|
|
goto error;
|
|
|
|
sock_un.sun_family = AF_UNIX;
|
|
strncpy (sock_un.sun_path, path, sizeof (sock_un.sun_path) - 1);
|
|
|
|
if (connect (self->main_socket, (struct sockaddr *) &sock_un,
|
|
sizeof (struct sockaddr_un)) < 0)
|
|
goto error;
|
|
|
|
return self;
|
|
|
|
error:
|
|
sp_client_close (self);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
ShmClient *
|
|
sp_writer_accept_client (ShmPipe * self)
|
|
{
|
|
ShmClient *client = NULL;
|
|
int fd;
|
|
struct CommandBuffer cb = { 0 };
|
|
int pathlen = strlen (self->shm_area->shm_area_name) + 1;
|
|
|
|
|
|
fd = accept (self->main_socket, NULL, NULL);
|
|
|
|
if (fd < 0) {
|
|
fprintf (stderr, "Could not client connection");
|
|
return NULL;
|
|
}
|
|
|
|
cb.payload.new_shm_area.size = self->shm_area->shm_area_len;
|
|
cb.payload.new_shm_area.path_size = pathlen;
|
|
if (!send_command (fd, &cb, COMMAND_NEW_SHM_AREA, self->shm_area->id)) {
|
|
fprintf (stderr, "Sending new shm area failed: %s", strerror (errno));
|
|
goto error;
|
|
}
|
|
|
|
if (send (fd, self->shm_area->shm_area_name, pathlen, MSG_NOSIGNAL) !=
|
|
pathlen) {
|
|
fprintf (stderr, "Sending new shm area path failed: %s", strerror (errno));
|
|
goto error;
|
|
}
|
|
|
|
client = spalloc_new (ShmClient);
|
|
client->fd = fd;
|
|
|
|
/* Prepend ot linked list */
|
|
client->next = self->clients;
|
|
self->clients = client;
|
|
self->num_clients++;
|
|
|
|
return client;
|
|
|
|
error:
|
|
shutdown (fd, SHUT_RDWR);
|
|
close (fd);
|
|
return NULL;
|
|
}
|
|
|
|
static int
|
|
sp_shmbuf_dec (ShmPipe * self, ShmBuffer * buf, ShmBuffer * prev_buf,
|
|
ShmClient * client, void **tag)
|
|
{
|
|
int i;
|
|
int had_client = 0;
|
|
|
|
/* Remove client from the list of buffer users. Here we make sure that
|
|
* if a client closes connection but already decremented the use count
|
|
* for this buffer, but other clients didn't have time to decrement
|
|
* buffer will not be freed too early in sp_writer_close_client.
|
|
*/
|
|
for (i = 0; i < buf->num_clients; i++) {
|
|
if (buf->clients[i] == client->fd) {
|
|
buf->clients[i] = -1;
|
|
had_client = 1;
|
|
break;
|
|
}
|
|
}
|
|
assert (had_client);
|
|
|
|
buf->use_count--;
|
|
|
|
if (buf->use_count == 0) {
|
|
/* Remove from linked list */
|
|
if (prev_buf)
|
|
prev_buf->next = buf->next;
|
|
else
|
|
self->buffers = buf->next;
|
|
|
|
if (tag)
|
|
*tag = buf->tag;
|
|
shm_alloc_space_block_dec (buf->ablock);
|
|
sp_shm_area_dec (self, buf->shm_area);
|
|
spalloc_free1 (sizeof (ShmBuffer) + sizeof (int) * buf->num_clients, buf);
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
void
|
|
sp_writer_close_client (ShmPipe * self, ShmClient * client,
|
|
sp_buffer_free_callback callback, void *user_data)
|
|
{
|
|
ShmBuffer *buffer = NULL, *prev_buf = NULL;
|
|
ShmClient *item = NULL, *prev_item = NULL;
|
|
|
|
shutdown (client->fd, SHUT_RDWR);
|
|
close (client->fd);
|
|
|
|
again:
|
|
for (buffer = self->buffers; buffer; buffer = buffer->next) {
|
|
int i;
|
|
void *tag = NULL;
|
|
|
|
for (i = 0; i < buffer->num_clients; i++) {
|
|
if (buffer->clients[i] == client->fd) {
|
|
if (!sp_shmbuf_dec (self, buffer, prev_buf, client, &tag)) {
|
|
if (callback)
|
|
callback (tag, user_data);
|
|
goto again;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
prev_buf = buffer;
|
|
}
|
|
|
|
for (item = self->clients; item; item = item->next) {
|
|
if (item == client)
|
|
break;
|
|
prev_item = item;
|
|
}
|
|
assert (item);
|
|
|
|
if (prev_item)
|
|
prev_item->next = client->next;
|
|
else
|
|
self->clients = client->next;
|
|
|
|
self->num_clients--;
|
|
|
|
spalloc_free (ShmClient, client);
|
|
}
|
|
|
|
int
|
|
sp_get_fd (ShmPipe * self)
|
|
{
|
|
return self->main_socket;
|
|
}
|
|
|
|
const char *
|
|
sp_get_shm_area_name (ShmPipe * self)
|
|
{
|
|
if (self->shm_area)
|
|
return self->shm_area->shm_area_name;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
int
|
|
sp_writer_get_client_fd (ShmClient * client)
|
|
{
|
|
return client->fd;
|
|
}
|
|
|
|
int
|
|
sp_writer_pending_writes (ShmPipe * self)
|
|
{
|
|
return (self->buffers != NULL);
|
|
}
|
|
|
|
const char *
|
|
sp_writer_get_path (ShmPipe * pipe)
|
|
{
|
|
return pipe->socket_path;
|
|
}
|
|
|
|
ShmBuffer *
|
|
sp_writer_get_pending_buffers (ShmPipe * self)
|
|
{
|
|
return self->buffers;
|
|
}
|
|
|
|
ShmBuffer *
|
|
sp_writer_get_next_buffer (ShmBuffer * buffer)
|
|
{
|
|
return buffer->next;
|
|
}
|
|
|
|
void *
|
|
sp_writer_buf_get_tag (ShmBuffer * buffer)
|
|
{
|
|
return buffer->tag;
|
|
}
|
|
|
|
size_t
|
|
sp_writer_get_max_buf_size (ShmPipe * self)
|
|
{
|
|
if (self->shm_area == NULL)
|
|
return 0;
|
|
|
|
return self->shm_area->shm_area_len;
|
|
}
|