res_stir_shaken: refactor utility function

Refactors temp file utility function into file.c.

ASTERISK-29809 #close

Change-Id: Ife478708c8f2b127239cb73c1755ef18c0bf431b
This commit is contained in:
Naveen Albert 2021-12-16 19:41:56 +00:00 committed by Friendly Automation
parent dffc4408e7
commit 78b7ab11a4
3 changed files with 41 additions and 34 deletions

View File

@ -146,6 +146,23 @@ int ast_filecopy(const char *oldname, const char *newname, const char *fmt);
*/
FILE *ast_file_mkftemp(char *template, mode_t mode);
/*!
* \brief Create a temporary file located at path
*
* \note The directory containing path will be created if it does not exist
* \note This function assumes path does not end with a '/'
*
* \param path The directory path to create the file in
* \param filename Function allocates memory and stores full filename (including path) here
* \param template_name mkstemp template to use. Must end with XXXXXX.
*
* \note filename will need to be freed with ast_free if this function succeeds
*
* \retval -1 on failure
* \return file descriptor on success
*/
int ast_file_fdtemp(const char *path, char **filename, const char *template_name);
/*!
* \brief Callback called for each file found when reading directories
* \param dir_name the name of the directory

View File

@ -199,6 +199,26 @@ FILE *ast_file_mkftemp(char *template, mode_t mode)
return p;
}
int ast_file_fdtemp(const char *path, char **filename, const char *template_name)
{
int fd;
if (ast_asprintf(filename, "%s/%s", path, template_name) < 0) {
ast_log(LOG_ERROR, "Failed to set up temporary file path\n");
return -1;
}
ast_mkdir(path, 0644);
if ((fd = mkstemp(*filename)) < 0) {
ast_log(LOG_NOTICE, "Failed to create temporary file\n");
ast_free(*filename);
return -1;
}
return fd;
}
int ast_stopstream(struct ast_channel *tmp)
{
ast_channel_lock(tmp);

View File

@ -20,6 +20,7 @@
#include "asterisk/utils.h"
#include "asterisk/logger.h"
#include "asterisk/file.h"
#include "curl.h"
#include "general.h"
#include "stir_shaken.h"
@ -151,42 +152,11 @@ static CURL *get_curl_instance(struct curl_cb_data *data)
return curl;
}
/*!
* \brief Create a temporary file located at path
*
* \note This function assumes path does not end with a '/'
*
* \param path The directory path to create the file in
* \param filename Function allocates memory and stores full filename (including path) here
*
* \retval -1 on failure
* \return file descriptor on success
*/
static int create_temp_file(const char *path, char **filename)
{
const char *template_name = "certXXXXXX";
int fd;
if (ast_asprintf(filename, "%s/%s", path, template_name) < 0) {
ast_log(LOG_ERROR, "Failed to set up temporary file path for CURL\n");
return -1;
}
ast_mkdir(path, 0644);
if ((fd = mkstemp(*filename)) < 0) {
ast_log(LOG_NOTICE, "Failed to create temporary file for CURL\n");
ast_free(*filename);
return -1;
}
return fd;
}
char *curl_public_key(const char *public_cert_url, const char *path, struct curl_cb_data *data)
{
FILE *public_key_file;
RAII_VAR(char *, tmp_filename, NULL, ast_free);
const char *template_name = "certXXXXXX";
char *filename;
char *serial;
int fd;
@ -199,9 +169,9 @@ char *curl_public_key(const char *public_cert_url, const char *path, struct curl
/* For now, it's fine to pass in path as is - it shouldn't end with a '/'. However,
* if we decide to change how certificates are stored in the future (configurable paths),
* then we will need to check to see if path ends with '/', copy everything up to the '/',
* and use this new variable for create_temp_file as well as for ast_asprintf below.
* and use this new variable for ast_create_temp_file as well as for ast_asprintf below.
*/
fd = create_temp_file(path, &tmp_filename);
fd = ast_file_fdtemp(path, &tmp_filename, template_name);
if (fd == -1) {
ast_log(LOG_ERROR, "Failed to get temporary file descriptor for CURL\n");
return NULL;