rename dynamic library in wrapper

Author: yongman

src/redis-server-wrap.c

@@ -2,23 +2,20 @@
 #include <stdlib.h>
 #include <dlfcn.h>
 #include <signal.h>
+#include <unistd.h>
 
 #include <jemalloc/jemalloc.h>
 
 #include "server.h"
 
 #define LIB_REDIS_PATH "./redis-server.so"
+#define LIB_REDIS_BUSY_PATH "./redis-server.so.busy"
 
 #define malloc(size) je_malloc(size)
-#define calloc(count,size) je_calloc(count,size)
-#define realloc(ptr,size) je_realloc(ptr,size)
 #define free(ptr) je_free(ptr)
-#define mallocx(size,flags) je_mallocx(size,flags)
-#define dallocx(ptr,flags) je_dallocx(ptr,flags)
 
 typedef int (*ENTRY)(int, char **, struct wrapperContext *);
 typedef int (*RELOAD)(int);
-typedef char *(*GETDB)();
 
 int main(int argc, char ** argv)
 {
@@ -28,6 +25,7 @@ int main(int argc, char ** argv)
  RELOAD reload = NULL;
  int ret;
  char *mem = NULL;
+ int err = 0;
 
  ctx.reload = 0;
 
@@ -42,9 +40,30 @@ int main(int argc, char ** argv)
  }
  free(mem);
 
+ /* check so file */
+ if (access(LIB_REDIS_BUSY_PATH, R_OK) != 0) {
+ err = errno;
+ fprintf(stderr, "file %s access error:%s\n", LIB_REDIS_BUSY_PATH, strerror(err));
+
+ if (access(LIB_REDIS_PATH, R_OK) != 0) {
+ err = errno;
+ fprintf(stderr, "file %s access error:%s\n", LIB_REDIS_PATH, strerror(err));
+ fprintf(stderr, "dynamic library file not exists, please check.\n");
+ exit(EXIT_FAILURE);
+ } else {
+ /* rename so file to busy */
+ if (rename(LIB_REDIS_PATH, LIB_REDIS_BUSY_PATH) != 0) {
+ err = errno;
+ fprintf(stderr, "rename file failed:%s\n", strerror(err));
+ exit(EXIT_FAILURE);
+ }
+ }
+ }
+
+
  for(;;) {
  fprintf(stderr, "load dynamic library\n");
- handle = dlopen(LIB_REDIS_PATH, RTLD_LAZY);
+ handle = dlopen(LIB_REDIS_BUSY_PATH, RTLD_LAZY);
  if (!handle) {
  fprintf(stderr, "%s\n", dlerror());
  exit(EXIT_FAILURE);
@@ -71,6 +90,19 @@ int main(int argc, char ** argv)
  }
  ctx.reload = 1;
  ctx.reloadTimes++;
+
+ /* use new so file */
+ if (access(LIB_REDIS_PATH, R_OK) == 0) {
+ /* rename so file */
+ if (rename(LIB_REDIS_PATH, LIB_REDIS_BUSY_PATH) != 0) {
+ err = errno;
+ fprintf(stderr, "rename %s to %s failed: %s", LIB_REDIS_PATH,
+ LIB_REDIS_BUSY_PATH, strerror(err));
+ fprintf(stderr, "use old so file instead");
+ }
+ } else {
+ fprintf(stderr, "no newer so file found, using old so file instead");
+ }
  }
  exit(EXIT_SUCCESS);
 }