sqlite3_load_extension.go 918 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. // Copyright (C) 2014 Yasuhiro Matsumoto <mattn.jp@gmail.com>.
  2. //
  3. // Use of this source code is governed by an MIT-style
  4. // license that can be found in the LICENSE file.
  5. // +build !sqlite_omit_load_extension
  6. package sqlite3
  7. /*
  8. #include <sqlite3-binding.h>
  9. #include <stdlib.h>
  10. */
  11. import "C"
  12. import (
  13. "errors"
  14. "unsafe"
  15. )
  16. func (c *SQLiteConn) loadExtensions(extensions []string) error {
  17. rv := C.sqlite3_enable_load_extension(c.db, 1)
  18. if rv != C.SQLITE_OK {
  19. return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
  20. }
  21. for _, extension := range extensions {
  22. cext := C.CString(extension)
  23. defer C.free(unsafe.Pointer(cext))
  24. rv = C.sqlite3_load_extension(c.db, cext, nil, nil)
  25. if rv != C.SQLITE_OK {
  26. return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
  27. }
  28. }
  29. rv = C.sqlite3_enable_load_extension(c.db, 0)
  30. if rv != C.SQLITE_OK {
  31. return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
  32. }
  33. return nil
  34. }