Livox-SDK
487 lines
16 KiB
C
487 lines
16 KiB
C
/* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
* contributor license agreements. See the NOTICE file distributed with
|
|
* this work for additional information regarding copyright ownership.
|
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
|
* (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "apr_arch_file_io.h"
|
|
#include "apr_file_io.h"
|
|
#include "apr_general.h"
|
|
#include "apr_strings.h"
|
|
#include "apr_escape.h"
|
|
#if APR_HAVE_ERRNO_H
|
|
#include <errno.h>
|
|
#endif
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#if APR_HAVE_SYS_TYPES_H
|
|
#include <sys/types.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_STAT_H
|
|
#include <sys/stat.h>
|
|
#endif
|
|
#if APR_HAVE_PROCESS_H
|
|
#include <process.h> /* for getpid() on Win32 */
|
|
#endif
|
|
#include "apr_arch_misc.h"
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_pipe_timeout_set(apr_file_t *thepipe,
|
|
apr_interval_time_t timeout)
|
|
{
|
|
/* Always OK to unset timeouts */
|
|
if (timeout == -1) {
|
|
thepipe->timeout = timeout;
|
|
return APR_SUCCESS;
|
|
}
|
|
if (!thepipe->pipe) {
|
|
return APR_ENOTIMPL;
|
|
}
|
|
if (timeout && !(thepipe->pOverlapped)) {
|
|
/* Cannot be nonzero if a pipe was opened blocking
|
|
*/
|
|
return APR_EINVAL;
|
|
}
|
|
thepipe->timeout = timeout;
|
|
return APR_SUCCESS;
|
|
}
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_pipe_timeout_get(apr_file_t *thepipe,
|
|
apr_interval_time_t *timeout)
|
|
{
|
|
/* Always OK to get the timeout (even if it's unset ... -1) */
|
|
*timeout = thepipe->timeout;
|
|
return APR_SUCCESS;
|
|
}
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_pipe_create(apr_file_t **in,
|
|
apr_file_t **out,
|
|
apr_pool_t *p)
|
|
{
|
|
/* Unix creates full blocking pipes. */
|
|
return apr_file_pipe_create_pools(in, out, APR_FULL_BLOCK, p, p);
|
|
}
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_pipe_create_ex(apr_file_t **in,
|
|
apr_file_t **out,
|
|
apr_int32_t blocking,
|
|
apr_pool_t *p)
|
|
{
|
|
return apr_file_pipe_create_pools(in, out, blocking, p, p);
|
|
}
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_pipe_create_pools(apr_file_t **in,
|
|
apr_file_t **out,
|
|
apr_int32_t blocking,
|
|
apr_pool_t *pool_in,
|
|
apr_pool_t *pool_out)
|
|
{
|
|
#ifdef _WIN32_WCE
|
|
return APR_ENOTIMPL;
|
|
#else
|
|
SECURITY_ATTRIBUTES sa;
|
|
static unsigned long id = 0;
|
|
DWORD dwPipeMode;
|
|
DWORD dwOpenMode;
|
|
|
|
sa.nLength = sizeof(sa);
|
|
|
|
#if APR_HAS_UNICODE_FS
|
|
IF_WIN_OS_IS_UNICODE
|
|
sa.bInheritHandle = FALSE;
|
|
#endif
|
|
#if APR_HAS_ANSI_FS
|
|
ELSE_WIN_OS_IS_ANSI
|
|
sa.bInheritHandle = TRUE;
|
|
#endif
|
|
sa.lpSecurityDescriptor = NULL;
|
|
|
|
(*in) = (apr_file_t *)apr_pcalloc(pool_in, sizeof(apr_file_t));
|
|
(*in)->pool = pool_in;
|
|
(*in)->fname = NULL;
|
|
(*in)->pipe = 1;
|
|
(*in)->timeout = -1;
|
|
(*in)->ungetchar = -1;
|
|
(*in)->eof_hit = 0;
|
|
(*in)->filePtr = 0;
|
|
(*in)->bufpos = 0;
|
|
(*in)->dataRead = 0;
|
|
(*in)->direction = 0;
|
|
(*in)->pOverlapped = NULL;
|
|
#if APR_FILES_AS_SOCKETS
|
|
(void) apr_pollset_create(&(*in)->pollset, 1, p, 0);
|
|
#endif
|
|
(*out) = (apr_file_t *)apr_pcalloc(pool_out, sizeof(apr_file_t));
|
|
(*out)->pool = pool_out;
|
|
(*out)->fname = NULL;
|
|
(*out)->pipe = 1;
|
|
(*out)->timeout = -1;
|
|
(*out)->ungetchar = -1;
|
|
(*out)->eof_hit = 0;
|
|
(*out)->filePtr = 0;
|
|
(*out)->bufpos = 0;
|
|
(*out)->dataRead = 0;
|
|
(*out)->direction = 0;
|
|
(*out)->pOverlapped = NULL;
|
|
#if APR_FILES_AS_SOCKETS
|
|
(void) apr_pollset_create(&(*out)->pollset, 1, p, 0);
|
|
#endif
|
|
if (apr_os_level >= APR_WIN_NT) {
|
|
char rand[8];
|
|
int pid = getpid();
|
|
#define FMT_PIPE_NAME "\\\\.\\pipe\\apr-pipe-%x.%lx."
|
|
/* ^ ^ ^
|
|
* pid | |
|
|
* | |
|
|
* id |
|
|
* |
|
|
* hex-escaped rand[8] (16 bytes)
|
|
*/
|
|
char name[sizeof FMT_PIPE_NAME + 2 * sizeof(pid)
|
|
+ 2 * sizeof(id)
|
|
+ 2 * sizeof(rand)];
|
|
apr_size_t pos;
|
|
|
|
/* Create the read end of the pipe */
|
|
dwOpenMode = PIPE_ACCESS_INBOUND;
|
|
#ifdef FILE_FLAG_FIRST_PIPE_INSTANCE
|
|
dwOpenMode |= FILE_FLAG_FIRST_PIPE_INSTANCE;
|
|
#endif
|
|
if (blocking == APR_WRITE_BLOCK /* READ_NONBLOCK */
|
|
|| blocking == APR_FULL_NONBLOCK) {
|
|
dwOpenMode |= FILE_FLAG_OVERLAPPED;
|
|
(*in)->pOverlapped =
|
|
(OVERLAPPED*) apr_pcalloc((*in)->pool, sizeof(OVERLAPPED));
|
|
(*in)->pOverlapped->hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
(*in)->timeout = 0;
|
|
}
|
|
dwPipeMode = 0;
|
|
|
|
apr_generate_random_bytes(rand, sizeof rand);
|
|
pos = apr_snprintf(name, sizeof name, FMT_PIPE_NAME, pid, id++);
|
|
apr_escape_hex(name + pos, rand, sizeof rand, 0, NULL);
|
|
|
|
(*in)->filehand = CreateNamedPipe(name,
|
|
dwOpenMode,
|
|
dwPipeMode,
|
|
1, /* nMaxInstances, */
|
|
0, /* nOutBufferSize, */
|
|
65536, /* nInBufferSize, */
|
|
1, /* nDefaultTimeOut, */
|
|
&sa);
|
|
if ((*in)->filehand == INVALID_HANDLE_VALUE) {
|
|
apr_status_t rv = apr_get_os_error();
|
|
file_cleanup(*in);
|
|
return rv;
|
|
}
|
|
|
|
/* Create the write end of the pipe */
|
|
dwOpenMode = FILE_ATTRIBUTE_NORMAL;
|
|
if (blocking == APR_READ_BLOCK /* WRITE_NONBLOCK */
|
|
|| blocking == APR_FULL_NONBLOCK) {
|
|
dwOpenMode |= FILE_FLAG_OVERLAPPED;
|
|
(*out)->pOverlapped =
|
|
(OVERLAPPED*) apr_pcalloc((*out)->pool, sizeof(OVERLAPPED));
|
|
(*out)->pOverlapped->hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
(*out)->timeout = 0;
|
|
}
|
|
|
|
(*out)->filehand = CreateFile(name,
|
|
GENERIC_WRITE, /* access mode */
|
|
0, /* share mode */
|
|
&sa, /* Security attributes */
|
|
OPEN_EXISTING, /* dwCreationDisposition */
|
|
dwOpenMode, /* Pipe attributes */
|
|
NULL); /* handle to template file */
|
|
if ((*out)->filehand == INVALID_HANDLE_VALUE) {
|
|
apr_status_t rv = apr_get_os_error();
|
|
file_cleanup(*out);
|
|
file_cleanup(*in);
|
|
return rv;
|
|
}
|
|
}
|
|
else {
|
|
/* Pipes on Win9* are blocking. Live with it. */
|
|
if (!CreatePipe(&(*in)->filehand, &(*out)->filehand, &sa, 65536)) {
|
|
return apr_get_os_error();
|
|
}
|
|
}
|
|
|
|
apr_pool_cleanup_register((*in)->pool, (void *)(*in), file_cleanup,
|
|
apr_pool_cleanup_null);
|
|
apr_pool_cleanup_register((*out)->pool, (void *)(*out), file_cleanup,
|
|
apr_pool_cleanup_null);
|
|
return APR_SUCCESS;
|
|
#endif /* _WIN32_WCE */
|
|
}
|
|
|
|
|
|
APR_DECLARE(apr_status_t) apr_file_namedpipe_create(const char *filename,
|
|
apr_fileperms_t perm,
|
|
apr_pool_t *pool)
|
|
{
|
|
/* Not yet implemented, interface not suitable.
|
|
* Win32 requires the named pipe to be *opened* at the time it's
|
|
* created, and to do so, blocking or non blocking must be elected.
|
|
*/
|
|
return APR_ENOTIMPL;
|
|
}
|
|
|
|
|
|
/* XXX: Problem; we need to choose between blocking and nonblocking based
|
|
* on how *thefile was opened, and we don't have that information :-/
|
|
* Hack; assume a blocking socket, since the most common use for the fn
|
|
* would be to handle stdio-style or blocking pipes. Win32 doesn't have
|
|
* select() blocking for pipes anyways :(
|
|
*/
|
|
APR_DECLARE(apr_status_t) apr_os_pipe_put_ex(apr_file_t **file,
|
|
apr_os_file_t *thefile,
|
|
int register_cleanup,
|
|
apr_pool_t *pool)
|
|
{
|
|
(*file) = apr_pcalloc(pool, sizeof(apr_file_t));
|
|
(*file)->pool = pool;
|
|
(*file)->pipe = 1;
|
|
(*file)->timeout = -1;
|
|
(*file)->ungetchar = -1;
|
|
(*file)->filehand = *thefile;
|
|
#if APR_FILES_AS_SOCKETS
|
|
(void) apr_pollset_create(&(*file)->pollset, 1, pool, 0);
|
|
#endif
|
|
if (register_cleanup) {
|
|
apr_pool_cleanup_register(pool, *file, file_cleanup,
|
|
apr_pool_cleanup_null);
|
|
}
|
|
|
|
return APR_SUCCESS;
|
|
}
|
|
|
|
|
|
APR_DECLARE(apr_status_t) apr_os_pipe_put(apr_file_t **file,
|
|
apr_os_file_t *thefile,
|
|
apr_pool_t *pool)
|
|
{
|
|
return apr_os_pipe_put_ex(file, thefile, 0, pool);
|
|
}
|
|
|
|
static apr_status_t create_socket_pipe(SOCKET *rd, SOCKET *wr)
|
|
{
|
|
static int id = 0;
|
|
FD_SET rs;
|
|
SOCKET ls;
|
|
struct timeval socktm;
|
|
struct sockaddr_in pa;
|
|
struct sockaddr_in la;
|
|
struct sockaddr_in ca;
|
|
int nrd;
|
|
apr_status_t rv = APR_SUCCESS;
|
|
int ll = sizeof(la);
|
|
int lc = sizeof(ca);
|
|
unsigned long bm = 1;
|
|
int uid[2];
|
|
int iid[2];
|
|
|
|
*rd = INVALID_SOCKET;
|
|
*wr = INVALID_SOCKET;
|
|
|
|
/* Create the unique socket identifier
|
|
* so that we know the connection originated
|
|
* from us.
|
|
*/
|
|
uid[0] = getpid();
|
|
uid[1] = id++;
|
|
if ((ls = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET) {
|
|
return apr_get_netos_error();
|
|
}
|
|
|
|
pa.sin_family = AF_INET;
|
|
pa.sin_port = 0;
|
|
pa.sin_addr.s_addr = inet_addr("127.0.0.1");
|
|
|
|
if (bind(ls, (SOCKADDR *)&pa, sizeof(pa)) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
if (getsockname(ls, (SOCKADDR *)&la, &ll) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
if (listen(ls, 1) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
if ((*wr = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
if (connect(*wr, (SOCKADDR *)&la, sizeof(la)) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
if (send(*wr, (char *)uid, sizeof(uid), 0) != sizeof(uid)) {
|
|
if ((rv = apr_get_netos_error()) == 0) {
|
|
rv = APR_EINVAL;
|
|
}
|
|
goto cleanup;
|
|
}
|
|
if (ioctlsocket(ls, FIONBIO, &bm) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
for (;;) {
|
|
int ns;
|
|
int nc = 0;
|
|
/* Listening socket is nonblocking by now.
|
|
* The accept should create the socket
|
|
* immediatelly because we are connected already.
|
|
* However on buys systems this can take a while
|
|
* until winsock gets a chance to handle the events.
|
|
*/
|
|
FD_ZERO(&rs);
|
|
FD_SET(ls, &rs);
|
|
|
|
socktm.tv_sec = 1;
|
|
socktm.tv_usec = 0;
|
|
if ((ns = select(0, &rs, NULL, NULL, &socktm)) == SOCKET_ERROR) {
|
|
/* Accept still not signaled */
|
|
Sleep(100);
|
|
continue;
|
|
}
|
|
if (ns == 0) {
|
|
/* No connections in the last second */
|
|
continue;
|
|
}
|
|
if ((*rd = accept(ls, (SOCKADDR *)&ca, &lc)) == INVALID_SOCKET) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
/* Verify the connection by reading the send identification.
|
|
*/
|
|
do {
|
|
if (nc++)
|
|
Sleep(1);
|
|
nrd = recv(*rd, (char *)iid, sizeof(iid), 0);
|
|
rv = nrd == SOCKET_ERROR ? apr_get_netos_error() : APR_SUCCESS;
|
|
} while (APR_STATUS_IS_EAGAIN(rv));
|
|
|
|
if (nrd == sizeof(iid)) {
|
|
if (memcmp(uid, iid, sizeof(uid)) == 0) {
|
|
/* Wow, we recived what we send.
|
|
* Put read side of the pipe to the blocking
|
|
* mode and return.
|
|
*/
|
|
bm = 0;
|
|
if (ioctlsocket(*rd, FIONBIO, &bm) == SOCKET_ERROR) {
|
|
rv = apr_get_netos_error();
|
|
goto cleanup;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
else if (nrd == SOCKET_ERROR) {
|
|
goto cleanup;
|
|
}
|
|
closesocket(*rd);
|
|
}
|
|
/* We don't need the listening socket any more */
|
|
closesocket(ls);
|
|
return 0;
|
|
|
|
cleanup:
|
|
/* Don't leak resources */
|
|
if (*rd != INVALID_SOCKET)
|
|
closesocket(*rd);
|
|
if (*wr != INVALID_SOCKET)
|
|
closesocket(*wr);
|
|
|
|
*rd = INVALID_SOCKET;
|
|
*wr = INVALID_SOCKET;
|
|
closesocket(ls);
|
|
return rv;
|
|
}
|
|
|
|
static apr_status_t socket_pipe_cleanup(void *thefile)
|
|
{
|
|
apr_file_t *file = thefile;
|
|
if (file->filehand != INVALID_HANDLE_VALUE) {
|
|
shutdown((SOCKET)file->filehand, SD_BOTH);
|
|
closesocket((SOCKET)file->filehand);
|
|
file->filehand = INVALID_HANDLE_VALUE;
|
|
}
|
|
return APR_SUCCESS;
|
|
}
|
|
|
|
apr_status_t apr_file_socket_pipe_create(apr_file_t **in,
|
|
apr_file_t **out,
|
|
apr_pool_t *p)
|
|
{
|
|
apr_status_t rv;
|
|
SOCKET rd;
|
|
SOCKET wr;
|
|
|
|
if ((rv = create_socket_pipe(&rd, &wr)) != APR_SUCCESS) {
|
|
return rv;
|
|
}
|
|
(*in) = (apr_file_t *)apr_pcalloc(p, sizeof(apr_file_t));
|
|
(*in)->pool = p;
|
|
(*in)->fname = NULL;
|
|
(*in)->pipe = 1;
|
|
(*in)->timeout = -1;
|
|
(*in)->ungetchar = -1;
|
|
(*in)->eof_hit = 0;
|
|
(*in)->filePtr = 0;
|
|
(*in)->bufpos = 0;
|
|
(*in)->dataRead = 0;
|
|
(*in)->direction = 0;
|
|
(*in)->pOverlapped = (OVERLAPPED*)apr_pcalloc(p, sizeof(OVERLAPPED));
|
|
(*in)->filehand = (HANDLE)rd;
|
|
|
|
(*out) = (apr_file_t *)apr_pcalloc(p, sizeof(apr_file_t));
|
|
(*out)->pool = p;
|
|
(*out)->fname = NULL;
|
|
(*out)->pipe = 1;
|
|
(*out)->timeout = -1;
|
|
(*out)->ungetchar = -1;
|
|
(*out)->eof_hit = 0;
|
|
(*out)->filePtr = 0;
|
|
(*out)->bufpos = 0;
|
|
(*out)->dataRead = 0;
|
|
(*out)->direction = 0;
|
|
(*out)->pOverlapped = (OVERLAPPED*)apr_pcalloc(p, sizeof(OVERLAPPED));
|
|
(*out)->filehand = (HANDLE)wr;
|
|
|
|
apr_pool_cleanup_register(p, (void *)(*in), socket_pipe_cleanup,
|
|
apr_pool_cleanup_null);
|
|
apr_pool_cleanup_register(p, (void *)(*out), socket_pipe_cleanup,
|
|
apr_pool_cleanup_null);
|
|
|
|
return rv;
|
|
}
|
|
|
|
apr_status_t apr_file_socket_pipe_close(apr_file_t *file)
|
|
{
|
|
apr_status_t stat;
|
|
if (!file->pipe)
|
|
return apr_file_close(file);
|
|
if ((stat = socket_pipe_cleanup(file)) == APR_SUCCESS) {
|
|
apr_pool_cleanup_kill(file->pool, file, socket_pipe_cleanup);
|
|
|
|
if (file->mutex) {
|
|
apr_thread_mutex_destroy(file->mutex);
|
|
}
|
|
|
|
return APR_SUCCESS;
|
|
}
|
|
return stat;
|
|
}
|
|
|