Add per-user limits to chan_sip

git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@1223 65c4cc65-6c06-0410-ace0-fbb531ad65f3
1.0
Mark Spencer 22 years ago
parent 159de43db3
commit ecf50fcee8

@ -184,7 +184,7 @@ build.h:
endif endif
asterisk: .version build.h editline/libedit.a db1-ast/libdb1.a $(OBJS) asterisk: .version build.h editline/libedit.a db1-ast/libdb1.a $(OBJS)
gcc $(DEBUG) -o asterisk -rdynamic $(OBJS) $(LIBS) $(LIBEDIT) db1-ast/libdb1.a $(CC) $(DEBUG) -o asterisk -rdynamic $(OBJS) $(LIBS) $(LIBEDIT) db1-ast/libdb1.a
subdirs: subdirs:
for x in $(SUBDIRS); do $(MAKE) -C $$x || exit 1 ; done for x in $(SUBDIRS); do $(MAKE) -C $$x || exit 1 ; done

@ -35,7 +35,7 @@ CFLAGS+=-fPIC
all: $(APPS) all: $(APPS)
clean: clean:
rm -f *.so *.o look rm -f *.so *.o look .depend
%.so : %.o %.so : %.o
$(CC) -shared -Xlinker -x -o $@ $< $(CC) -shared -Xlinker -x -o $@ $<

@ -268,6 +268,8 @@ struct sip_user {
int insecure; int insecure;
int canreinvite; int canreinvite;
int dtmfmode; int dtmfmode;
int inUse;
int incominglimit;
struct ast_ha *ha; struct ast_ha *ha;
struct sip_user *next; struct sip_user *next;
}; };
@ -882,6 +884,37 @@ static void __sip_destroy(struct sip_pvt *p, int lockowner)
free(p); free(p);
} }
} }
static int find_user(struct sip_pvt *p, int event)
{
char name[256] = "";
struct sip_user *u;
strncpy(name, p->username, sizeof(name) - 1);
ast_pthread_mutex_lock(&userl.lock);
u = userl.users;
while(u) {
if (!strcasecmp(u->name, name)) {
break;
}
u = u->next;
}
if(event == 0) {
u->inUse--;
} else {
if (u->incominglimit > 0 ) {
if (u->inUse >= u->incominglimit) {
ast_log(LOG_ERROR, "Call from user '%s' rejected due to usage limit of %d\n", u->name, u->incominglimit);
ast_pthread_mutex_unlock(&userl.lock);
return -1;
}
}
u->inUse++;
ast_log(LOG_DEBUG, "Call from user '%s' is %d out of %d\n", u->name, u->inUse, u->incominglimit);
}
ast_pthread_mutex_unlock(&userl.lock);
return 0;
}
static void sip_destroy(struct sip_pvt *p) static void sip_destroy(struct sip_pvt *p)
{ {
ast_pthread_mutex_lock(&iflock); ast_pthread_mutex_lock(&iflock);
@ -904,6 +937,7 @@ static int sip_hangup(struct ast_channel *ast)
return 0; return 0;
} }
ast_pthread_mutex_lock(&p->lock); ast_pthread_mutex_lock(&p->lock);
/* find_user(p,0); */
/* Determine how to disconnect */ /* Determine how to disconnect */
if (p->owner != ast) { if (p->owner != ast) {
ast_log(LOG_WARNING, "Huh? We aren't the owner?\n"); ast_log(LOG_WARNING, "Huh? We aren't the owner?\n");
@ -3667,20 +3701,22 @@ static void receive_message(struct sip_pvt *p, struct sip_request *req)
static int sip_show_users(int fd, int argc, char *argv[]) static int sip_show_users(int fd, int argc, char *argv[])
{ {
#define FORMAT "%-15.15s %-15.15s %-15.15s %-15.15s %-5.5s\n" #define FORMAT "%-15.15s %-10.15s %-10.10s %-10.15s %-5.5s %-5.15s\n"
#define FORMAT2 "%-15.15s %-10.15s %-10.10s %-11.15s %-6.5s %d/%d\n"
struct sip_user *user; struct sip_user *user;
if (argc != 3) if (argc != 3)
return RESULT_SHOWUSAGE; return RESULT_SHOWUSAGE;
ast_pthread_mutex_lock(&userl.lock); ast_pthread_mutex_lock(&userl.lock);
ast_cli(fd, FORMAT, "Username", "Secret", "Authen", "Def.Context", "A/C"); ast_cli(fd, FORMAT, "Username", "Secret", "Authen", "Def.Context", "A/C", "inUse/Limit");
for(user=userl.users;user;user=user->next) { for(user=userl.users;user;user=user->next) {
ast_cli(fd, FORMAT, user->name, user->secret, user->methods, ast_cli(fd, FORMAT2, user->name, user->secret, user->methods,
user->context, user->context,user->ha ? "Yes" : "No",
user->ha ? "Yes" : "No"); user->inUse,user->incominglimit);
} }
ast_pthread_mutex_unlock(&userl.lock); ast_pthread_mutex_unlock(&userl.lock);
return RESULT_SUCCESS; return RESULT_SUCCESS;
#undef FORMAT #undef FORMAT
#undef FORMAT2
} }
static int sip_show_peers(int fd, int argc, char *argv[]) static int sip_show_peers(int fd, int argc, char *argv[])
@ -4494,16 +4530,28 @@ static int handle_request(struct sip_pvt *p, struct sip_request *req, struct soc
/* Initialize the context if it hasn't been already */ /* Initialize the context if it hasn't been already */
if (!strlen(p->context)) if (!strlen(p->context))
strncpy(p->context, context, sizeof(p->context) - 1); strncpy(p->context, context, sizeof(p->context) - 1);
/* Check number of concurrent calls -vs- incoming limit HERE */
res = find_user(p,1);
if (res) {
if (res < 0) {
ast_log(LOG_DEBUG, "Failed to place call for user %s, too many calls\n", p->username);
p->needdestroy = 1;
}
return 0;
}
/* Get destination right away */ /* Get destination right away */
gotdest = get_destination(p, NULL); gotdest = get_destination(p, NULL);
get_rdnis(p, NULL); get_rdnis(p, NULL);
build_contact(p); build_contact(p);
if (gotdest) { if (gotdest) {
if (gotdest < 0) if (gotdest < 0) {
transmit_response(p, "404 Not Found", req); transmit_response(p, "404 Not Found", req);
else find_user(p,0);
} else {
transmit_response(p, "484 Address Incomplete", req); transmit_response(p, "484 Address Incomplete", req);
find_user(p,0);
}
p->needdestroy = 1; p->needdestroy = 1;
} else { } else {
/* If no extension was specified, use the s one */ /* If no extension was specified, use the s one */
@ -4627,6 +4675,7 @@ static int handle_request(struct sip_pvt *p, struct sip_request *req, struct soc
transmit_response(p, "200 OK", req); transmit_response(p, "200 OK", req);
transmit_response_reliable(p, "487 Request Terminated", &p->initreq); transmit_response_reliable(p, "487 Request Terminated", &p->initreq);
} else if (!strcasecmp(cmd, "BYE")) { } else if (!strcasecmp(cmd, "BYE")) {
find_user(p,0);
copy_request(&p->initreq, req); copy_request(&p->initreq, req);
check_via(p, req); check_via(p, req);
p->alreadygone = 1; p->alreadygone = 1;
@ -5141,6 +5190,10 @@ static struct sip_user *build_user(char *name, struct ast_variable *v)
if (user) { if (user) {
memset(user, 0, sizeof(struct sip_user)); memset(user, 0, sizeof(struct sip_user));
strncpy(user->name, name, sizeof(user->name)-1); strncpy(user->name, name, sizeof(user->name)-1);
/* set the usage flag to a sane staring value*/
user->inUse = 0;
user->canreinvite = REINVITE_INVITE; user->canreinvite = REINVITE_INVITE;
/* JK02: set default context */ /* JK02: set default context */
strcpy(user->context, context); strcpy(user->context, context);
@ -5181,6 +5234,10 @@ static struct sip_user *build_user(char *name, struct ast_variable *v)
user->pickupgroup = ast_get_group(v->value); user->pickupgroup = ast_get_group(v->value);
} else if (!strcasecmp(v->name, "accountcode")) { } else if (!strcasecmp(v->name, "accountcode")) {
strncpy(user->accountcode, v->value, sizeof(user->accountcode)-1); strncpy(user->accountcode, v->value, sizeof(user->accountcode)-1);
} else if (!strcasecmp(v->name, "incominglimit")) {
user->incominglimit = atoi(v->value);
if (user->incominglimit < 0)
user->incominglimit = 0;
} else if (!strcasecmp(v->name, "amaflags")) { } else if (!strcasecmp(v->name, "amaflags")) {
format = ast_cdr_amaflags2int(v->value); format = ast_cdr_amaflags2int(v->value);
if (format < 0) { if (format < 0) {

Loading…
Cancel
Save