safearray_windows.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337
  1. // +build windows
  2. package ole
  3. import (
  4. "unsafe"
  5. )
  6. var (
  7. procSafeArrayAccessData, _ = modoleaut32.FindProc("SafeArrayAccessData")
  8. procSafeArrayAllocData, _ = modoleaut32.FindProc("SafeArrayAllocData")
  9. procSafeArrayAllocDescriptor, _ = modoleaut32.FindProc("SafeArrayAllocDescriptor")
  10. procSafeArrayAllocDescriptorEx, _ = modoleaut32.FindProc("SafeArrayAllocDescriptorEx")
  11. procSafeArrayCopy, _ = modoleaut32.FindProc("SafeArrayCopy")
  12. procSafeArrayCopyData, _ = modoleaut32.FindProc("SafeArrayCopyData")
  13. procSafeArrayCreate, _ = modoleaut32.FindProc("SafeArrayCreate")
  14. procSafeArrayCreateEx, _ = modoleaut32.FindProc("SafeArrayCreateEx")
  15. procSafeArrayCreateVector, _ = modoleaut32.FindProc("SafeArrayCreateVector")
  16. procSafeArrayCreateVectorEx, _ = modoleaut32.FindProc("SafeArrayCreateVectorEx")
  17. procSafeArrayDestroy, _ = modoleaut32.FindProc("SafeArrayDestroy")
  18. procSafeArrayDestroyData, _ = modoleaut32.FindProc("SafeArrayDestroyData")
  19. procSafeArrayDestroyDescriptor, _ = modoleaut32.FindProc("SafeArrayDestroyDescriptor")
  20. procSafeArrayGetDim, _ = modoleaut32.FindProc("SafeArrayGetDim")
  21. procSafeArrayGetElement, _ = modoleaut32.FindProc("SafeArrayGetElement")
  22. procSafeArrayGetElemsize, _ = modoleaut32.FindProc("SafeArrayGetElemsize")
  23. procSafeArrayGetIID, _ = modoleaut32.FindProc("SafeArrayGetIID")
  24. procSafeArrayGetLBound, _ = modoleaut32.FindProc("SafeArrayGetLBound")
  25. procSafeArrayGetUBound, _ = modoleaut32.FindProc("SafeArrayGetUBound")
  26. procSafeArrayGetVartype, _ = modoleaut32.FindProc("SafeArrayGetVartype")
  27. procSafeArrayLock, _ = modoleaut32.FindProc("SafeArrayLock")
  28. procSafeArrayPtrOfIndex, _ = modoleaut32.FindProc("SafeArrayPtrOfIndex")
  29. procSafeArrayUnaccessData, _ = modoleaut32.FindProc("SafeArrayUnaccessData")
  30. procSafeArrayUnlock, _ = modoleaut32.FindProc("SafeArrayUnlock")
  31. procSafeArrayPutElement, _ = modoleaut32.FindProc("SafeArrayPutElement")
  32. //procSafeArrayRedim, _ = modoleaut32.FindProc("SafeArrayRedim") // TODO
  33. //procSafeArraySetIID, _ = modoleaut32.FindProc("SafeArraySetIID") // TODO
  34. procSafeArrayGetRecordInfo, _ = modoleaut32.FindProc("SafeArrayGetRecordInfo")
  35. procSafeArraySetRecordInfo, _ = modoleaut32.FindProc("SafeArraySetRecordInfo")
  36. )
  37. // safeArrayAccessData returns raw array pointer.
  38. //
  39. // AKA: SafeArrayAccessData in Windows API.
  40. // Todo: Test
  41. func safeArrayAccessData(safearray *SafeArray) (element uintptr, err error) {
  42. err = convertHresultToError(
  43. procSafeArrayAccessData.Call(
  44. uintptr(unsafe.Pointer(safearray)),
  45. uintptr(unsafe.Pointer(&element))))
  46. return
  47. }
  48. // safeArrayUnaccessData releases raw array.
  49. //
  50. // AKA: SafeArrayUnaccessData in Windows API.
  51. func safeArrayUnaccessData(safearray *SafeArray) (err error) {
  52. err = convertHresultToError(procSafeArrayUnaccessData.Call(uintptr(unsafe.Pointer(safearray))))
  53. return
  54. }
  55. // safeArrayAllocData allocates SafeArray.
  56. //
  57. // AKA: SafeArrayAllocData in Windows API.
  58. func safeArrayAllocData(safearray *SafeArray) (err error) {
  59. err = convertHresultToError(procSafeArrayAllocData.Call(uintptr(unsafe.Pointer(safearray))))
  60. return
  61. }
  62. // safeArrayAllocDescriptor allocates SafeArray.
  63. //
  64. // AKA: SafeArrayAllocDescriptor in Windows API.
  65. func safeArrayAllocDescriptor(dimensions uint32) (safearray *SafeArray, err error) {
  66. err = convertHresultToError(
  67. procSafeArrayAllocDescriptor.Call(uintptr(dimensions), uintptr(unsafe.Pointer(&safearray))))
  68. return
  69. }
  70. // safeArrayAllocDescriptorEx allocates SafeArray.
  71. //
  72. // AKA: SafeArrayAllocDescriptorEx in Windows API.
  73. func safeArrayAllocDescriptorEx(variantType VT, dimensions uint32) (safearray *SafeArray, err error) {
  74. err = convertHresultToError(
  75. procSafeArrayAllocDescriptorEx.Call(
  76. uintptr(variantType),
  77. uintptr(dimensions),
  78. uintptr(unsafe.Pointer(&safearray))))
  79. return
  80. }
  81. // safeArrayCopy returns copy of SafeArray.
  82. //
  83. // AKA: SafeArrayCopy in Windows API.
  84. func safeArrayCopy(original *SafeArray) (safearray *SafeArray, err error) {
  85. err = convertHresultToError(
  86. procSafeArrayCopy.Call(
  87. uintptr(unsafe.Pointer(original)),
  88. uintptr(unsafe.Pointer(&safearray))))
  89. return
  90. }
  91. // safeArrayCopyData duplicates SafeArray into another SafeArray object.
  92. //
  93. // AKA: SafeArrayCopyData in Windows API.
  94. func safeArrayCopyData(original *SafeArray, duplicate *SafeArray) (err error) {
  95. err = convertHresultToError(
  96. procSafeArrayCopyData.Call(
  97. uintptr(unsafe.Pointer(original)),
  98. uintptr(unsafe.Pointer(duplicate))))
  99. return
  100. }
  101. // safeArrayCreate creates SafeArray.
  102. //
  103. // AKA: SafeArrayCreate in Windows API.
  104. func safeArrayCreate(variantType VT, dimensions uint32, bounds *SafeArrayBound) (safearray *SafeArray, err error) {
  105. sa, _, err := procSafeArrayCreate.Call(
  106. uintptr(variantType),
  107. uintptr(dimensions),
  108. uintptr(unsafe.Pointer(bounds)))
  109. safearray = (*SafeArray)(unsafe.Pointer(&sa))
  110. return
  111. }
  112. // safeArrayCreateEx creates SafeArray.
  113. //
  114. // AKA: SafeArrayCreateEx in Windows API.
  115. func safeArrayCreateEx(variantType VT, dimensions uint32, bounds *SafeArrayBound, extra uintptr) (safearray *SafeArray, err error) {
  116. sa, _, err := procSafeArrayCreateEx.Call(
  117. uintptr(variantType),
  118. uintptr(dimensions),
  119. uintptr(unsafe.Pointer(bounds)),
  120. extra)
  121. safearray = (*SafeArray)(unsafe.Pointer(sa))
  122. return
  123. }
  124. // safeArrayCreateVector creates SafeArray.
  125. //
  126. // AKA: SafeArrayCreateVector in Windows API.
  127. func safeArrayCreateVector(variantType VT, lowerBound int32, length uint32) (safearray *SafeArray, err error) {
  128. sa, _, err := procSafeArrayCreateVector.Call(
  129. uintptr(variantType),
  130. uintptr(lowerBound),
  131. uintptr(length))
  132. safearray = (*SafeArray)(unsafe.Pointer(sa))
  133. return
  134. }
  135. // safeArrayCreateVectorEx creates SafeArray.
  136. //
  137. // AKA: SafeArrayCreateVectorEx in Windows API.
  138. func safeArrayCreateVectorEx(variantType VT, lowerBound int32, length uint32, extra uintptr) (safearray *SafeArray, err error) {
  139. sa, _, err := procSafeArrayCreateVectorEx.Call(
  140. uintptr(variantType),
  141. uintptr(lowerBound),
  142. uintptr(length),
  143. extra)
  144. safearray = (*SafeArray)(unsafe.Pointer(sa))
  145. return
  146. }
  147. // safeArrayDestroy destroys SafeArray object.
  148. //
  149. // AKA: SafeArrayDestroy in Windows API.
  150. func safeArrayDestroy(safearray *SafeArray) (err error) {
  151. err = convertHresultToError(procSafeArrayDestroy.Call(uintptr(unsafe.Pointer(safearray))))
  152. return
  153. }
  154. // safeArrayDestroyData destroys SafeArray object.
  155. //
  156. // AKA: SafeArrayDestroyData in Windows API.
  157. func safeArrayDestroyData(safearray *SafeArray) (err error) {
  158. err = convertHresultToError(procSafeArrayDestroyData.Call(uintptr(unsafe.Pointer(safearray))))
  159. return
  160. }
  161. // safeArrayDestroyDescriptor destroys SafeArray object.
  162. //
  163. // AKA: SafeArrayDestroyDescriptor in Windows API.
  164. func safeArrayDestroyDescriptor(safearray *SafeArray) (err error) {
  165. err = convertHresultToError(procSafeArrayDestroyDescriptor.Call(uintptr(unsafe.Pointer(safearray))))
  166. return
  167. }
  168. // safeArrayGetDim is the amount of dimensions in the SafeArray.
  169. //
  170. // SafeArrays may have multiple dimensions. Meaning, it could be
  171. // multidimensional array.
  172. //
  173. // AKA: SafeArrayGetDim in Windows API.
  174. func safeArrayGetDim(safearray *SafeArray) (dimensions *uint32, err error) {
  175. l, _, err := procSafeArrayGetDim.Call(uintptr(unsafe.Pointer(safearray)))
  176. dimensions = (*uint32)(unsafe.Pointer(l))
  177. return
  178. }
  179. // safeArrayGetElementSize is the element size in bytes.
  180. //
  181. // AKA: SafeArrayGetElemsize in Windows API.
  182. func safeArrayGetElementSize(safearray *SafeArray) (length *uint32, err error) {
  183. l, _, err := procSafeArrayGetElemsize.Call(uintptr(unsafe.Pointer(safearray)))
  184. length = (*uint32)(unsafe.Pointer(l))
  185. return
  186. }
  187. // safeArrayGetElement retrieves element at given index.
  188. func safeArrayGetElement(safearray *SafeArray, index int32, pv unsafe.Pointer) error {
  189. return convertHresultToError(
  190. procSafeArrayGetElement.Call(
  191. uintptr(unsafe.Pointer(safearray)),
  192. uintptr(unsafe.Pointer(&index)),
  193. uintptr(pv)))
  194. }
  195. // safeArrayGetElementString retrieves element at given index and converts to string.
  196. func safeArrayGetElementString(safearray *SafeArray, index int32) (str string, err error) {
  197. var element *int16
  198. err = convertHresultToError(
  199. procSafeArrayGetElement.Call(
  200. uintptr(unsafe.Pointer(safearray)),
  201. uintptr(unsafe.Pointer(&index)),
  202. uintptr(unsafe.Pointer(&element))))
  203. str = BstrToString(*(**uint16)(unsafe.Pointer(&element)))
  204. SysFreeString(element)
  205. return
  206. }
  207. // safeArrayGetIID is the InterfaceID of the elements in the SafeArray.
  208. //
  209. // AKA: SafeArrayGetIID in Windows API.
  210. func safeArrayGetIID(safearray *SafeArray) (guid *GUID, err error) {
  211. err = convertHresultToError(
  212. procSafeArrayGetIID.Call(
  213. uintptr(unsafe.Pointer(safearray)),
  214. uintptr(unsafe.Pointer(&guid))))
  215. return
  216. }
  217. // safeArrayGetLBound returns lower bounds of SafeArray.
  218. //
  219. // SafeArrays may have multiple dimensions. Meaning, it could be
  220. // multidimensional array.
  221. //
  222. // AKA: SafeArrayGetLBound in Windows API.
  223. func safeArrayGetLBound(safearray *SafeArray, dimension uint32) (lowerBound int32, err error) {
  224. err = convertHresultToError(
  225. procSafeArrayGetLBound.Call(
  226. uintptr(unsafe.Pointer(safearray)),
  227. uintptr(dimension),
  228. uintptr(unsafe.Pointer(&lowerBound))))
  229. return
  230. }
  231. // safeArrayGetUBound returns upper bounds of SafeArray.
  232. //
  233. // SafeArrays may have multiple dimensions. Meaning, it could be
  234. // multidimensional array.
  235. //
  236. // AKA: SafeArrayGetUBound in Windows API.
  237. func safeArrayGetUBound(safearray *SafeArray, dimension uint32) (upperBound int32, err error) {
  238. err = convertHresultToError(
  239. procSafeArrayGetUBound.Call(
  240. uintptr(unsafe.Pointer(safearray)),
  241. uintptr(dimension),
  242. uintptr(unsafe.Pointer(&upperBound))))
  243. return
  244. }
  245. // safeArrayGetVartype returns data type of SafeArray.
  246. //
  247. // AKA: SafeArrayGetVartype in Windows API.
  248. func safeArrayGetVartype(safearray *SafeArray) (varType uint16, err error) {
  249. err = convertHresultToError(
  250. procSafeArrayGetVartype.Call(
  251. uintptr(unsafe.Pointer(safearray)),
  252. uintptr(unsafe.Pointer(&varType))))
  253. return
  254. }
  255. // safeArrayLock locks SafeArray for reading to modify SafeArray.
  256. //
  257. // This must be called during some calls to ensure that another process does not
  258. // read or write to the SafeArray during editing.
  259. //
  260. // AKA: SafeArrayLock in Windows API.
  261. func safeArrayLock(safearray *SafeArray) (err error) {
  262. err = convertHresultToError(procSafeArrayLock.Call(uintptr(unsafe.Pointer(safearray))))
  263. return
  264. }
  265. // safeArrayUnlock unlocks SafeArray for reading.
  266. //
  267. // AKA: SafeArrayUnlock in Windows API.
  268. func safeArrayUnlock(safearray *SafeArray) (err error) {
  269. err = convertHresultToError(procSafeArrayUnlock.Call(uintptr(unsafe.Pointer(safearray))))
  270. return
  271. }
  272. // safeArrayPutElement stores the data element at the specified location in the
  273. // array.
  274. //
  275. // AKA: SafeArrayPutElement in Windows API.
  276. func safeArrayPutElement(safearray *SafeArray, index int64, element uintptr) (err error) {
  277. err = convertHresultToError(
  278. procSafeArrayPutElement.Call(
  279. uintptr(unsafe.Pointer(safearray)),
  280. uintptr(unsafe.Pointer(&index)),
  281. uintptr(unsafe.Pointer(element))))
  282. return
  283. }
  284. // safeArrayGetRecordInfo accesses IRecordInfo info for custom types.
  285. //
  286. // AKA: SafeArrayGetRecordInfo in Windows API.
  287. //
  288. // XXX: Must implement IRecordInfo interface for this to return.
  289. func safeArrayGetRecordInfo(safearray *SafeArray) (recordInfo interface{}, err error) {
  290. err = convertHresultToError(
  291. procSafeArrayGetRecordInfo.Call(
  292. uintptr(unsafe.Pointer(safearray)),
  293. uintptr(unsafe.Pointer(&recordInfo))))
  294. return
  295. }
  296. // safeArraySetRecordInfo mutates IRecordInfo info for custom types.
  297. //
  298. // AKA: SafeArraySetRecordInfo in Windows API.
  299. //
  300. // XXX: Must implement IRecordInfo interface for this to return.
  301. func safeArraySetRecordInfo(safearray *SafeArray, recordInfo interface{}) (err error) {
  302. err = convertHresultToError(
  303. procSafeArraySetRecordInfo.Call(
  304. uintptr(unsafe.Pointer(safearray)),
  305. uintptr(unsafe.Pointer(&recordInfo))))
  306. return
  307. }