271f76590f
Summary: Adjust code logic Signed-off-by: chenrun1 <chenrun1@xiaomi.com>
147 lines
4.7 KiB
C
147 lines
4.7 KiB
C
/****************************************************************************
|
|
* libs/libc/misc/lib_pathbuffer.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.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <nuttx/config.h>
|
|
#include <nuttx/mutex.h>
|
|
#include <nuttx/lib/lib.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
/****************************************************************************
|
|
* Pre-processor definitions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Private Types
|
|
****************************************************************************/
|
|
|
|
struct pathbuffer_s
|
|
{
|
|
mutex_t lock; /* Lock for the buffer */
|
|
unsigned int free_bitmap; /* Bitmap of free buffer */
|
|
char buffer[CONFIG_LIBC_MAX_PATHBUFFER][PATH_MAX];
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Data
|
|
****************************************************************************/
|
|
|
|
static struct pathbuffer_s g_pathbuffer =
|
|
{
|
|
NXMUTEX_INITIALIZER,
|
|
(1u << CONFIG_LIBC_MAX_PATHBUFFER) - 1,
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Name: lib_get_pathbuffer
|
|
*
|
|
* Description:
|
|
* The lib_get_pathbuffer() function returns a pointer to a temporary
|
|
* buffer. The buffer is allocated from a pool of pre-allocated buffers
|
|
* and if the pool is exhausted, a new buffer is allocated through
|
|
* kmm_malloc(). The size of the buffer is PATH_MAX, and must freed by
|
|
* calling lib_put_pathbuffer().
|
|
*
|
|
* Returned Value:
|
|
* On success, lib_get_pathbuffer() returns a pointer to a temporary
|
|
* buffer. On failure, NULL is returned.
|
|
*
|
|
****************************************************************************/
|
|
|
|
FAR char *lib_get_pathbuffer(void)
|
|
{
|
|
int index;
|
|
|
|
/* Try to find a free buffer */
|
|
|
|
nxmutex_lock(&g_pathbuffer.lock);
|
|
index = ffs(g_pathbuffer.free_bitmap) - 1;
|
|
if (index >= 0 && index < CONFIG_LIBC_MAX_PATHBUFFER)
|
|
{
|
|
g_pathbuffer.free_bitmap &= ~(1u << index);
|
|
nxmutex_unlock(&g_pathbuffer.lock);
|
|
return g_pathbuffer.buffer[index];
|
|
}
|
|
|
|
nxmutex_unlock(&g_pathbuffer.lock);
|
|
|
|
/* If no free buffer is found, allocate a new one if
|
|
* CONFIG_LIBC_PATHBUFFER_MALLOC is enabled
|
|
*/
|
|
|
|
#ifdef CONFIG_LIBC_PATHBUFFER_MALLOC
|
|
return lib_malloc(PATH_MAX);
|
|
#else
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Name: lib_put_pathbuffer
|
|
*
|
|
* Description:
|
|
* The lib_put_pathbuffer() function frees a temporary buffer that was
|
|
* allocated by lib_get_pathbuffer(). If the buffer was allocated
|
|
* dynamically, it is freed by calling kmm_free(). Otherwise, the buffer
|
|
* is marked as free in the pool of pre-allocated buffers.
|
|
*
|
|
* Returned Value:
|
|
* None
|
|
*
|
|
****************************************************************************/
|
|
|
|
void lib_put_pathbuffer(FAR char *buffer)
|
|
{
|
|
int index;
|
|
|
|
nxmutex_lock(&g_pathbuffer.lock);
|
|
index = (buffer - &g_pathbuffer.buffer[0][0]) / PATH_MAX;
|
|
|
|
if (index >= 0 && index < CONFIG_LIBC_MAX_PATHBUFFER)
|
|
{
|
|
/* Mark the corresponding bit as free */
|
|
|
|
g_pathbuffer.free_bitmap |= 1u << index;
|
|
nxmutex_unlock(&g_pathbuffer.lock);
|
|
return;
|
|
}
|
|
|
|
nxmutex_unlock(&g_pathbuffer.lock);
|
|
|
|
/* Free the buffer if it was dynamically allocated */
|
|
|
|
#ifdef CONFIG_LIBC_PATHBUFFER_MALLOC
|
|
return lib_free(buffer);
|
|
#endif
|
|
}
|