| /* | 
 |  * Copyright (C) 2008 The Android Open Source Project | 
 |  * | 
 |  * Licensed under the Apache License, Version 2.0 (the "License"); | 
 |  * you may not use this file except in compliance with the License. | 
 |  * You may obtain a copy of the License at | 
 |  * | 
 |  *      http://www.apache.org/licenses/LICENSE-2.0 | 
 |  * | 
 |  * Unless required by applicable law or agreed to in writing, software | 
 |  * distributed under the License is distributed on an "AS IS" BASIS, | 
 |  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
 |  * See the License for the specific language governing permissions and | 
 |  * limitations under the License. | 
 |  */ | 
 |  | 
 | #define LOG_TAG "IMemory" | 
 |  | 
 | #include <stdint.h> | 
 | #include <stdio.h> | 
 | #include <stdlib.h> | 
 | #include <fcntl.h> | 
 | #include <unistd.h> | 
 |  | 
 | #include <sys/types.h> | 
 | #include <sys/mman.h> | 
 |  | 
 | #include <binder/IMemory.h> | 
 | #include <utils/KeyedVector.h> | 
 | #include <utils/threads.h> | 
 | #include <utils/Atomic.h> | 
 | #include <binder/Parcel.h> | 
 | #include <utils/CallStack.h> | 
 |  | 
 | #define VERBOSE   0 | 
 |  | 
 | namespace android { | 
 | // --------------------------------------------------------------------------- | 
 |  | 
 | class HeapCache : public IBinder::DeathRecipient | 
 | { | 
 | public: | 
 |     HeapCache(); | 
 |     virtual ~HeapCache(); | 
 |  | 
 |     virtual void binderDied(const wp<IBinder>& who); | 
 |  | 
 |     sp<IMemoryHeap> find_heap(const sp<IBinder>& binder); | 
 |     void free_heap(const sp<IBinder>& binder); | 
 |     sp<IMemoryHeap> get_heap(const sp<IBinder>& binder); | 
 |     void dump_heaps(); | 
 |  | 
 | private: | 
 |     // For IMemory.cpp | 
 |     struct heap_info_t { | 
 |         sp<IMemoryHeap> heap; | 
 |         int32_t         count; | 
 |     }; | 
 |  | 
 |     void free_heap(const wp<IBinder>& binder); | 
 |  | 
 |     Mutex mHeapCacheLock; | 
 |     KeyedVector< wp<IBinder>, heap_info_t > mHeapCache; | 
 | }; | 
 |  | 
 | static sp<HeapCache> gHeapCache = new HeapCache(); | 
 |  | 
 | /******************************************************************************/ | 
 |  | 
 | enum { | 
 |     HEAP_ID = IBinder::FIRST_CALL_TRANSACTION | 
 | }; | 
 |  | 
 | class BpMemoryHeap : public BpInterface<IMemoryHeap> | 
 | { | 
 | public: | 
 |     BpMemoryHeap(const sp<IBinder>& impl); | 
 |     virtual ~BpMemoryHeap(); | 
 |  | 
 |     virtual int getHeapID() const; | 
 |     virtual void* getBase() const; | 
 |     virtual size_t getSize() const; | 
 |     virtual uint32_t getFlags() const; | 
 |     virtual uint32_t getOffset() const; | 
 |  | 
 | private: | 
 |     friend class IMemory; | 
 |     friend class HeapCache; | 
 |  | 
 |     // for debugging in this module | 
 |     static inline sp<IMemoryHeap> find_heap(const sp<IBinder>& binder) { | 
 |         return gHeapCache->find_heap(binder); | 
 |     } | 
 |     static inline void free_heap(const sp<IBinder>& binder) { | 
 |         gHeapCache->free_heap(binder); | 
 |     } | 
 |     static inline sp<IMemoryHeap> get_heap(const sp<IBinder>& binder) { | 
 |         return gHeapCache->get_heap(binder); | 
 |     } | 
 |     static inline void dump_heaps() { | 
 |         gHeapCache->dump_heaps(); | 
 |     } | 
 |  | 
 |     void assertMapped() const; | 
 |     void assertReallyMapped() const; | 
 |  | 
 |     mutable volatile int32_t mHeapId; | 
 |     mutable void*       mBase; | 
 |     mutable size_t      mSize; | 
 |     mutable uint32_t    mFlags; | 
 |     mutable uint32_t    mOffset; | 
 |     mutable bool        mRealHeap; | 
 |     mutable Mutex       mLock; | 
 | }; | 
 |  | 
 | // ---------------------------------------------------------------------------- | 
 |  | 
 | enum { | 
 |     GET_MEMORY = IBinder::FIRST_CALL_TRANSACTION | 
 | }; | 
 |  | 
 | class BpMemory : public BpInterface<IMemory> | 
 | { | 
 | public: | 
 |     BpMemory(const sp<IBinder>& impl); | 
 |     virtual ~BpMemory(); | 
 |     virtual sp<IMemoryHeap> getMemory(ssize_t* offset=0, size_t* size=0) const; | 
 |  | 
 | private: | 
 |     mutable sp<IMemoryHeap> mHeap; | 
 |     mutable ssize_t mOffset; | 
 |     mutable size_t mSize; | 
 | }; | 
 |  | 
 | /******************************************************************************/ | 
 |  | 
 | void* IMemory::fastPointer(const sp<IBinder>& binder, ssize_t offset) const | 
 | { | 
 |     sp<IMemoryHeap> realHeap = BpMemoryHeap::get_heap(binder); | 
 |     void* const base = realHeap->base(); | 
 |     if (base == MAP_FAILED) | 
 |         return 0; | 
 |     return static_cast<char*>(base) + offset; | 
 | } | 
 |  | 
 | void* IMemory::pointer() const { | 
 |     ssize_t offset; | 
 |     sp<IMemoryHeap> heap = getMemory(&offset); | 
 |     void* const base = heap!=0 ? heap->base() : MAP_FAILED; | 
 |     if (base == MAP_FAILED) | 
 |         return 0; | 
 |     return static_cast<char*>(base) + offset; | 
 | } | 
 |  | 
 | size_t IMemory::size() const { | 
 |     size_t size; | 
 |     getMemory(NULL, &size); | 
 |     return size; | 
 | } | 
 |  | 
 | ssize_t IMemory::offset() const { | 
 |     ssize_t offset; | 
 |     getMemory(&offset); | 
 |     return offset; | 
 | } | 
 |  | 
 | /******************************************************************************/ | 
 |  | 
 | BpMemory::BpMemory(const sp<IBinder>& impl) | 
 |     : BpInterface<IMemory>(impl), mOffset(0), mSize(0) | 
 | { | 
 | } | 
 |  | 
 | BpMemory::~BpMemory() | 
 | { | 
 | } | 
 |  | 
 | sp<IMemoryHeap> BpMemory::getMemory(ssize_t* offset, size_t* size) const | 
 | { | 
 |     if (mHeap == 0) { | 
 |         Parcel data, reply; | 
 |         data.writeInterfaceToken(IMemory::getInterfaceDescriptor()); | 
 |         if (remote()->transact(GET_MEMORY, data, &reply) == NO_ERROR) { | 
 |             sp<IBinder> heap = reply.readStrongBinder(); | 
 |             ssize_t o = reply.readInt32(); | 
 |             size_t s = reply.readInt32(); | 
 |             if (heap != 0) { | 
 |                 mHeap = interface_cast<IMemoryHeap>(heap); | 
 |                 if (mHeap != 0) { | 
 |                     mOffset = o; | 
 |                     mSize = s; | 
 |                 } | 
 |             } | 
 |         } | 
 |     } | 
 |     if (offset) *offset = mOffset; | 
 |     if (size) *size = mSize; | 
 |     return mHeap; | 
 | } | 
 |  | 
 | // --------------------------------------------------------------------------- | 
 |  | 
 | IMPLEMENT_META_INTERFACE(Memory, "android.utils.IMemory"); | 
 |  | 
 | BnMemory::BnMemory() { | 
 | } | 
 |  | 
 | BnMemory::~BnMemory() { | 
 | } | 
 |  | 
 | status_t BnMemory::onTransact( | 
 |     uint32_t code, const Parcel& data, Parcel* reply, uint32_t flags) | 
 | { | 
 |     switch(code) { | 
 |         case GET_MEMORY: { | 
 |             CHECK_INTERFACE(IMemory, data, reply); | 
 |             ssize_t offset; | 
 |             size_t size; | 
 |             reply->writeStrongBinder( getMemory(&offset, &size)->asBinder() ); | 
 |             reply->writeInt32(offset); | 
 |             reply->writeInt32(size); | 
 |             return NO_ERROR; | 
 |         } break; | 
 |         default: | 
 |             return BBinder::onTransact(code, data, reply, flags); | 
 |     } | 
 | } | 
 |  | 
 |  | 
 | /******************************************************************************/ | 
 |  | 
 | BpMemoryHeap::BpMemoryHeap(const sp<IBinder>& impl) | 
 |     : BpInterface<IMemoryHeap>(impl), | 
 |         mHeapId(-1), mBase(MAP_FAILED), mSize(0), mFlags(0), mOffset(0), mRealHeap(false) | 
 | { | 
 | } | 
 |  | 
 | BpMemoryHeap::~BpMemoryHeap() { | 
 |     if (mHeapId != -1) { | 
 |         close(mHeapId); | 
 |         if (mRealHeap) { | 
 |             // by construction we're the last one | 
 |             if (mBase != MAP_FAILED) { | 
 |                 sp<IBinder> binder = const_cast<BpMemoryHeap*>(this)->asBinder(); | 
 |  | 
 |                 if (VERBOSE) { | 
 |                     ALOGD("UNMAPPING binder=%p, heap=%p, size=%d, fd=%d", | 
 |                             binder.get(), this, mSize, mHeapId); | 
 |                     CallStack stack(LOG_TAG); | 
 |                 } | 
 |  | 
 |                 munmap(mBase, mSize); | 
 |             } | 
 |         } else { | 
 |             // remove from list only if it was mapped before | 
 |             sp<IBinder> binder = const_cast<BpMemoryHeap*>(this)->asBinder(); | 
 |             free_heap(binder); | 
 |         } | 
 |     } | 
 | } | 
 |  | 
 | void BpMemoryHeap::assertMapped() const | 
 | { | 
 |     if (mHeapId == -1) { | 
 |         sp<IBinder> binder(const_cast<BpMemoryHeap*>(this)->asBinder()); | 
 |         sp<BpMemoryHeap> heap(static_cast<BpMemoryHeap*>(find_heap(binder).get())); | 
 |         heap->assertReallyMapped(); | 
 |         if (heap->mBase != MAP_FAILED) { | 
 |             Mutex::Autolock _l(mLock); | 
 |             if (mHeapId == -1) { | 
 |                 mBase   = heap->mBase; | 
 |                 mSize   = heap->mSize; | 
 |                 mOffset = heap->mOffset; | 
 |                 android_atomic_write( dup( heap->mHeapId ), &mHeapId ); | 
 |             } | 
 |         } else { | 
 |             // something went wrong | 
 |             free_heap(binder); | 
 |         } | 
 |     } | 
 | } | 
 |  | 
 | void BpMemoryHeap::assertReallyMapped() const | 
 | { | 
 |     if (mHeapId == -1) { | 
 |  | 
 |         // remote call without mLock held, worse case scenario, we end up | 
 |         // calling transact() from multiple threads, but that's not a problem, | 
 |         // only mmap below must be in the critical section. | 
 |  | 
 |         Parcel data, reply; | 
 |         data.writeInterfaceToken(IMemoryHeap::getInterfaceDescriptor()); | 
 |         status_t err = remote()->transact(HEAP_ID, data, &reply); | 
 |         int parcel_fd = reply.readFileDescriptor(); | 
 |         ssize_t size = reply.readInt32(); | 
 |         uint32_t flags = reply.readInt32(); | 
 |         uint32_t offset = reply.readInt32(); | 
 |  | 
 |         ALOGE_IF(err, "binder=%p transaction failed fd=%d, size=%ld, err=%d (%s)", | 
 |                 asBinder().get(), parcel_fd, size, err, strerror(-err)); | 
 |  | 
 |         int fd = dup( parcel_fd ); | 
 |         ALOGE_IF(fd==-1, "cannot dup fd=%d, size=%ld, err=%d (%s)", | 
 |                 parcel_fd, size, err, strerror(errno)); | 
 |  | 
 |         int access = PROT_READ; | 
 |         if (!(flags & READ_ONLY)) { | 
 |             access |= PROT_WRITE; | 
 |         } | 
 |  | 
 |         Mutex::Autolock _l(mLock); | 
 |         if (mHeapId == -1) { | 
 |             mRealHeap = true; | 
 |             mBase = mmap(0, size, access, MAP_SHARED, fd, offset); | 
 |             if (mBase == MAP_FAILED) { | 
 |                 ALOGE("cannot map BpMemoryHeap (binder=%p), size=%ld, fd=%d (%s)", | 
 |                         asBinder().get(), size, fd, strerror(errno)); | 
 |                 close(fd); | 
 |             } else { | 
 |                 mSize = size; | 
 |                 mFlags = flags; | 
 |                 mOffset = offset; | 
 |                 android_atomic_write(fd, &mHeapId); | 
 |             } | 
 |         } | 
 |     } | 
 | } | 
 |  | 
 | int BpMemoryHeap::getHeapID() const { | 
 |     assertMapped(); | 
 |     return mHeapId; | 
 | } | 
 |  | 
 | void* BpMemoryHeap::getBase() const { | 
 |     assertMapped(); | 
 |     return mBase; | 
 | } | 
 |  | 
 | size_t BpMemoryHeap::getSize() const { | 
 |     assertMapped(); | 
 |     return mSize; | 
 | } | 
 |  | 
 | uint32_t BpMemoryHeap::getFlags() const { | 
 |     assertMapped(); | 
 |     return mFlags; | 
 | } | 
 |  | 
 | uint32_t BpMemoryHeap::getOffset() const { | 
 |     assertMapped(); | 
 |     return mOffset; | 
 | } | 
 |  | 
 | // --------------------------------------------------------------------------- | 
 |  | 
 | IMPLEMENT_META_INTERFACE(MemoryHeap, "android.utils.IMemoryHeap"); | 
 |  | 
 | BnMemoryHeap::BnMemoryHeap() { | 
 | } | 
 |  | 
 | BnMemoryHeap::~BnMemoryHeap() { | 
 | } | 
 |  | 
 | status_t BnMemoryHeap::onTransact( | 
 |         uint32_t code, const Parcel& data, Parcel* reply, uint32_t flags) | 
 | { | 
 |     switch(code) { | 
 |        case HEAP_ID: { | 
 |             CHECK_INTERFACE(IMemoryHeap, data, reply); | 
 |             reply->writeFileDescriptor(getHeapID()); | 
 |             reply->writeInt32(getSize()); | 
 |             reply->writeInt32(getFlags()); | 
 |             reply->writeInt32(getOffset()); | 
 |             return NO_ERROR; | 
 |         } break; | 
 |         default: | 
 |             return BBinder::onTransact(code, data, reply, flags); | 
 |     } | 
 | } | 
 |  | 
 | /*****************************************************************************/ | 
 |  | 
 | HeapCache::HeapCache() | 
 |     : DeathRecipient() | 
 | { | 
 | } | 
 |  | 
 | HeapCache::~HeapCache() | 
 | { | 
 | } | 
 |  | 
 | void HeapCache::binderDied(const wp<IBinder>& binder) | 
 | { | 
 |     //ALOGD("binderDied binder=%p", binder.unsafe_get()); | 
 |     free_heap(binder); | 
 | } | 
 |  | 
 | sp<IMemoryHeap> HeapCache::find_heap(const sp<IBinder>& binder) | 
 | { | 
 |     Mutex::Autolock _l(mHeapCacheLock); | 
 |     ssize_t i = mHeapCache.indexOfKey(binder); | 
 |     if (i>=0) { | 
 |         heap_info_t& info = mHeapCache.editValueAt(i); | 
 |         ALOGD_IF(VERBOSE, | 
 |                 "found binder=%p, heap=%p, size=%d, fd=%d, count=%d", | 
 |                 binder.get(), info.heap.get(), | 
 |                 static_cast<BpMemoryHeap*>(info.heap.get())->mSize, | 
 |                 static_cast<BpMemoryHeap*>(info.heap.get())->mHeapId, | 
 |                 info.count); | 
 |         android_atomic_inc(&info.count); | 
 |         return info.heap; | 
 |     } else { | 
 |         heap_info_t info; | 
 |         info.heap = interface_cast<IMemoryHeap>(binder); | 
 |         info.count = 1; | 
 |         //ALOGD("adding binder=%p, heap=%p, count=%d", | 
 |         //      binder.get(), info.heap.get(), info.count); | 
 |         mHeapCache.add(binder, info); | 
 |         return info.heap; | 
 |     } | 
 | } | 
 |  | 
 | void HeapCache::free_heap(const sp<IBinder>& binder)  { | 
 |     free_heap( wp<IBinder>(binder) ); | 
 | } | 
 |  | 
 | void HeapCache::free_heap(const wp<IBinder>& binder) | 
 | { | 
 |     sp<IMemoryHeap> rel; | 
 |     { | 
 |         Mutex::Autolock _l(mHeapCacheLock); | 
 |         ssize_t i = mHeapCache.indexOfKey(binder); | 
 |         if (i>=0) { | 
 |             heap_info_t& info(mHeapCache.editValueAt(i)); | 
 |             int32_t c = android_atomic_dec(&info.count); | 
 |             if (c == 1) { | 
 |                 ALOGD_IF(VERBOSE, | 
 |                         "removing binder=%p, heap=%p, size=%d, fd=%d, count=%d", | 
 |                         binder.unsafe_get(), info.heap.get(), | 
 |                         static_cast<BpMemoryHeap*>(info.heap.get())->mSize, | 
 |                         static_cast<BpMemoryHeap*>(info.heap.get())->mHeapId, | 
 |                         info.count); | 
 |                 rel = mHeapCache.valueAt(i).heap; | 
 |                 mHeapCache.removeItemsAt(i); | 
 |             } | 
 |         } else { | 
 |             ALOGE("free_heap binder=%p not found!!!", binder.unsafe_get()); | 
 |         } | 
 |     } | 
 | } | 
 |  | 
 | sp<IMemoryHeap> HeapCache::get_heap(const sp<IBinder>& binder) | 
 | { | 
 |     sp<IMemoryHeap> realHeap; | 
 |     Mutex::Autolock _l(mHeapCacheLock); | 
 |     ssize_t i = mHeapCache.indexOfKey(binder); | 
 |     if (i>=0)   realHeap = mHeapCache.valueAt(i).heap; | 
 |     else        realHeap = interface_cast<IMemoryHeap>(binder); | 
 |     return realHeap; | 
 | } | 
 |  | 
 | void HeapCache::dump_heaps() | 
 | { | 
 |     Mutex::Autolock _l(mHeapCacheLock); | 
 |     int c = mHeapCache.size(); | 
 |     for (int i=0 ; i<c ; i++) { | 
 |         const heap_info_t& info = mHeapCache.valueAt(i); | 
 |         BpMemoryHeap const* h(static_cast<BpMemoryHeap const *>(info.heap.get())); | 
 |         ALOGD("hey=%p, heap=%p, count=%d, (fd=%d, base=%p, size=%d)", | 
 |                 mHeapCache.keyAt(i).unsafe_get(), | 
 |                 info.heap.get(), info.count, | 
 |                 h->mHeapId, h->mBase, h->mSize); | 
 |     } | 
 | } | 
 |  | 
 |  | 
 | // --------------------------------------------------------------------------- | 
 | }; // namespace android |