mem.c 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <stddef.h>
  4. #include <stdarg.h>
  5. #include <unistd.h>
  6. #include <errno.h>
  7. #include <string.h>
  8. #include <fcntl.h>
  9. #include <sys/types.h>
  10. #include <sys/mman.h>
  11. #include "kern_util.h"
  12. #include "user.h"
  13. #include "user_util.h"
  14. #include "mem_user.h"
  15. #include "init.h"
  16. #include "os.h"
  17. #include "tempfile.h"
  18. #include "kern_constants.h"
  19. #include <sys/param.h>
  20. static char *tempdir = NULL;
  21. static void __init find_tempdir(void)
  22. {
  23. char *dirs[] = { "TMP", "TEMP", "TMPDIR", NULL };
  24. int i;
  25. char *dir = NULL;
  26. if(tempdir != NULL) return; /* We've already been called */
  27. for(i = 0; dirs[i]; i++){
  28. dir = getenv(dirs[i]);
  29. if((dir != NULL) && (*dir != '\0'))
  30. break;
  31. }
  32. if((dir == NULL) || (*dir == '\0'))
  33. dir = "/tmp";
  34. tempdir = malloc(strlen(dir) + 2);
  35. if(tempdir == NULL){
  36. fprintf(stderr, "Failed to malloc tempdir, "
  37. "errno = %d\n", errno);
  38. return;
  39. }
  40. strcpy(tempdir, dir);
  41. strcat(tempdir, "/");
  42. }
  43. /*
  44. * This proc still used in tt-mode
  45. * (file: kernel/tt/ptproxy/proxy.c, proc: start_debugger).
  46. * So it isn't 'static' yet.
  47. */
  48. int make_tempfile(const char *template, char **out_tempname, int do_unlink)
  49. {
  50. char *tempname;
  51. int fd;
  52. tempname = malloc(MAXPATHLEN);
  53. find_tempdir();
  54. if (template[0] != '/')
  55. strcpy(tempname, tempdir);
  56. else
  57. tempname[0] = '\0';
  58. strcat(tempname, template);
  59. fd = mkstemp(tempname);
  60. if(fd < 0){
  61. fprintf(stderr, "open - cannot create %s: %s\n", tempname,
  62. strerror(errno));
  63. goto out;
  64. }
  65. if(do_unlink && (unlink(tempname) < 0)){
  66. perror("unlink");
  67. goto out;
  68. }
  69. if(out_tempname){
  70. *out_tempname = tempname;
  71. } else {
  72. free(tempname);
  73. }
  74. return(fd);
  75. out:
  76. free(tempname);
  77. return -1;
  78. }
  79. #define TEMPNAME_TEMPLATE "vm_file-XXXXXX"
  80. /*
  81. * This proc is used in start_up.c
  82. * So it isn't 'static'.
  83. */
  84. int create_tmp_file(unsigned long long len)
  85. {
  86. int fd, err;
  87. char zero;
  88. fd = make_tempfile(TEMPNAME_TEMPLATE, NULL, 1);
  89. if(fd < 0) {
  90. exit(1);
  91. }
  92. err = fchmod(fd, 0777);
  93. if(err < 0){
  94. perror("os_mode_fd");
  95. exit(1);
  96. }
  97. if (lseek64(fd, len, SEEK_SET) < 0) {
  98. perror("os_seek_file");
  99. exit(1);
  100. }
  101. zero = 0;
  102. err = os_write_file(fd, &zero, 1);
  103. if(err != 1){
  104. errno = -err;
  105. perror("os_write_file");
  106. exit(1);
  107. }
  108. return(fd);
  109. }
  110. int create_mem_file(unsigned long long len)
  111. {
  112. int err, fd;
  113. fd = create_tmp_file(len);
  114. err = os_set_exec_close(fd, 1);
  115. if(err < 0){
  116. errno = -err;
  117. perror("exec_close");
  118. }
  119. return(fd);
  120. }