How to implement readlink to find the path

后端 未结 5 1916
悲&欢浪女
悲&欢浪女 2020-12-09 03:33

Using the readlink function used as a solution to How do I find the location of the executable in C?, how would I get the path into a char array? Also, what do the variables

相关标签:
5条回答
  • 2020-12-09 03:47
    char *
    readlink_malloc (const char *filename)
    {
      int size = 100;
      char *buffer = NULL;
    
      while (1)
        {
          buffer = (char *) xrealloc (buffer, size);
          int nchars = readlink (filename, buffer, size);
          if (nchars < 0)
            {
              free (buffer);
              return NULL;
            }
          if (nchars < size)
            return buffer;
          size *= 2;
        }
    }
    

    Taken from: http://www.delorie.com/gnu/docs/glibc/libc_279.html

    0 讨论(0)
  • 2020-12-09 03:48

    This Use the readlink() function properly for the correct uses of the readlink function.

    If you have your path in a std::string, you could do something like this:

    #include <unistd.h>
    #include <limits.h>
    
    std::string do_readlink(std::string const& path) {
        char buff[PATH_MAX];
        ssize_t len = ::readlink(path.c_str(), buff, sizeof(buff)-1);
        if (len != -1) {
          buff[len] = '\0';
          return std::string(buff);
        }
        /* handle error condition */
    }
    

    If you're only after a fixed path:

    std::string get_selfpath() {
        char buff[PATH_MAX];
        ssize_t len = ::readlink("/proc/self/exe", buff, sizeof(buff)-1);
        if (len != -1) {
          buff[len] = '\0';
          return std::string(buff);
        }
        /* handle error condition */
    }
    

    To use it:

    int main()
    {
      std::string selfpath = get_selfpath();
      std::cout << selfpath << std::endl;
      return 0;
    }
    
    0 讨论(0)
  • 2020-12-09 03:56
    #include <stdlib.h>
    #include <unistd.h>
    
    static char *exename(void)
    {
        char *buf;
        char *newbuf;
        size_t cap;
        ssize_t len;
    
        buf = NULL;
        for (cap = 64; cap <= 16384; cap *= 2) {
            newbuf = realloc(buf, cap);
            if (newbuf == NULL) {
                break;
            }
            buf = newbuf;
            len = readlink("/proc/self/exe", buf, cap);
            if (len < 0) {
                break;
            }
            if ((size_t)len < cap) {
                buf[len] = 0;
                return buf;
            }
        }
        free(buf);
        return NULL;
    }
    
    #include <stdio.h>
    
    int main(void)
    {
        char *e = exename();
        printf("%s\n", e ? e : "unknown");
        free(e);
        return 0;
    }
    

    This uses the traditional "when you don't know the right buffer size, reallocate increasing powers of two" trick. We assume that allocating less than 64 bytes for a pathname is not worth the effort. We also assume that an executable pathname as long as 16384 (2**14) bytes has to indicate some kind of anomaly in how the program was installed, and it's not useful to know the pathname as we'll soon encounter bigger problems to worry about.

    There is no need to bother with constants like PATH_MAX. Reserving so much memory is overkill for almost all pathnames, and as noted in another answer, it's not guaranteed to be the actual upper limit anyway. For this application, we can pick a common-sense upper limit such as 16384. Even for applications with no common-sense upper limit, reallocating increasing powers of two is a good approach. You only need log n calls for a n-byte result, and the amount of memory capacity you waste is proportional to the length of the result. It also avoids race conditions where the length of the string changes between the realloc() and the readlink().

    0 讨论(0)
  • 2020-12-09 04:01

    Let's look at what the manpage says:

     readlink() places the contents of the symbolic link path in the buffer
     buf, which has size bufsiz.  readlink does not append a NUL character to
     buf.
    

    OK. Should be simple enough. Given your buffer of 1024 chars:

     char buf[1024];
    
     /* The manpage says it won't null terminate.  Let's zero the buffer. */
     memset(buf, 0, sizeof(buf));
    
     /* Note we use sizeof(buf)-1 since we may need an extra char for NUL. */
     if (readlink("/proc/self/exe", buf, sizeof(buf)-1) < 0)
     {
        /* There was an error...  Perhaps the path does not exist
         * or the buffer is not big enough.  errno has the details. */
        perror("readlink");
        return -1;
     }
    
    0 讨论(0)
  • 2020-12-09 04:08

    Accepted answer is almost correct, except you can't rely on PATH_MAX because it is

    not guaranteed to be defined per POSIX if the system does not have such limit.

    (From readlink(2) manpage)

    Also, when it's defined it doesn't always represent the "true" limit. (See http://insanecoding.blogspot.fr/2007/11/pathmax-simply-isnt.html )

    The readlink's manpage also give a way to do that on symlink :

    Using a statically sized buffer might not provide enough room for the symbolic link contents. The required size for the buffer can be obtained from the stat.st_size value returned by a call to lstat(2) on the link. However, the number of bytes written by readlink() and read‐ linkat() should be checked to make sure that the size of the symbolic link did not increase between the calls.

    However in the case of /proc/self/exe/ as for most of /proc files, stat.st_size would be 0. The only remaining solution I see is to resize buffer while it doesn't fit.

    I suggest the use of vector<char> as follow for this purpose:

    std::string get_selfpath()
    {
        std::vector<char> buf(400);
        ssize_t len;
    
        do
        {
            buf.resize(buf.size() + 100);
            len = ::readlink("/proc/self/exe", &(buf[0]), buf.size());
        } while (buf.size() == len);
    
        if (len > 0)
        {
            buf[len] = '\0';
            return (std::string(&(buf[0])));
        }
        /* handle error */
        return "";
    }
    
    0 讨论(0)
提交回复
热议问题