NASP/cuda-samples
53
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2026-01-09 20:07:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug 5502937: Set TEMPDIR as socket create dir to avoid socket create permission isue on systems like WSL

Browse Source
This commit is contained in:
shawnz 2025-09-05 11:55:44 +08:00
parent 3e6d1858af
commit b0a2f2049f
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Common/helper_multiprocess.cpp
View File

@ -188,7 +188,7 @@ int ipcCreateSocket(ipcHandle *&handle, const char *name,
char path_name[50]; char path_name[50];
// Create unique name for the socket with path if SOCK_FOLDER is set. // Create unique name for the socket with path if SOCK_FOLDER is set.
sprintf(path_name, "%s%u", SOCK_FOLDER, getpid()); sprintf(path_name, "%s/%u", getSocketFolder().c_str(), getpid());
unlink(path_name); unlink(path_name);
memset(&servaddr, 0, sizeof(servaddr)); memset(&servaddr, 0, sizeof(servaddr));
@ -230,7 +230,7 @@ int ipcOpenSocket(ipcHandle *&handle) {
char temp[50]; char temp[50];
// Create unique name for the socket with path if SOCK_FOLDER is set. // Create unique name for the socket with path if SOCK_FOLDER is set.
sprintf(temp, "%s%u", SOCK_FOLDER, getpid()); sprintf(temp, "%s/%u", getSocketFolder().c_str(), getpid());
strcpy(cliaddr.sun_path, temp); strcpy(cliaddr.sun_path, temp);
if (bind(sock, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0) { if (bind(sock, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0) {
@ -367,7 +367,7 @@ int ipcSendShareableHandle(ipcHandle *handle,
memset(&cliaddr, 0, sizeof(cliaddr)); memset(&cliaddr, 0, sizeof(cliaddr));
cliaddr.sun_family = AF_UNIX; cliaddr.sun_family = AF_UNIX;
char temp[20]; char temp[20];
sprintf(temp, "%s%u", SOCK_FOLDER, process); sprintf(temp, "%s/%u", getSocketFolder().c_str(), process);
strcpy(cliaddr.sun_path, temp); strcpy(cliaddr.sun_path, temp);
len = sizeof(cliaddr); len = sizeof(cliaddr);

10
Common/helper_multiprocess.h
View File

@ -52,14 +52,12 @@
#include <memory.h> #include <memory.h>
#include <sys/un.h> #include <sys/un.h>
#endif #endif
#include <filesystem>
#include <vector> #include <vector>
// Define "/tmp" as socket creating folder for QNX inline std::string getSocketFolder() {
#if defined(__QNX__) return std::filesystem::temp_directory_path().string();
#define SOCK_FOLDER "/tmp/" }
#else
#define SOCK_FOLDER ""
#endif
typedef struct sharedMemoryInfo_st { typedef struct sharedMemoryInfo_st {
void *addr; void *addr;