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; }