static int dcca_cb( struct msg ** msg, struct avp * avp, struct session * sess, void * opaque, enum disp_action * act) { struct msg * m; struct avp_hdr * val_app_id = NULL, * val_rt = NULL, * val_rn = NULL, * val_io = NULL, * val_oo = NULL, *val_imsi = NULL; struct avp *groupedavp = NULL, *groupedavp2 = NULL; union avp_value validity_time, total_octets, result_code, rating_group; struct avp * a = NULL, * aa = NULL; int input_octets = 0, output_octets = 0; char* imsi = NULL; LOG_N("CCR processor"); TRACE_ENTRY("%p %p %p %p", msg, avp, sess, act); if (msg == NULL) return EINVAL; // // Read AVPs from request // // CHECK_FCT( dcca_read_avps_from_message(*msg) ); // Read AVPs CHECK_FCT( dcca_cb_read(msg, dcca_dict.Auth_Application_Id, &val_app_id) ); CHECK_FCT( dcca_cb_read(msg, dcca_dict.CC_Request_Type, &val_rt) ); CHECK_FCT( dcca_cb_read(msg, dcca_dict.CC_Request_Number, &val_rn) ); // Read IMSI CHECK_FCT( get_imsi(*msg, &a) ); if (a) { CHECK_FCT( fd_msg_avp_hdr( a, &val_imsi ) ); int len = val_imsi->avp_value->os.len; imsi = malloc(len * sizeof(char)); int i; for (i = 0; i < len; i++) { imsi[i] = val_imsi->avp_value->os.data[i]; } imsi[len] = 0; } // Read Input / Output Octets CHECK_FCT( fd_msg_search_avp ( *msg, dcca_dict.Multiple_Services_Credit_Control, &a) ); if (a) { CHECK_FCT( avp_search_child ( a, dcca_dict.Used_Service_Unit, &a) ); if (a) { CHECK_FCT( avp_search_child ( a, dcca_dict.CC_Input_Octets, &aa) ); if (aa) { CHECK_FCT( fd_msg_avp_hdr( aa, &val_io ) ); input_octets = val_io->avp_value->i32; } CHECK_FCT( avp_search_child ( a, dcca_dict.CC_Output_Octets, &aa) ); if (aa) { CHECK_FCT( fd_msg_avp_hdr( aa, &val_oo ) ); output_octets = val_oo->avp_value->i32; } } } LOG_N("IMSI: %s", imsi); LOG_N("IN: %i", input_octets); LOG_N("OUT: %i", output_octets); // Create response message CHECK_FCT( fd_msg_new_answer_from_req ( fd_g_config->cnf_dict, msg, 0 ) ); m = *msg; CHECK_FCT( fd_msg_rescode_set( m, "DIAMETER_SUCCESS", NULL, NULL, 1 ) ); // Put params to response CHECK_FCT( dcca_cb_put(m, dcca_dict.Auth_Application_Id, val_app_id) ); CHECK_FCT( dcca_cb_put(m, dcca_dict.CC_Request_Type, val_rt) ); CHECK_FCT( dcca_cb_put(m, dcca_dict.CC_Request_Number, val_rn) ); // Multiple Services CC Group total_octets.i64 = rest_api_vsca(imsi, input_octets, output_octets); CHECK_FCT( fd_msg_avp_new ( dcca_dict.Multiple_Services_Credit_Control, 0, &groupedavp ) ); // Granted Service Unit CHECK_FCT( fd_msg_avp_new ( dcca_dict.Granted_Service_Unit, 0, &groupedavp2 ) ); CHECK_FCT( dcca_cb_put_value_to_avp(groupedavp2, dcca_dict.CC_Total_Octets, &total_octets) ); CHECK_FCT( fd_msg_avp_add( groupedavp, MSG_BRW_LAST_CHILD, groupedavp2 ) ); // Result code result_code.i32 = 2001; CHECK_FCT( dcca_cb_put_value_to_avp(groupedavp, dcca_dict.Result_Code, &result_code) ); // Rating group rating_group.i32 = 1001; CHECK_FCT( dcca_cb_put_value_to_avp(groupedavp, dcca_dict.Rating_Group, &rating_group) ); // Validity Time validity_time.i32 = 60; CHECK_FCT( dcca_cb_put_value_to_avp(groupedavp, dcca_dict.Validity_Time, &validity_time) ); CHECK_FCT( fd_msg_avp_add( m, MSG_BRW_LAST_CHILD, groupedavp ) ); // Send the answer *act = DISP_ACT_SEND; return 0; }
/* Callback for incoming Test-Request messages */ static int ta_tr_cb( struct msg ** msg, struct avp * avp, struct session * sess, void * opaque, enum disp_action * act) { struct msg *ans, *req; struct avp * src = NULL; struct avp_hdr * hdr = NULL; UsageServerSession* mi; TRACE_ENTRY("%p %p %p %p", msg, avp, sess, act); if (msg == NULL) return EINVAL; /* Value of Origin-Host */ fprintf(stderr, "Request received from "); CHECK_FCT( fd_msg_search_avp ( *msg, ta_origin_host, &src) ); if (src) { CHECK_FCT( fd_msg_avp_hdr( src, &hdr ) ); fprintf(stderr, "'%.*s'", (int)hdr->avp_value->os.len, hdr->avp_value->os.data); } else { fprintf(stderr, "no_Origin-Host"); } fprintf(stderr, ", replying...\n"); //get or create the session/session state req = *msg; /* Create answer header */ CHECK_FCT( fd_msg_new_answer_from_req ( fd_g_config->cnf_dict, msg, 0 ) ); ans = *msg; d_req_type reqType; //get the op type CHECK_FCT( fd_msg_search_avp ( req, ta_avp_optype, &src) ); CHECK_FCT( fd_msg_avp_hdr( src, &hdr ) ); reqType = hdr->avp_value->i32; if (reqType==d_start){ //create the session state mi= usageserver_session_alloc(); mi->leftQuota = DUMMY_INIT_QUOTA; }else{ //get the session state fd_sess_state_retrieve(ta_cli_reg, sess, &mi); } if (reqType!=d_start){//for update, stop, need to update the leftQuota //update the left quota in session state, according to the used quota avp in req CHECK_FCT( fd_msg_search_avp ( req, ta_avp_usedQuota, &src) ); CHECK_FCT( fd_msg_avp_hdr( src, &hdr ) ); uint64_t usedQuota = hdr->avp_value->u64; if (mi->leftQuota>=usedQuota){ mi->leftQuota-=usedQuota; }else{ fprintf(stderr, "fatal, the used should not be larger then the granted last time."); mi->leftQuota=0; } } if (reqType!=d_stop){//for start, update, need to reply with grantedQuota //set the granted quota AVP according to requested quota and left quota in session state CHECK_FCT( fd_msg_search_avp ( req, ta_avp_requestQuota, &src) ); CHECK_FCT( fd_msg_avp_hdr( src, &hdr ) ); uint64_t reqAmt = hdr->avp_value->u64; uint64_t grantAmt=0; if (mi->leftQuota>=reqAmt){ grantAmt = reqAmt; }else{ grantAmt = mi->leftQuota; } hdr->avp_value->u64 = grantAmt; CHECK_FCT( fd_msg_avp_new ( ta_avp_grantedQuota, 0, &avp ) ); CHECK_FCT( fd_msg_avp_setvalue( avp, hdr->avp_value ) ); CHECK_FCT( fd_msg_avp_add( ans, MSG_BRW_LAST_CHILD, avp ) ); fprintf(stderr, "add granted quota avp, %llu.\n", grantAmt); } fprintf(stderr, "session:leftQuota:%llu\n", mi->leftQuota); /* Set the Origin-Host, Origin-Realm, Result-Code AVPs */ CHECK_FCT( fd_msg_rescode_set( ans, "DIAMETER_SUCCESS", NULL, NULL, 1 ) ); //save the session state fd_sess_state_store(ta_cli_reg, sess, &mi); /* Send the answer */ CHECK_FCT( fd_msg_send( msg, NULL, NULL ) ); return 0; }
/* Callback for incoming Base Accounting Accounting-Request messages */ static int acct_cb( struct msg ** msg, struct avp * avp, struct session * sess, void * opaque, enum disp_action * act) { struct msg * m; struct avp * a = NULL; struct avp_hdr * art=NULL, *arn=NULL; /* We keep a pointer on the Accounting-Record-{Type, Number} AVPs from the query */ struct acct_record_list rl; TRACE_ENTRY("%p %p %p %p", msg, avp, sess, act); if (msg == NULL) return EINVAL; m = *msg; /* Prepare a new record list */ CHECK_FCT( acct_rec_prepare( &rl ) ); /* Maps the AVPs from the query with this record list */ CHECK_FCT( acct_rec_map( &rl, m ) ); /* Check that at least one AVP was mapped */ CHECK_FCT( acct_rec_validate( &rl ) ); /* Now, save these mapped AVPs in the database */ CHECK_FCT( acct_db_insert( &rl ) ); acct_rec_empty( &rl ); /* OK, we can send a positive reply now */ /* Get Accounting-Record-{Number,Type} values */ CHECK_FCT( fd_msg_search_avp ( m, acct_dict.Accounting_Record_Type, &a) ); if (a) { CHECK_FCT( fd_msg_avp_hdr( a, &art ) ); } CHECK_FCT( fd_msg_search_avp ( m, acct_dict.Accounting_Record_Number, &a) ); if (a) { CHECK_FCT( fd_msg_avp_hdr( a, &arn ) ); } /* Create the answer message */ CHECK_FCT( fd_msg_new_answer_from_req ( fd_g_config->cnf_dict, msg, 0 ) ); m = *msg; /* Set the Origin-Host, Origin-Realm, Result-Code AVPs */ CHECK_FCT( fd_msg_rescode_set( m, "DIAMETER_SUCCESS", NULL, NULL, 1 ) ); /* Add the mandatory AVPs in the ACA */ if (art) { CHECK_FCT( fd_msg_avp_new ( acct_dict.Accounting_Record_Type, 0, &a ) ); CHECK_FCT( fd_msg_avp_setvalue( a, art->avp_value ) ); CHECK_FCT( fd_msg_avp_add( m, MSG_BRW_LAST_CHILD, a ) ); } if (arn) { CHECK_FCT( fd_msg_avp_new ( acct_dict.Accounting_Record_Number, 0, &a ) ); CHECK_FCT( fd_msg_avp_setvalue( a, arn->avp_value ) ); CHECK_FCT( fd_msg_avp_add( m, MSG_BRW_LAST_CHILD, a ) ); } /* Send the answer */ *act = DISP_ACT_SEND; return 0; }