dso-data.c 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. #include "util.h"
  2. #include <stdlib.h>
  3. #include <sys/types.h>
  4. #include <sys/stat.h>
  5. #include <fcntl.h>
  6. #include <string.h>
  7. #include "machine.h"
  8. #include "symbol.h"
  9. #include "tests.h"
  10. #define TEST_ASSERT_VAL(text, cond) \
  11. do { \
  12. if (!(cond)) { \
  13. pr_debug("FAILED %s:%d %s\n", __FILE__, __LINE__, text); \
  14. return -1; \
  15. } \
  16. } while (0)
  17. static char *test_file(int size)
  18. {
  19. static char buf_templ[] = "/tmp/test-XXXXXX";
  20. char *templ = buf_templ;
  21. int fd, i;
  22. unsigned char *buf;
  23. fd = mkstemp(templ);
  24. buf = malloc(size);
  25. if (!buf) {
  26. close(fd);
  27. return NULL;
  28. }
  29. for (i = 0; i < size; i++)
  30. buf[i] = (unsigned char) ((int) i % 10);
  31. if (size != write(fd, buf, size))
  32. templ = NULL;
  33. close(fd);
  34. return templ;
  35. }
  36. #define TEST_FILE_SIZE (DSO__DATA_CACHE_SIZE * 20)
  37. struct test_data_offset {
  38. off_t offset;
  39. u8 data[10];
  40. int size;
  41. };
  42. struct test_data_offset offsets[] = {
  43. /* Fill first cache page. */
  44. {
  45. .offset = 10,
  46. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  47. .size = 10,
  48. },
  49. /* Read first cache page. */
  50. {
  51. .offset = 10,
  52. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  53. .size = 10,
  54. },
  55. /* Fill cache boundary pages. */
  56. {
  57. .offset = DSO__DATA_CACHE_SIZE - DSO__DATA_CACHE_SIZE % 10,
  58. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  59. .size = 10,
  60. },
  61. /* Read cache boundary pages. */
  62. {
  63. .offset = DSO__DATA_CACHE_SIZE - DSO__DATA_CACHE_SIZE % 10,
  64. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  65. .size = 10,
  66. },
  67. /* Fill final cache page. */
  68. {
  69. .offset = TEST_FILE_SIZE - 10,
  70. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  71. .size = 10,
  72. },
  73. /* Read final cache page. */
  74. {
  75. .offset = TEST_FILE_SIZE - 10,
  76. .data = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
  77. .size = 10,
  78. },
  79. /* Read final cache page. */
  80. {
  81. .offset = TEST_FILE_SIZE - 3,
  82. .data = { 7, 8, 9, 0, 0, 0, 0, 0, 0, 0 },
  83. .size = 3,
  84. },
  85. };
  86. int test__dso_data(void)
  87. {
  88. struct machine machine;
  89. struct dso *dso;
  90. char *file = test_file(TEST_FILE_SIZE);
  91. size_t i;
  92. TEST_ASSERT_VAL("No test file", file);
  93. memset(&machine, 0, sizeof(machine));
  94. dso = dso__new((const char *)file);
  95. /* Basic 10 bytes tests. */
  96. for (i = 0; i < ARRAY_SIZE(offsets); i++) {
  97. struct test_data_offset *data = &offsets[i];
  98. ssize_t size;
  99. u8 buf[10];
  100. memset(buf, 0, 10);
  101. size = dso__data_read_offset(dso, &machine, data->offset,
  102. buf, 10);
  103. TEST_ASSERT_VAL("Wrong size", size == data->size);
  104. TEST_ASSERT_VAL("Wrong data", !memcmp(buf, data->data, 10));
  105. }
  106. /* Read cross multiple cache pages. */
  107. {
  108. ssize_t size;
  109. int c;
  110. u8 *buf;
  111. buf = malloc(TEST_FILE_SIZE);
  112. TEST_ASSERT_VAL("ENOMEM\n", buf);
  113. /* First iteration to fill caches, second one to read them. */
  114. for (c = 0; c < 2; c++) {
  115. memset(buf, 0, TEST_FILE_SIZE);
  116. size = dso__data_read_offset(dso, &machine, 10,
  117. buf, TEST_FILE_SIZE);
  118. TEST_ASSERT_VAL("Wrong size",
  119. size == (TEST_FILE_SIZE - 10));
  120. for (i = 0; i < (size_t)size; i++)
  121. TEST_ASSERT_VAL("Wrong data",
  122. buf[i] == (i % 10));
  123. }
  124. free(buf);
  125. }
  126. dso__delete(dso);
  127. unlink(file);
  128. return 0;
  129. }