433 lines
17 KiB
C
433 lines
17 KiB
C
//-----------------------------------------------------------------------------
|
|
// StringVar.c
|
|
// Defines the routines specific to the string type.
|
|
//-----------------------------------------------------------------------------
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// String type
|
|
//-----------------------------------------------------------------------------
|
|
typedef struct {
|
|
Variable_HEAD
|
|
char *data;
|
|
} udt_StringVar;
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Declaration of string variable functions.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringVar_Initialize(udt_StringVar*, udt_Cursor*);
|
|
static int StringVar_PostDefine(udt_StringVar*);
|
|
static int StringVar_SetValue(udt_StringVar*, unsigned, PyObject*);
|
|
static PyObject *StringVar_GetValue(udt_StringVar*, unsigned);
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Python type declarations
|
|
//-----------------------------------------------------------------------------
|
|
static PyTypeObject g_StringVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.STRING", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
static PyTypeObject g_UnicodeVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.UNICODE", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
static PyTypeObject g_FixedCharVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.FIXED_CHAR", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
static PyTypeObject g_FixedUnicodeVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.FIXED_UNICODE", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
static PyTypeObject g_RowidVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.ROWID", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
static PyTypeObject g_BinaryVarType = {
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, // ob_size
|
|
"cx_Oracle.BINARY", // tp_name
|
|
sizeof(udt_StringVar), // tp_basicsize
|
|
0, // tp_itemsize
|
|
(destructor) Variable_Free, // tp_dealloc
|
|
0, // tp_print
|
|
0, // tp_getattr
|
|
0, // tp_setattr
|
|
0, // tp_compare
|
|
(reprfunc) Variable_Repr, // tp_repr
|
|
0, // tp_as_number
|
|
0, // tp_as_sequence
|
|
0, // tp_as_mapping
|
|
0, // tp_hash
|
|
0, // tp_call
|
|
0, // tp_str
|
|
(getattrofunc) Variable_GetAttr, // tp_getattro
|
|
0, // tp_setattro
|
|
0, // tp_as_buffer
|
|
Py_TPFLAGS_DEFAULT, // tp_flags
|
|
0 // tp_doc
|
|
};
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// variable type declarations
|
|
//-----------------------------------------------------------------------------
|
|
static udt_VariableType vt_String = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) NULL,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_StringVarType, // Python type
|
|
SQLT_CHR, // Oracle type
|
|
SQLCS_IMPLICIT, // charset form
|
|
MAX_STRING_CHARS, // element length (default)
|
|
1, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
static udt_VariableType vt_NationalCharString = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) StringVar_PostDefine,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_UnicodeVarType, // Python type
|
|
SQLT_CHR, // Oracle type
|
|
SQLCS_NCHAR, // charset form
|
|
MAX_STRING_CHARS, // element length (default)
|
|
1, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
static udt_VariableType vt_FixedChar = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) NULL,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_FixedCharVarType, // Python type
|
|
SQLT_AFC, // Oracle type
|
|
SQLCS_IMPLICIT, // charset form
|
|
2000, // element length (default)
|
|
1, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
static udt_VariableType vt_FixedNationalChar = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) StringVar_PostDefine,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_FixedUnicodeVarType, // Python type
|
|
SQLT_AFC, // Oracle type
|
|
SQLCS_NCHAR, // charset form
|
|
2000, // element length (default)
|
|
1, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
static udt_VariableType vt_Rowid = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) NULL,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_RowidVarType, // Python type
|
|
SQLT_CHR, // Oracle type
|
|
SQLCS_IMPLICIT, // charset form
|
|
18, // element length (default)
|
|
0, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
static udt_VariableType vt_Binary = {
|
|
(InitializeProc) StringVar_Initialize,
|
|
(FinalizeProc) NULL,
|
|
(PreDefineProc) NULL,
|
|
(PostDefineProc) NULL,
|
|
(IsNullProc) NULL,
|
|
(SetValueProc) StringVar_SetValue,
|
|
(GetValueProc) StringVar_GetValue,
|
|
&g_BinaryVarType, // Python type
|
|
SQLT_BIN, // Oracle type
|
|
SQLCS_IMPLICIT, // charset form
|
|
MAX_STRING_CHARS, // element length (default)
|
|
1, // is variable length
|
|
1, // can be copied
|
|
1 // can be in array
|
|
};
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringVar_Initialize()
|
|
// Initialize the variable.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringVar_Initialize(
|
|
udt_StringVar *var, // variable to initialize
|
|
udt_Cursor *cursor) // cursor to use
|
|
{
|
|
var->actualLength = (ub2*) PyMem_Malloc(var->allocatedElements *
|
|
sizeof(ub2));
|
|
if (!var->actualLength) {
|
|
PyErr_NoMemory();
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringVar_PostDefine()
|
|
// Set the character set information when values are fetched from this
|
|
// variable.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringVar_PostDefine(
|
|
udt_StringVar *var) // variable to initialize
|
|
{
|
|
ub2 charsetId;
|
|
sword status;
|
|
|
|
status = OCIAttrSet(var->defineHandle, OCI_HTYPE_DEFINE,
|
|
&var->type->charsetForm, 0, OCI_ATTR_CHARSET_FORM,
|
|
var->environment->errorHandle);
|
|
if (Environment_CheckForError(var->environment, status,
|
|
"StringVar_PostDefine(): setting charset form") < 0)
|
|
return -1;
|
|
|
|
charsetId = OCI_UTF16ID;
|
|
status = OCIAttrSet(var->defineHandle, OCI_HTYPE_DEFINE, &charsetId, 0,
|
|
OCI_ATTR_CHARSET_ID, var->environment->errorHandle);
|
|
if (Environment_CheckForError(var->environment, status,
|
|
"StringVar_PostDefine(): setting charset id") < 0)
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringVar_SetValue()
|
|
// Set the value of the variable.
|
|
//-----------------------------------------------------------------------------
|
|
static int StringVar_SetValue(
|
|
udt_StringVar *var, // variable to set value for
|
|
unsigned pos, // array position to set
|
|
PyObject *value) // value to set
|
|
{
|
|
PyObject *encodedString;
|
|
Py_ssize_t bufferSize;
|
|
const void *buffer;
|
|
ub2 actualLength;
|
|
|
|
// get the buffer data and size for binding
|
|
encodedString = NULL;
|
|
if (var->type->charsetForm == SQLCS_IMPLICIT) {
|
|
if (PyString_Check(value)) {
|
|
buffer = PyString_AS_STRING(value);
|
|
bufferSize = PyString_GET_SIZE(value);
|
|
} else if (PyBuffer_Check(value)) {
|
|
if (PyObject_AsReadBuffer(value, &buffer, &bufferSize) < 0)
|
|
return -1;
|
|
} else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"expecting string or buffer data");
|
|
return -1;
|
|
}
|
|
actualLength = (ub2) bufferSize;
|
|
} else {
|
|
if (!PyUnicode_Check(value)) {
|
|
PyErr_SetString(PyExc_TypeError, "expecting unicode data");
|
|
return -1;
|
|
}
|
|
#ifdef Py_UNICODE_WIDE
|
|
int one = 1;
|
|
int byteOrder = (IS_LITTLE_ENDIAN) ? -1 : 1;
|
|
encodedString = PyUnicode_EncodeUTF16(PyUnicode_AS_UNICODE(value),
|
|
PyUnicode_GET_SIZE(value), NULL, byteOrder);
|
|
if (!encodedString)
|
|
return -1;
|
|
buffer = PyString_AS_STRING(encodedString);
|
|
bufferSize = PyString_GET_SIZE(encodedString);
|
|
#else
|
|
buffer = PyUnicode_AS_UNICODE(value);
|
|
bufferSize = sizeof(Py_UNICODE) * PyUnicode_GET_SIZE(value);
|
|
#endif
|
|
actualLength = bufferSize / 2;
|
|
}
|
|
|
|
// ensure that the buffer is not too large
|
|
if (bufferSize > var->maxLength) {
|
|
if (bufferSize > var->environment->maxStringBytes) {
|
|
PyErr_SetString(PyExc_ValueError, "string data too large");
|
|
Py_XDECREF(encodedString);
|
|
return -1;
|
|
}
|
|
if (Variable_Resize( (udt_Variable*) var, bufferSize) < 0) {
|
|
Py_XDECREF(encodedString);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// keep a copy of the string
|
|
var->actualLength[pos] = actualLength;
|
|
if (bufferSize)
|
|
memcpy(var->data + var->maxLength * pos, buffer, bufferSize);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// StringVar_GetValue()
|
|
// Returns the value stored at the given array position.
|
|
//-----------------------------------------------------------------------------
|
|
static PyObject *StringVar_GetValue(
|
|
udt_StringVar *var, // variable to determine value for
|
|
unsigned pos) // array position
|
|
{
|
|
char *data;
|
|
|
|
data = var->data + pos * var->maxLength;
|
|
if (var->type->charsetForm == SQLCS_IMPLICIT)
|
|
return PyString_FromStringAndSize(data, var->actualLength[pos]);
|
|
#ifdef Py_UNICODE_WIDE
|
|
ub4 bytes = var->actualLength[pos] * 2;
|
|
return PyUnicode_DecodeUTF16(data, bytes, NULL, NULL);
|
|
#else
|
|
return PyUnicode_FromUnicode((Py_UNICODE*) data, var->actualLength[pos]);
|
|
#endif
|
|
}
|
|
|