mem.c 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  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[MAXPATHLEN];
  51. int fd;
  52. find_tempdir();
  53. if (*template != '/')
  54. strcpy(tempname, tempdir);
  55. else
  56. *tempname = 0;
  57. strcat(tempname, template);
  58. fd = mkstemp(tempname);
  59. if(fd < 0){
  60. fprintf(stderr, "open - cannot create %s: %s\n", tempname,
  61. strerror(errno));
  62. return -1;
  63. }
  64. if(do_unlink && (unlink(tempname) < 0)){
  65. perror("unlink");
  66. return -1;
  67. }
  68. if(out_tempname){
  69. *out_tempname = strdup(tempname);
  70. if(*out_tempname == NULL){
  71. perror("strdup");
  72. return -1;
  73. }
  74. }
  75. return(fd);
  76. }
  77. #define TEMPNAME_TEMPLATE "vm_file-XXXXXX"
  78. /*
  79. * This proc is used in start_up.c
  80. * So it isn't 'static'.
  81. */
  82. int create_tmp_file(unsigned long long len)
  83. {
  84. int fd, err;
  85. char zero;
  86. fd = make_tempfile(TEMPNAME_TEMPLATE, NULL, 1);
  87. if(fd < 0) {
  88. exit(1);
  89. }
  90. err = fchmod(fd, 0777);
  91. if(err < 0){
  92. perror("os_mode_fd");
  93. exit(1);
  94. }
  95. if (lseek64(fd, len, SEEK_SET) < 0) {
  96. perror("os_seek_file");
  97. exit(1);
  98. }
  99. zero = 0;
  100. err = os_write_file(fd, &zero, 1);
  101. if(err != 1){
  102. errno = -err;
  103. perror("os_write_file");
  104. exit(1);
  105. }
  106. return(fd);
  107. }
  108. static int create_anon_file(unsigned long long len)
  109. {
  110. void *addr;
  111. int fd;
  112. fd = open("/dev/anon", O_RDWR);
  113. if(fd < 0) {
  114. perror("opening /dev/anon");
  115. exit(1);
  116. }
  117. addr = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd, 0);
  118. if(addr == MAP_FAILED){
  119. perror("mapping physmem file");
  120. exit(1);
  121. }
  122. munmap(addr, len);
  123. return(fd);
  124. }
  125. extern int have_devanon;
  126. int create_mem_file(unsigned long long len)
  127. {
  128. int err, fd;
  129. if(have_devanon)
  130. fd = create_anon_file(len);
  131. else fd = create_tmp_file(len);
  132. err = os_set_exec_close(fd, 1);
  133. if(err < 0){
  134. errno = -err;
  135. perror("exec_close");
  136. }
  137. return(fd);
  138. }