int AAAAVPBuildBuffer(AAA_AVP *avp, unsigned char *dest) {
unsigned char* p = dest;
if (!avp || !dest) {
ERROR("trying to build msg buffer from/to NULL avp");
return 0;
}
/* AVP HEADER */
/* avp code */
set_4bytes(p,avp->code);
p +=4;
/* flags */
(*p++) = (unsigned char)avp->flags;
/* avp length */
set_3bytes(p, (AVP_HDR_SIZE(avp->flags)+avp->data.len) );
p += 3;
/* vendor id */
if ((avp->flags&0x80)!=0) {
set_4bytes(p,avp->vendorId);
p +=4;
}
/* data */
memcpy( p, avp->data.s, avp->data.len);
p += to_32x_len( avp->data.len );
return p-dest;
}
/**
* Groups a list of avps into a data buffer
* @param avps
*/
str AAAGroupAVPS(AAA_AVP_LIST avps)
{
AAA_AVP *avp;
unsigned char *p;
str buf={0,0};
/* count and add the avps */
for(avp=avps.head;avp;avp=avp->next) {
buf.len += AVP_HDR_SIZE(avp->flags)+ to_32x_len( avp->data.len );
}
if (!buf.len) return buf;
/* allocate some memory */
buf.s = (char*)shm_malloc( buf.len );
if (!buf.s) {
LM_ERR("hss3g_group_avps: no more free memory!\n");
buf.len=0;
return buf;
}
memset(buf.s, 0, buf.len);
/* fill in the buffer */
p = (unsigned char*) buf.s;
for(avp=avps.head;avp;avp=avp->next) {
/* AVP HEADER */
/* avp code */
set_4bytes(p,avp->code);
p +=4;
/* flags */
(*p++) = (unsigned char)avp->flags;
/* avp length */
set_3bytes(p, (AVP_HDR_SIZE(avp->flags)+avp->data.len) );
p += 3;
/* vendor id */
if ((avp->flags&0x80)!=0) {
set_4bytes(p,avp->vendorId);
p +=4;
}
/* data */
memcpy( p, avp->data.s, avp->data.len);
p += to_32x_len( avp->data.len );
}
if ((char*)p-buf.s!=buf.len) {
LM_ERR("BUG:hss3g_group_avps: mismatch between len and buf!\n");
shm_free( buf.s );
buf.s = 0;
buf.len = 0;
return buf;
}
return buf;
}
/* from a AAAMessage structure, a buffer to be send is build
*/
AAAReturnCode AAABuildMsgBuffer( AAAMessage *msg )
{
unsigned char *p;
AAA_AVP *avp;
AAA_AVP *mem; // group member
/* first let's compute the length of the buffer */
msg->buf.len = AAA_MSG_HDR_SIZE; /* AAA message header size */
/* count and add the avps */
for(avp=msg->avpList.head;avp;avp=avp->next) {
msg->buf.len += AVP_HDR_SIZE(avp->flags)+ to_32x_len( avp->data.len );
}
// DBG("xxxx len=%d\n",msg->buf.len);
/* allocate some memory */
msg->buf.s = (char*)ad_malloc( msg->buf.len );
if (!msg->buf.s) {
ERROR("ERROR:AAABuildMsgBuffer: no more free memory!\n");
goto error;
}
memset(msg->buf.s, 0, msg->buf.len);
/* fill in the buffer */
p = (unsigned char*)msg->buf.s;
/* DIAMETER HEADER */
/* message length */
((unsigned int*)p)[0] =htonl(msg->buf.len);
/* Diameter Version */
*p = 1;
p += VER_SIZE + MESSAGE_LENGTH_SIZE;
/* command code */
((unsigned int*)p)[0] = htonl(msg->commandCode);
/* flags */
*p = (unsigned char)msg->flags;
p += FLAGS_SIZE + COMMAND_CODE_SIZE;
/* application-ID */
((unsigned int*)p)[0] = htonl(msg->applicationId);
p += APPLICATION_ID_SIZE;
/* hop by hop id */
((unsigned int*)p)[0] = msg->hopbyhopId;
p += HOP_BY_HOP_IDENTIFIER_SIZE;
/* end to end id */
((unsigned int*)p)[0] = msg->endtoendId;
p += END_TO_END_IDENTIFIER_SIZE;
/* AVPS */
for(avp=msg->avpList.head;avp;avp=avp->next) {
/* AVP HEADER */
/* avp code */
set_4bytes(p,avp->code);
p +=4;
/* flags */
(*p++) = (unsigned char)avp->flags;
/* avp length */
set_3bytes(p, (AVP_HDR_SIZE(avp->flags)+avp->data.len) );
p += 3;
/* vendor id */
if ((avp->flags&0x80)!=0) {
set_4bytes(p,avp->vendorId);
p +=4;
}
/* data */
if (!avp->groupedHead) {
memcpy( p, avp->data.s, avp->data.len);
p += to_32x_len( avp->data.len );
} else {
// group members
for(mem=avp->groupedHead;mem;mem=mem->next)
p+=AAAAVPBuildBuffer(mem, p);
}
}
if ((char*)p-msg->buf.s!=msg->buf.len) {
ERROR("BUG: build_buf_from_msg: mismatch between len and buf!\n");
ad_free( msg->buf.s );
msg->buf.s = 0;
msg->buf.len = 0;
goto error;
}
// DBG("Message: %.*s\n", msg->buf.len, msg->buf.s);
return AAA_ERR_SUCCESS;
error:
return -1;
}
