ion drivers: add 'query client infomation' interface for vpu

diff --git a/drivers/gpu/ion/ion.c b/drivers/gpu/ion/ion.c
index 378e6b1feef347a1f01db2f601b9d0bfa4406dc6..f5074c0348d44f159ecb4cd7a6f48a2f949918bc 100755 (executable)
--- a/drivers/gpu/ion/ion.c
+++ b/drivers/gpu/ion/ion.c
@@ -1111,6 +1111,40 @@ static long ion_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
                        return err;
                break;
        }
+        case ION_GET_CLIENT: 
+        {
+                struct ion_handle *handle;
+               struct ion_client_data data;
+                struct rb_node *n;
+
+               if (copy_from_user(&data, (void __user *)arg,
+                                  sizeof(struct ion_client_data)))
+                       return -EFAULT;
+
+               mutex_lock(&client->lock);
+                switch (data.type) {
+                        case ION_TYPE_GET_TOTAL_SIZE:
+                                data.total_size = 0;
+                               for (n = rb_first(&client->handles); n; n = rb_next(n)) {
+                                       handle = rb_entry(n, struct ion_handle, node);
+                                        data.total_size += handle->buffer->size;
+                                }
+                                break;
+                        case ION_TYPE_SIZE_GET_COUNT:
+                                data.count = 0;
+                               for (n = rb_first(&client->handles); n; n = rb_next(n)) {
+                                       handle = rb_entry(n, struct ion_handle, node);
+                                        if(handle->buffer->size == data.size)
+                                                data.count++;
+                                }
+                                break;
+                }
+               mutex_unlock(&client->lock);
+               if (copy_to_user((void __user *)arg, &data,
+                                sizeof(struct ion_client_data)))
+                       return -EFAULT;
+                break;
+        }
        default:
                return -ENOTTY;
        }

diff --git a/include/linux/ion.h b/include/linux/ion.h
index 202766b7d0e5b319306fb100dcb198b3e005eafc..1063072b28fb21229fd2644c343c5d3d5996953d 100755 (executable)
--- a/include/linux/ion.h
+++ b/include/linux/ion.h
@@ -312,6 +312,16 @@ struct ion_flush_data {
        void *virt;
        size_t size;
 };
+struct ion_client_data {
+#define ION_TYPE_GET_TOTAL_SIZE  0
+#define ION_TYPE_SIZE_GET_COUNT  1
+        unsigned int type;
+        union {
+                size_t size;
+                size_t total_size;
+        };
+        unsigned int count;
+};
 #define ION_IOC_MAGIC          'I'
 
 /**
@@ -371,4 +381,5 @@ struct ion_flush_data {
 #define ION_CACHE_CLEAN                _IOWR(ION_IOC_MAGIC, 8, struct ion_flush_data)
 #define ION_CACHE_INVALID      _IOWR(ION_IOC_MAGIC, 9, struct ion_flush_data)
 #define ION_GET_PHYS           _IOWR(ION_IOC_MAGIC, 10, unsigned long)
+#define ION_GET_CLIENT         _IOWR(ION_IOC_MAGIC, 11, struct ion_client_data)
 #endif /* _LINUX_ION_H */