Commit 29d1d9a3 authored by jimw@mysql.com's avatar jimw@mysql.com

Clean up merge for fix for Bug #8350.

parents 793a505c f391223f
...@@ -47,6 +47,7 @@ ...@@ -47,6 +47,7 @@
extern "C" int gethostname(char *name, int namelen); extern "C" int gethostname(char *name, int namelen);
#endif #endif
static void time_out_user_resource_limits(THD *thd, USER_CONN *uc);
#ifndef NO_EMBEDDED_ACCESS_CHECKS #ifndef NO_EMBEDDED_ACCESS_CHECKS
static int check_for_max_user_connections(THD *thd, USER_CONN *uc); static int check_for_max_user_connections(THD *thd, USER_CONN *uc);
#endif #endif
...@@ -442,6 +443,7 @@ static int check_for_max_user_connections(THD *thd, USER_CONN *uc) ...@@ -442,6 +443,7 @@ static int check_for_max_user_connections(THD *thd, USER_CONN *uc)
error=1; error=1;
goto end; goto end;
} }
time_out_user_resource_limits(thd, uc);
if (uc->user_resources.connections && if (uc->user_resources.connections &&
uc->user_resources.connections <= uc->conn_per_hour) uc->user_resources.connections <= uc->conn_per_hour)
{ {
...@@ -543,36 +545,56 @@ bool is_update_query(enum enum_sql_command command) ...@@ -543,36 +545,56 @@ bool is_update_query(enum enum_sql_command command)
} }
/* /*
Check if maximum queries per hour limit has been reached Reset per-hour user resource limits when it has been more than
returns 0 if OK. an hour since they were last checked
In theory we would need a mutex in the USER_CONN structure for this to SYNOPSIS:
be 100 % safe, but as the worst scenario is that we would miss counting time_out_user_resource_limits()
a couple of queries, this isn't critical. thd Thread handler
*/ uc User connection details
NOTE:
This assumes that the LOCK_user_conn mutex has been acquired, so it is
safe to test and modify members of the USER_CONN structure.
*/
static bool check_mqh(THD *thd, uint check_command) static void time_out_user_resource_limits(THD *thd, USER_CONN *uc)
{ {
#ifdef NO_EMBEDDED_ACCESS_CHECKS bool error= 0;
return(0);
#else
bool error=0;
time_t check_time = thd->start_time ? thd->start_time : time(NULL); time_t check_time = thd->start_time ? thd->start_time : time(NULL);
USER_CONN *uc=thd->user_connect; DBUG_ENTER("time_out_user_resource_limits");
DBUG_ENTER("check_mqh");
DBUG_ASSERT(uc != 0);
/* If more than a hour since last check, reset resource checking */ /* If more than a hour since last check, reset resource checking */
if (check_time - uc->intime >= 3600) if (check_time - uc->intime >= 3600)
{ {
(void) pthread_mutex_lock(&LOCK_user_conn);
uc->questions=1; uc->questions=1;
uc->updates=0; uc->updates=0;
uc->conn_per_hour=0; uc->conn_per_hour=0;
uc->intime=check_time; uc->intime=check_time;
(void) pthread_mutex_unlock(&LOCK_user_conn);
} }
DBUG_VOID_RETURN;
}
/*
Check if maximum queries per hour limit has been reached
returns 0 if OK.
*/
static bool check_mqh(THD *thd, uint check_command)
{
#ifndef NO_EMBEDDED_ACCESS_CHECKS
bool error= 0;
time_t check_time = thd->start_time ? thd->start_time : time(NULL);
USER_CONN *uc=thd->user_connect;
DBUG_ENTER("check_mqh");
DBUG_ASSERT(uc != 0);
(void) pthread_mutex_lock(&LOCK_user_conn);
time_out_user_resource_limits(thd, uc);
/* Check that we have not done too many questions / hour */ /* Check that we have not done too many questions / hour */
if (uc->user_resources.questions && if (uc->user_resources.questions &&
uc->questions++ >= uc->user_resources.questions) uc->questions++ >= uc->user_resources.questions)
...@@ -595,7 +617,10 @@ static bool check_mqh(THD *thd, uint check_command) ...@@ -595,7 +617,10 @@ static bool check_mqh(THD *thd, uint check_command)
} }
} }
end: end:
(void) pthread_mutex_unlock(&LOCK_user_conn);
DBUG_RETURN(error); DBUG_RETURN(error);
#else
return (0);
#endif /* NO_EMBEDDED_ACCESS_CHECKS */ #endif /* NO_EMBEDDED_ACCESS_CHECKS */
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment