CGI程序与diameter的交互

考虑到网关（CGI程序）和diameter客户端（NAS）可能不在一台机器上，还是使用了socket 编程，不是使用的是本地socket，一方面方便，一方面容易改成远程的。

 

客户端接口

VerifyAdaptor_Result_T Verify_UsrInfo_WithPw(const char* InUsrName, const char* InUsrPw);

 

服务器接口

VerifyAdaptor_Result_T Initialize_VerifyAdaptor();
 void Destroy_VerifyAdaptor();

 VerifyAdaptor_Result_T Retrieve_UsrInfo_WithPw(char* OutUsrName, unsigned int NameLenCap, char* OUtUsrPw, unsigned int PwLenCap, int* OutSocketID);
 VerifyAdaptor_Result_T NotifyVerifyResult(VerifyAdaptor_Result_T InResult, int OutPutSocketID);

 

实现源代码

#include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <sys/types.h> #include <sys/socket.h> #include <sys/un.h> #include <sys/time.h> #include <unistd.h> #include "verify_adaptor.h" #define SOCK_PATH "/tmp/socket" enum { COMMAND_VERIFY_USR_PW, COMMAND_VERIFY_RESULT }; VerifyAdaptor_Result_T Verify_UsrInfo_WithPw(const char* InUsrName, const char* InUsrPw) { int s, t, len; struct sockaddr_un remote; int command, userNameLen, userPwLen; if ((s = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) { return VERIFY_FAIL; } remote.sun_family = AF_UNIX; strcpy(remote.sun_path, SOCK_PATH); len = strlen(remote.sun_path) + sizeof(remote.sun_family); if (connect(s, (struct sockaddr *)&remote, len) == -1) { goto err; } //send user info command = COMMAND_VERIFY_USR_PW; send(s, &command, sizeof(command), 0); userNameLen = strlen(InUsrName)+1; send(s, &userNameLen, sizeof(userNameLen), 0); send(s, InUsrName, userNameLen, 0); userPwLen = strlen(InUsrPw) + 1; send(s, &userPwLen, sizeof(userPwLen),0); send(s, InUsrPw, userPwLen, 0); //wait for the verify result; { fd_set rfds; struct timeval tv; int retval; /* Watch stdin (fd 0) to see when it has input. */ FD_ZERO(&rfds); FD_SET(s, &rfds); /* Wait up to five seconds. */ tv.tv_sec = 5; tv.tv_usec = 0; retval = select(s+1, &rfds, NULL, NULL, &tv); /* Don't rely on the value of tv now! */ if (retval) { /* FD_ISSET(0, &rfds) will be true. */ //read the data recv(s,&command, sizeof(command), 0); if (command == COMMAND_VERIFY_RESULT) { int result = 0; recv(s, &result, sizeof(result), 0); if (result == VERIFY_PASS) goto success; } } } err: close(s); return VERIFY_FAIL; success: close(s); return VERIFY_PASS; } static int ServerSocketFileID = 0; VerifyAdaptor_Result_T Initialize_VerifyAdaptor() { struct sockaddr_un local; char str[100]; int len; if ((ServerSocketFileID = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) { return VERIFY_FAIL; } local.sun_family = AF_UNIX; strcpy(local.sun_path, SOCK_PATH); unlink(local.sun_path); len = strlen(local.sun_path) + sizeof(local.sun_family); if (bind(ServerSocketFileID, (struct sockaddr *)&local, len) == -1) { close(ServerSocketFileID); ServerSocketFileID = 0; return VERIFY_FAIL; } chmod(SOCK_PATH, S_IRWXU|S_IRWXG|S_IRWXO); if (listen(ServerSocketFileID, 5) == -1) { close(ServerSocketFileID); ServerSocketFileID = 0; return VERIFY_FAIL; } return VERIFY_PASS; } void Destroy_VerifyAdaptor() { if (ServerSocketFileID > 0) close(ServerSocketFileID); ServerSocketFileID = 0; unlink(SOCK_PATH); } VerifyAdaptor_Result_T Retrieve_UsrInfo_WithPw(char* OutUsrName, unsigned int NameLenCap, char* OUtUsrPw, unsigned int PwLenCap, int* OutSocketID) { struct sockaddr_un remote; int t,s,command, n; if (ServerSocketFileID <= 0) return VERIFY_FAIL; t = sizeof(remote); if ((s = accept(ServerSocketFileID, (struct sockaddr *)&remote, &t)) == -1) { return VERIFY_FAIL; } n = recv(s, &command, sizeof(command), 0); if (n != sizeof(command)) goto err; if (command == COMMAND_VERIFY_USR_PW) { int usrNamelen, usrPwLen; n = recv(s, &usrNamelen, sizeof(usrNamelen), 0); if (n != sizeof(usrNamelen)) goto err; if (usrNamelen > NameLenCap) goto err; n = recv(s, OutUsrName, usrNamelen, 0); if (n != usrNamelen) goto err; n = recv(s, &usrPwLen, sizeof(usrPwLen), 0); if (n != sizeof(usrPwLen) goto err; if (usrPwLen > PwLenCap) goto err; n = recv(s, OUtUsrPw, usrPwLen, 0); if (n != usrPwLen) goto err; *OutSocketID = s; return VERIFY_PASS; } err: close(s); return VERIFY_FAIL; } VerifyAdaptor_Result_T NotifyVerifyResult(VerifyAdaptor_Result_T InResult, int OutPutSocketID) { int command,result,n; if (ServerSocketFileID <= 0 || OutPutSocketID <= 0) return VERIFY_FAIL; command = COMMAND_VERIFY_RESULT; n = send(OutPutSocketID, &command, sizeof(command), 0); if (n != sizeof(command)) goto err; result = InResult; n = send(OutPutSocketID, &result, sizeof(result), 0); if (n != sizeof(result)) goto err; close(OutPutSocketID); return VERIFY_PASS; err: close(OutPutSocketID); return VERIFY_FAIL; }