Make rlmain runnable again.

facebookresearch · Feb 11, 2022 · 65d3e33 · 65d3e33
1 parent b96ef90
commit 65d3e33
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/sys/unix/rlmain.cc b/sys/unix/rlmain.cc
@@ -116,11 +116,26 @@ main(int argc, char **argv)
         fopen("nle.ttyrec.bz2", "a"), fclose);
 
     nle_settings settings;
-    strncpy(settings.hackdir, getenv("HACKDIR"), sizeof(settings.hackdir));
+    char *hackdir = getenv("HACKDIR");
+
+    if (hackdir)
+        strncpy(settings.hackdir, hackdir, sizeof(settings.hackdir));
+    else
+        throw std::runtime_error(
+            "Set HACKDIR env variable to NetHack installation");
 
     ScopedTC tc;
 
-    Instance nle("libnethack.so", &obs, ttyrec.get(), nullptr, &settings);
+    char *cwd = getcwd(nullptr, 0); /* glibc's allocating version. */
+    std::string path(cwd);
+    free(cwd);
+
+    if (path.back() != '/')
+        path.append("/");
+
+    path.append("libnethack.so");
+
+    Instance nle(path, &obs, ttyrec.get(), nullptr, &settings);
     if (argc > 1 && argv[1][0] == 'r') {
         randgame(nle, &obs, 3, &settings);
     } else {