TempFile.cpp 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. /*
  2. * Copyright (c) 2020-2021, the SerenityOS developers.
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include "TempFile.h"
  7. #include <AK/Random.h>
  8. #include <LibCore/File.h>
  9. #include <fcntl.h>
  10. #include <stdlib.h>
  11. #include <sys/stat.h>
  12. namespace Core {
  13. NonnullOwnPtr<TempFile> TempFile::create(Type type)
  14. {
  15. return adopt_own(*new TempFile(type));
  16. }
  17. DeprecatedString TempFile::create_temp(Type type)
  18. {
  19. char name_template[] = "/tmp/tmp.XXXXXX";
  20. switch (type) {
  21. case Type::File: {
  22. auto fd = mkstemp(name_template);
  23. VERIFY(fd >= 0);
  24. close(fd);
  25. break;
  26. }
  27. case Type::Directory: {
  28. auto fd = mkdtemp(name_template);
  29. VERIFY(fd != nullptr);
  30. break;
  31. }
  32. }
  33. return DeprecatedString { name_template };
  34. }
  35. TempFile::TempFile(Type type)
  36. : m_type(type)
  37. , m_path(create_temp(type))
  38. {
  39. }
  40. TempFile::~TempFile()
  41. {
  42. File::RecursionMode recursion_allowed { File::RecursionMode::Disallowed };
  43. if (m_type == Type::Directory)
  44. recursion_allowed = File::RecursionMode::Allowed;
  45. auto rc = File::remove(m_path.characters(), recursion_allowed);
  46. if (rc.is_error()) {
  47. warnln("File::remove failed: {}", rc.error().string_literal());
  48. }
  49. }
  50. }