static int
ora_instr_mb(text *txt, text *pattern, int start, int nth)
{
int c_len_txt, c_len_pat;
int b_len_pat;
int *pos_txt;
const char *str_txt, *str_pat;
int beg, end, i, dx;
str_txt = VARDATA_ANY(txt);
c_len_txt = ora_mb_strlen(txt, NULL, &pos_txt);
str_pat = VARDATA_ANY(pattern);
b_len_pat = VARSIZE_ANY_EXHDR(pattern);
c_len_pat = pg_mbstrlen_with_len(str_pat, b_len_pat);
if (start > 0)
{
dx = 1;
beg = start - 1;
end = c_len_txt - c_len_pat + 1;
if (beg >= end)
return 0; /* out of range */
}
else
{
dx = -1;
beg = Min(c_len_txt + start, c_len_txt - c_len_pat);
end = -1;
if (beg <= end)
return 0; /* out of range */
}
for (i = beg; i != end; i += dx)
{
if (memcmp(str_txt + pos_txt[i], str_pat, b_len_pat) == 0)
{
if (--nth == 0)
return i + 1;
}
}
return 0;
}
Datum
plvstr_rvrs(PG_FUNCTION_ARGS)
{
text *str;
int start;
int end;
int len;
int i;
int new_len;
text *result;
char *data;
char *sizes = NULL;
int *positions = NULL;
bool mb_encode;
if (PG_ARGISNULL(0))
PG_RETURN_NULL();
str = PG_GETARG_TEXT_PP(0);
mb_encode = pg_database_encoding_max_length() > 1;
if (!mb_encode)
len = VARSIZE_ANY_EXHDR(str);
else
len = ora_mb_strlen(str, &sizes, &positions);
start = PG_ARGISNULL(1) ? 1 : PG_GETARG_INT32(1);
end = PG_ARGISNULL(2) ? (start < 0 ? -len : len) : PG_GETARG_INT32(2);
if ((start > end && start > 0) || (start < end && start < 0))
PARAMETER_ERROR("Second parameter is bigger than third.");
if (start < 0)
{
int new_start, new_end;
new_start = len + start + 1;
new_end = len + end + 1;
start = new_end;
end = new_start;
}
start = start != 0 ? start : 1;
end = end < len ? end : len;
new_len = end - start + 1;
new_len = new_len >= 0 ? new_len : 0;
if (mb_encode)
{
int max_size;
int cur_size;
char *p;
int j;
int fz_size;
fz_size = VARSIZE_ANY_EXHDR(str);
if ((max_size = (new_len*pg_database_encoding_max_length())) > fz_size)
result = palloc(fz_size + VARHDRSZ);
else
result = palloc(max_size + VARHDRSZ);
data = (char*) VARDATA(result);
cur_size = 0;
p = VARDATA_ANY(str);
for (i = end - 1; i>= start - 1; i--)
{
for (j=0; j<sizes[i]; j++)
*data++ = *(p+positions[i]+j);
cur_size += sizes[i];
}
SET_VARSIZE(result, cur_size + VARHDRSZ);
}
else
{
char *p = VARDATA_ANY(str);
result = palloc(new_len + VARHDRSZ);
data = (char*) VARDATA(result);
SET_VARSIZE(result, new_len + VARHDRSZ);
for (i = end - 1; i >= start - 1; i--)
*data++ = p[i];
}
PG_RETURN_TEXT_P(result);
}
static text*
plvsubst_string(text *template_in, ArrayType *vals_in, text *c_subst, FunctionCallInfo fcinfo)
{
ArrayType *v = vals_in;
int nitems,
*dims,
ndims;
char *p;
int16 typlen;
bool typbyval;
char typalign;
char typdelim;
Oid typelem;
Oid typiofunc;
FmgrInfo proc;
int i = 0, items = 0;
StringInfo sinfo;
const char *template_str;
int template_len;
char *sizes;
int *positions;
int subst_mb_len;
int subst_len;
const bits8 *bitmap;
int bitmask;
if (v != NULL && (ndims = ARR_NDIM(v)) > 0)
{
if (ndims != 1)
ereport(ERROR,
(errcode(ERRCODE_INVALID_PARAMETER_VALUE),
errmsg("invalid parameter"),
errdetail("Array of arguments has wrong dimension: %d", ndims)));
p = ARR_DATA_PTR(v);
dims = ARR_DIMS(v);
nitems = ArrayGetNItems(ndims, dims);
bitmap = ARR_NULLBITMAP(v);
get_type_io_data(ARR_ELEMTYPE(v), IOFunc_output,
&typlen, &typbyval,
&typalign, &typdelim,
&typelem, &typiofunc);
fmgr_info_cxt(typiofunc, &proc, fcinfo->flinfo->fn_mcxt);
}
else
{
nitems = 0;
p = NULL;
bitmap = NULL;
}
template_str = VARDATA(template_in);
template_len = ora_mb_strlen(template_in, &sizes, &positions);
subst_mb_len = ora_mb_strlen1(c_subst);
subst_len = VARSIZE_ANY_EXHDR(c_subst);
sinfo = makeStringInfo();
bitmask = 1;
for (i = 0; i < template_len; i++)
{
if (strncmp(&template_str[positions[i]], VARDATA(c_subst), subst_len) == 0)
{
Datum itemvalue;
char *value;
if (items++ < nitems)
{
if (bitmap && (*bitmap & bitmask) == 0)
value = pstrdup("NULL");
else
{
itemvalue = fetch_att(p, typbyval, typlen);
value = DatumGetCString(FunctionCall3(&proc,
itemvalue,
ObjectIdGetDatum(typelem),
Int32GetDatum(-1)));
p = att_addlength_pointer(p, typlen, p);
p = (char *) att_align_nominal(p, typalign);
}
appendStringInfoString(sinfo, value);
pfree(value);
if (bitmap)
{
bitmask <<= 1;
if (bitmask == 0x100)
{
bitmap++;
bitmask = 1;
}
}
}
else
ereport(ERROR,
(errcode(ERRCODE_SYNTAX_ERROR),
errmsg("too few parameters specified for template string")));
i += subst_mb_len - 1;
}
else
appendBinaryStringInfo(sinfo, &template_str[positions[i]], sizes[i]);
}
return cstring_to_text(sinfo->data);
}
