fis-gtm/sr_port/op_setzextract.c

112 lines
3.4 KiB
C

/****************************************************************
* *
* Copyright 2006, 2012 Fidelity Information Services, Inc *
* *
* This source code contains the intellectual property *
* of its copyright holder(s), and is made available *
* under a license. If you do not know the terms of *
* the license, please stop and do not read further. *
* *
****************************************************************/
#include "mdef.h"
#include "gtm_string.h"
#include "stringpool.h"
#include "min_max.h"
#include "op.h"
GBLREF spdesc stringpool;
error_def(ERR_MAXSTRLEN);
/*
* ----------------------------------------------------------
* Set version of $extract
*
* Arguments:
* src - source mval
* expr - expression string mval to be inserted into source
* schar - starting character index to be replaced
* echar - ending character index to be replaced
* dst - destination mval where the result is saved.
*
* Return:
* none
* ----------------------------------------------------------
*/
void op_setzextract(mval *src, mval *expr, int schar, int echar, mval *dst)
{
size_t strlen, padlen;
int pfxlen, sfxoff, sfxlen, srclen;
unsigned char *srcptr, *pfx, *straddr;
padlen = pfxlen = sfxlen = 0;
MV_FORCE_STR(expr); /* Expression to put into piece place */
if (MV_DEFINED(src))
{
MV_FORCE_STR(src); /* Make sure is string prior to length check */
srclen = src->str.len;
} else
/* Source is not defined -- treat as a null string */
srclen = 0;
schar = MAX(schar, 1); /* schar starts at 1 at a minimum */
/* There are four cases in the spec:
*
* 1) schar > echar or echar < 1 -- glvn and naked indicator are not changed. This is
* handled by generated code in m_set
* 2) echar '< schar-1 > srclen -- dst = src_$J("",schar-1-srclen)_expr
* 3) schar-1 '> srclen < echar -- dst = $E(src,1,schar-1)_expr
* 4) All others -- dst = $E(src,1,schar-1)_expr_$E(src,echar+1,srclen)
*/
if ((schar - 1) > srclen)
{ /* Case #2 -- echar test handled in generated code */
pfxlen = srclen;
padlen = (size_t)schar - 1 - (size_t)srclen;
/* Note, no suffix, just expr after the padlen */
} else
{ /* (schar-1) <= srclen) (Case #3 and common part of Default case) */
pfxlen = schar - 1;
/* Test if truly default case */
if (srclen >= echar) /* If (srclen < echar) then was case 3 and we are done. else.. */
{ /* Default case */
sfxoff = echar;
sfxlen = srclen - echar;
}
}
/* Calculate total string len. delim_cnt has needed padding delimiters for null fields */
strlen = (size_t)pfxlen + padlen + (size_t)expr->str.len + (size_t)sfxlen;
if (MAX_STRLEN < strlen)
rts_error(VARLSTCNT(1) ERR_MAXSTRLEN);
ENSURE_STP_FREE_SPACE((int)strlen);
pfx = (unsigned char *)src->str.addr;
straddr = stringpool.free;
/* copy prefix */
if (0 < pfxlen)
{
memcpy(straddr, pfx, pfxlen);
straddr += pfxlen;
}
/* insert padding */
while (0 < padlen--)
*straddr++ = ' ';
/* copy expression */
if (0 < expr->str.len)
{
memcpy(straddr, expr->str.addr, expr->str.len);
straddr += expr->str.len;
}
/* copy suffix */
if (0 < sfxlen)
{
memcpy(straddr, pfx + sfxoff, sfxlen);
straddr += sfxlen;
}
assert((straddr - stringpool.free) == strlen);
dst->mvtype = MV_STR;
dst->str.len = INTCAST(straddr - stringpool.free);
dst->str.addr = (char *)stringpool.free;
stringpool.free = straddr;
}