Skip to content

Commit

Permalink
open file with flag O_CLOEXEC
Browse files Browse the repository at this point in the history
  • Loading branch information
happyfish100 committed Jun 24, 2022
1 parent aa55061 commit e254b8e
Show file tree
Hide file tree
Showing 12 changed files with 44 additions and 33 deletions.
3 changes: 3 additions & 0 deletions HISTORY
Original file line number Diff line number Diff line change
@@ -1,4 +1,7 @@

Version 1.59 2022-06-24
* open file with flag O_CLOEXEC

Version 1.58 2022-06-04
* add function conn_pool_connect_server_ex1 to support service name
* add function conn_pool_get_connection_ex to support service name
Expand Down
4 changes: 2 additions & 2 deletions php-fastcommon/fastcommon.c
Original file line number Diff line number Diff line change
Expand Up @@ -1237,8 +1237,8 @@ static PHPFileContext *fetch_file_context(const char *filename)

static int fc_open_file(PHPFileContext *ctx)
{
if ((ctx->fd = open(ctx->filename, O_WRONLY |
O_CREAT | O_APPEND, 0644)) < 0)
if ((ctx->fd = open(ctx->filename, O_WRONLY | O_CREAT |
O_APPEND | O_CLOEXEC, 0644)) < 0)
{
logError("file: "__FILE__", line: %d, "
"open file \"%s\" to write fail, "
Expand Down
3 changes: 2 additions & 1 deletion src/buffered_file_writer.c
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,8 @@ int buffered_file_writer_open_ex(BufferedFileWriter *writer,
}

snprintf(writer->filename, sizeof(writer->filename), "%s", filename);
writer->fd = open(writer->filename, O_WRONLY | O_CREAT | O_TRUNC, mode);
writer->fd = open(writer->filename, O_WRONLY |
O_CREAT | O_TRUNC | O_CLOEXEC, mode);
if (writer->fd < 0)
{
result = errno != 0 ? errno : EIO;
Expand Down
9 changes: 5 additions & 4 deletions src/connection_pool.c
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ int conn_pool_connect_server_ex1(ConnectionInfo *conn,
}

if ((conn->sock=socketCreateEx2(conn->socket_domain, conn->ip_addr,
O_NONBLOCK, bind_ipaddr, &result)) < 0)
O_NONBLOCK | O_CLOEXEC, bind_ipaddr, &result)) < 0)
{
return result;
}
Expand Down Expand Up @@ -160,8 +160,8 @@ int conn_pool_async_connect_server_ex(ConnectionInfo *conn,
close(conn->sock);
}

if ((conn->sock=socketCreateEx2(conn->socket_domain,
conn->ip_addr, O_NONBLOCK, bind_ipaddr,
if ((conn->sock=socketCreateEx2(conn->socket_domain, conn->ip_addr,
O_NONBLOCK | O_CLOEXEC, bind_ipaddr,
&result)) < 0)
{
return result;
Expand All @@ -181,7 +181,8 @@ int conn_pool_async_connect_server_ex(ConnectionInfo *conn,
return result;
}

static inline void conn_pool_get_key(const ConnectionInfo *conn, char *key, int *key_len)
static inline void conn_pool_get_key(const ConnectionInfo *conn,
char *key, int *key_len)
{
*key_len = sprintf(key, "%s_%u", conn->ip_addr, conn->port);
}
Expand Down
5 changes: 3 additions & 2 deletions src/id_generator.c
Original file line number Diff line number Diff line change
Expand Up @@ -123,13 +123,14 @@ int id_generator_init_extra_ex(struct idg_context *context, const char *filename
mid = ntohl(ip_addr.s_addr) & ((1 << mid_bits) - 1);
}

if ((context->fd = open(filename, O_RDWR)) < 0)
if ((context->fd = open(filename, O_RDWR | O_CLOEXEC)) < 0)
{
if (errno == ENOENT)
{
mode_t old_mode;
old_mode = umask(0);
if ((context->fd=open(filename, O_RDWR | O_CREAT, mode)) < 0)
if ((context->fd=open(filename, O_RDWR | O_CREAT |
O_CLOEXEC, mode)) < 0)
{
result = errno != 0 ? errno : EACCES;
}
Expand Down
4 changes: 2 additions & 2 deletions src/logger.c
Original file line number Diff line number Diff line change
Expand Up @@ -158,8 +158,8 @@ static int log_print_header(LogContext *pContext)
static int log_open(LogContext *pContext)
{
int result;
if ((pContext->log_fd = open(pContext->log_filename, O_WRONLY | \
O_CREAT | O_APPEND | pContext->fd_flags, 0644)) < 0)
if ((pContext->log_fd = open(pContext->log_filename, O_WRONLY | O_CREAT |
O_APPEND | O_CLOEXEC | pContext->fd_flags, 0644)) < 0)
{
fprintf(stderr, "open log file \"%s\" to write fail, " \
"errno: %d, error info: %s\n", \
Expand Down
4 changes: 2 additions & 2 deletions src/pthread_func.h
Original file line number Diff line number Diff line change
Expand Up @@ -114,9 +114,9 @@ static inline void fc_timedwait_sec(pthread_mutex_t *lock,
{
struct timespec ts;

PTHREAD_MUTEX_LOCK(lock);
ts.tv_sec = get_current_time() + timeout;
ts.tv_nsec = 0;
PTHREAD_MUTEX_LOCK(lock);
pthread_cond_timedwait(cond, lock, &ts);
PTHREAD_MUTEX_UNLOCK(lock);
}
Expand All @@ -128,9 +128,9 @@ static inline void fc_timedwait_ms(pthread_mutex_t *lock,
struct timespec ts;

expires_ms = get_current_time_ms() + timeout_ms;
PTHREAD_MUTEX_LOCK(lock);
ts.tv_sec = expires_ms / 1000;
ts.tv_nsec = (expires_ms % 1000) * (1000 * 1000);
PTHREAD_MUTEX_LOCK(lock);
pthread_cond_timedwait(cond, lock, &ts);
PTHREAD_MUTEX_UNLOCK(lock);
}
Expand Down
6 changes: 3 additions & 3 deletions src/sched_thread.c
Original file line number Diff line number Diff line change
Expand Up @@ -127,9 +127,9 @@ static int sched_init_entries(ScheduleEntry *entries, const int count)

if (pEntry->interval <= 0)
{
logError("file: "__FILE__", line: %d, " \
"shedule interval %d <= 0", \
__LINE__, pEntry->interval);
logError("file: "__FILE__", line: %d, "
"shedule id: %d, interval %d <= 0",
__LINE__, pEntry->id, pEntry->interval);
return EINVAL;
}

Expand Down
15 changes: 8 additions & 7 deletions src/shared_func.c
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,7 @@ int getUserProcIds(const char *progName, const bool bAllOwners, \
if ((bAllOwners || (statbuf.st_uid == myuid)) && S_ISDIR(statbuf.st_mode))
{
sprintf(filepath, "%s/cmdline", fullpath);
if ((fd = open(filepath, O_RDONLY))<0)
if ((fd=open(filepath, O_RDONLY | O_CLOEXEC))<0)
{
continue;
}
Expand Down Expand Up @@ -734,7 +734,7 @@ int fc_get_file_line_count_ex(const char *filename,
return ENOMEM;
}

fd = open(filename, O_RDONLY);
fd = open(filename, O_RDONLY | O_CLOEXEC);
if (fd < 0) {
result = errno != 0 ? errno : EACCES;
logError("file: "__FILE__", line: %d, "
Expand Down Expand Up @@ -1314,7 +1314,7 @@ int getFileContent(const char *filename, char **buff, int64_t *file_size)
}
}

fd = open(filename, O_RDONLY);
fd = open(filename, O_RDONLY | O_CLOEXEC);
if (fd < 0)
{
*buff = NULL;
Expand Down Expand Up @@ -1368,7 +1368,7 @@ int getFileContentEx(const char *filename, char *buff,
return EINVAL;
}

fd = open(filename, O_RDONLY);
fd = open(filename, O_RDONLY | O_CLOEXEC);
if (fd < 0)
{
*size = 0;
Expand Down Expand Up @@ -1406,7 +1406,7 @@ int writeToFile(const char *filename, const char *buff, const int file_size)
int fd;
int result;

fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0644);
if (fd < 0)
{
result = errno != 0 ? errno : EIO;
Expand Down Expand Up @@ -1480,7 +1480,7 @@ int fc_copy_file(const char *src_filename, const char *dest_filename)
int bytes;
char buff[16 * 1024];

src_fd = open(src_filename, O_RDONLY);
src_fd = open(src_filename, O_RDONLY | O_CLOEXEC);
if (src_fd < 0)
{
result = errno != 0 ? errno : ENOENT;
Expand All @@ -1490,7 +1490,8 @@ int fc_copy_file(const char *src_filename, const char *dest_filename)
return result;
}

dest_fd = open(dest_filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
dest_fd = open(dest_filename, O_WRONLY | O_CREAT |
O_TRUNC | O_CLOEXEC, 0644);
if (dest_fd < 0)
{
result = errno != 0 ? errno : EIO;
Expand Down
2 changes: 1 addition & 1 deletion src/shared_func.h
Original file line number Diff line number Diff line change
Expand Up @@ -1198,7 +1198,7 @@ int fc_safe_write_file_init(SafeWriteFileInfo *fi,

static inline int fc_safe_write_file_open(SafeWriteFileInfo *fi)
{
const int flags = O_WRONLY | O_CREAT | O_TRUNC;
const int flags = O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC;
int result;

if ((fi->fd=open(fi->tmp_filename, flags, 0644)) < 0) {
Expand Down
16 changes: 9 additions & 7 deletions src/sockopt.c
Original file line number Diff line number Diff line change
Expand Up @@ -1476,7 +1476,8 @@ int tcprecvfile(int sock, const char *filename, const int64_t file_bytes, \
recv_func = tcprecvdata_ex;
}

write_fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
write_fd = open(filename, O_WRONLY | O_CREAT |
O_TRUNC | O_CLOEXEC, 0644);
if (write_fd < 0)
{
return errno != 0 ? errno : EACCES;
Expand Down Expand Up @@ -1544,7 +1545,7 @@ int tcprecvfile(int sock, const char *filename, const int64_t file_bytes, \
break;
}

read_fd = open(filename, O_RDONLY);
read_fd = open(filename, O_RDONLY | O_CLOEXEC);
if (read_fd < 0)
{
return errno != 0 ? errno : EACCES;
Expand Down Expand Up @@ -1627,7 +1628,8 @@ int tcprecvfile_ex(int sock, const char *filename, const int64_t file_bytes, \
recv_func = tcprecvdata_ex;
}

fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
fd = open(filename, O_WRONLY | O_CREAT |
O_TRUNC | O_CLOEXEC, 0644);
if (fd < 0)
{
return errno != 0 ? errno : EACCES;
Expand Down Expand Up @@ -1761,7 +1763,7 @@ int tcpsendfile_ex(int sock, const char *filename, const int64_t file_offset, \
int64_t remain_bytes;
#endif

fd = open(filename, O_RDONLY);
fd = open(filename, O_RDONLY | O_CLOEXEC);
if (fd < 0)
{
*total_send_bytes = 0;
Expand All @@ -1779,7 +1781,7 @@ int tcpsendfile_ex(int sock, const char *filename, const int64_t file_offset, \

if (flags & O_NONBLOCK)
{
if (fcntl(sock, F_SETFL, flags & ~O_NONBLOCK) == -1)
if (fcntl(sock, F_SETFL, flags & ~O_NONBLOCK) < 0)
{
*total_send_bytes = 0;
return errno != 0 ? errno : EACCES;
Expand Down Expand Up @@ -1873,7 +1875,7 @@ int tcpsendfile_ex(int sock, const char *filename, const int64_t file_offset, \

if (flags & O_NONBLOCK) //restore
{
if (fcntl(sock, F_SETFL, flags) == -1)
if (fcntl(sock, F_SETFL, flags) < 0)
{
result = errno != 0 ? errno : EACCES;
}
Expand Down Expand Up @@ -2124,7 +2126,7 @@ int tcpsetnonblockopt(int fd)
return errno != 0 ? errno : EACCES;
}

if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
if (fcntl(fd, F_SETFL, flags | (O_NONBLOCK | O_CLOEXEC)) < 0)
{
logError("file: "__FILE__", line: %d, " \
"fcntl failed, errno: %d, error info: %s.", \
Expand Down
6 changes: 4 additions & 2 deletions src/system_info.c
Original file line number Diff line number Diff line change
Expand Up @@ -843,7 +843,7 @@ int get_device_block_size(const char *device, int *block_size)
int fd;
size_t bs;

if ((fd=open(device, O_RDONLY)) < 0) {
if ((fd=open(device, O_RDONLY | O_CLOEXEC)) < 0) {
result = errno != 0 ? errno : ENOENT;
logError("file: "__FILE__", line: %d, "
"open device %s fail, errno: %d, error info: %s",
Expand Down Expand Up @@ -885,7 +885,9 @@ static int get_block_size_by_write(const char *path, int *block_size)

snprintf(tmp_filename, sizeof(tmp_filename),
"%s/.blksize-test.tmp", path);
if ((fd=open(tmp_filename, O_WRONLY | O_CREAT | O_DIRECT, 0755)) < 0) {
if ((fd=open(tmp_filename, O_WRONLY | O_CREAT |
O_DIRECT | O_CLOEXEC, 0755)) < 0)
{
result = errno != 0 ? errno : ENOENT;
logError("file: "__FILE__", line: %d, "
"open file %s fail, errno: %d, error info: %s",
Expand Down

0 comments on commit e254b8e

Please sign in to comment.