/crypto/heimdal/lib/roken/write_pid.c

https://bitbucket.org/freebsd/freebsd-head/ · C · 90 lines · 50 code · 8 blank · 32 comment · 13 complexity · c50e2b9bd2334dcd21c662a4bcb4cb3d MD5 · raw file

  1. /*
  2. * Copyright (c) 1999 - 2001 Kungliga Tekniska Hรถgskolan
  3. * (Royal Institute of Technology, Stockholm, Sweden).
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions
  8. * are met:
  9. *
  10. * 1. Redistributions of source code must retain the above copyright
  11. * notice, this list of conditions and the following disclaimer.
  12. *
  13. * 2. Redistributions in binary form must reproduce the above copyright
  14. * notice, this list of conditions and the following disclaimer in the
  15. * documentation and/or other materials provided with the distribution.
  16. *
  17. * 3. Neither the name of the Institute nor the names of its contributors
  18. * may be used to endorse or promote products derived from this software
  19. * without specific prior written permission.
  20. *
  21. * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24. * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31. * SUCH DAMAGE.
  32. */
  33. #include <config.h>
  34. #include "roken.h"
  35. ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
  36. pid_file_write (const char *progname)
  37. {
  38. char *ret = NULL;
  39. FILE *fp;
  40. if (asprintf (&ret, "%s%s.pid", _PATH_VARRUN, progname) < 0 || ret == NULL)
  41. return NULL;
  42. fp = fopen (ret, "w");
  43. if (fp == NULL) {
  44. free (ret);
  45. return NULL;
  46. }
  47. fprintf (fp, "%u", (unsigned)getpid());
  48. fclose (fp);
  49. return ret;
  50. }
  51. ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
  52. pid_file_delete (char **filename)
  53. {
  54. if (*filename != NULL) {
  55. unlink (*filename);
  56. free (*filename);
  57. *filename = NULL;
  58. }
  59. }
  60. #ifndef HAVE_PIDFILE
  61. static char *pidfile_path;
  62. static void
  63. pidfile_cleanup(void)
  64. {
  65. if(pidfile_path != NULL)
  66. pid_file_delete(&pidfile_path);
  67. }
  68. ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
  69. pidfile(const char *basename)
  70. {
  71. if(pidfile_path != NULL)
  72. return;
  73. if(basename == NULL)
  74. basename = getprogname();
  75. pidfile_path = pid_file_write(basename);
  76. #if defined(HAVE_ATEXIT)
  77. atexit(pidfile_cleanup);
  78. #elif defined(HAVE_ON_EXIT)
  79. on_exit(pidfile_cleanup);
  80. #endif
  81. }
  82. #endif