OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2013 Google Inc. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license that can be | |
5 * found in the LICENSE file. | |
6 */ | |
7 | |
8 #include "SkPurgeableMemoryBlock.h" | |
9 | |
10 #include "android/ashmem.h" | |
11 #include <sys/mman.h> | |
12 #include <unistd.h> | |
13 | |
14 bool SkPurgeableMemoryBlock::IsSupported() { | |
15 return true; | |
16 } | |
17 | |
18 #ifdef SK_DEBUG | |
19 bool SkPurgeableMemoryBlock::PlatformSupportsPurgingAllUnpinnedBlocks() { | |
20 return false; | |
21 } | |
22 | |
23 bool SkPurgeableMemoryBlock::PurgeAllUnpinnedBlocks() { | |
24 return false; | |
25 } | |
26 | |
27 bool SkPurgeableMemoryBlock::purge() { | |
28 SkASSERT(!fPinned); | |
29 if (-1 != fFD) { | |
30 ashmem_purge_all_caches(fFD); | |
31 return true; | |
32 } else { | |
33 return false; | |
34 } | |
35 } | |
36 #endif | |
37 | |
38 // ashmem likes lengths on page boundaries. | |
39 static size_t round_to_page_size(size_t size) { | |
40 const size_t mask = getpagesize() - 1; | |
41 size_t newSize = (size + mask) & ~mask; | |
42 return newSize; | |
43 } | |
44 | |
45 SkPurgeableMemoryBlock::SkPurgeableMemoryBlock(size_t size) | |
46 : fAddr(NULL) | |
47 , fSize(round_to_page_size(size)) | |
48 , fPinned(false) | |
49 , fFD(-1) { | |
50 } | |
51 | |
52 SkPurgeableMemoryBlock::~SkPurgeableMemoryBlock() { | |
53 if (-1 != fFD) { | |
54 munmap(fAddr, fSize); | |
55 close(fFD); | |
56 } | |
57 } | |
58 | |
59 void* SkPurgeableMemoryBlock::pin(SkPurgeableMemoryBlock::PinResult* pinResult)
{ | |
60 SkASSERT(!fPinned); | |
61 if (-1 == fFD) { | |
62 int fd = ashmem_create_region(NULL, fSize); | |
63 if (-1 == fd) { | |
64 SkDebugf("ashmem_create_region failed\n"); | |
65 return NULL; | |
66 } | |
67 | |
68 int err = ashmem_set_prot_region(fd, PROT_READ | PROT_WRITE); | |
69 if (err != 0) { | |
70 SkDebugf("ashmem_set_prot_region failed\n"); | |
71 close(fd); | |
72 return NULL; | |
73 } | |
74 | |
75 void* addr = mmap(NULL, fSize, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd,
0); | |
76 if (-1 == (long) addr) { | |
77 SkDebugf("mmap failed\n"); | |
78 close(fd); | |
79 return NULL; | |
80 } | |
81 fAddr = addr; | |
82 fFD = fd; | |
83 (void) ashmem_pin_region(fd, 0, 0); | |
84 *pinResult = kUninitialized_PinResult; | |
85 fPinned = true; | |
86 } else { | |
87 int pin = ashmem_pin_region(fFD, 0, 0); | |
88 if (ASHMEM_NOT_PURGED == pin) { | |
89 fPinned = true; | |
90 *pinResult = kRetained_PinResult; | |
91 } else if (ASHMEM_WAS_PURGED == pin) { | |
92 fPinned = true; | |
93 *pinResult = kUninitialized_PinResult; | |
94 } else { | |
95 // Failed. | |
96 munmap(fAddr, fSize); | |
97 close(fFD); | |
98 fFD = -1; | |
99 fAddr = NULL; | |
100 } | |
101 } | |
102 return fAddr; | |
103 } | |
104 | |
105 void SkPurgeableMemoryBlock::unpin() { | |
106 if (-1 != fFD) { | |
107 ashmem_unpin_region(fFD, 0, 0); | |
108 fPinned = false; | |
109 } | |
110 } | |
OLD | NEW |