0
0
Fork 0
mirror of https://github.com/go-gitea/gitea synced 2024-12-05 05:27:59 +01:00
gitea/vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go

85 lines
1.8 KiB
Go
Raw Normal View History

// Copyright (C) 2019 Yasuhiro Matsumoto <mattn.jp@gmail.com>.
2016-11-03 23:16:01 +01:00
//
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.
2018-10-23 07:28:10 +02:00
2016-11-03 23:16:01 +01:00
// +build !sqlite_omit_load_extension
package sqlite3
/*
#ifndef USE_LIBSQLITE3
#include <sqlite3-binding.h>
#else
#include <sqlite3.h>
#endif
#include <stdlib.h>
*/
import "C"
import (
"errors"
"unsafe"
)
func (c *SQLiteConn) loadExtensions(extensions []string) error {
rv := C.sqlite3_enable_load_extension(c.db, 1)
if rv != C.SQLITE_OK {
return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
}
for _, extension := range extensions {
2020-08-14 03:54:46 +02:00
if err := c.loadExtension(extension, nil); err != nil {
C.sqlite3_enable_load_extension(c.db, 0)
2020-08-14 03:54:46 +02:00
return err
2016-11-03 23:16:01 +01:00
}
}
rv = C.sqlite3_enable_load_extension(c.db, 0)
if rv != C.SQLITE_OK {
return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
}
2020-08-14 03:54:46 +02:00
2016-11-03 23:16:01 +01:00
return nil
}
// LoadExtension load the sqlite3 extension.
2016-11-03 23:16:01 +01:00
func (c *SQLiteConn) LoadExtension(lib string, entry string) error {
rv := C.sqlite3_enable_load_extension(c.db, 1)
if rv != C.SQLITE_OK {
return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
}
2020-08-14 03:54:46 +02:00
if err := c.loadExtension(lib, &entry); err != nil {
C.sqlite3_enable_load_extension(c.db, 0)
return err
}
2016-11-03 23:16:01 +01:00
2020-08-14 03:54:46 +02:00
rv = C.sqlite3_enable_load_extension(c.db, 0)
2016-11-03 23:16:01 +01:00
if rv != C.SQLITE_OK {
return errors.New(C.GoString(C.sqlite3_errmsg(c.db)))
}
2020-08-14 03:54:46 +02:00
return nil
}
func (c *SQLiteConn) loadExtension(lib string, entry *string) error {
clib := C.CString(lib)
defer C.free(unsafe.Pointer(clib))
var centry *C.char
if entry != nil {
centry = C.CString(*entry)
2020-08-14 03:54:46 +02:00
defer C.free(unsafe.Pointer(centry))
}
var errMsg *C.char
defer C.sqlite3_free(unsafe.Pointer(errMsg))
rv := C.sqlite3_load_extension(c.db, clib, centry, &errMsg)
2016-11-03 23:16:01 +01:00
if rv != C.SQLITE_OK {
2020-08-14 03:54:46 +02:00
return errors.New(C.GoString(errMsg))
2016-11-03 23:16:01 +01:00
}
return nil
}