2 * "$Id: quotas.c 9793 2011-05-20 03:49:49Z mike $"
4 * Quota routines for the CUPS scheduler.
6 * Copyright 2007-2011 by Apple Inc.
7 * Copyright 1997-2007 by Easy Software Products.
9 * These coded instructions, statements, and computer programs are the
10 * property of Apple Inc. and are protected by Federal copyright
11 * law. Distribution and use rights are outlined in the file "LICENSE.txt"
12 * which should have been included with this file. If this file is
13 * file is missing or damaged, see the license at "http://www.cups.org/".
17 * cupsdFindQuota() - Find a quota record.
18 * cupsdFreeQuotas() - Free quotas for a printer.
19 * cupsdUpdateQuota() - Update quota data for the specified printer and user.
20 * add_quota() - Add a quota record for this printer and user.
21 * compare_quotas() - Compare two quota records...
25 * Include necessary headers...
35 static cupsd_quota_t *add_quota(cupsd_printer_t *p, const char *username);
36 static int compare_quotas(const cupsd_quota_t *q1,
37 const cupsd_quota_t *q2);
41 * 'cupsdFindQuota()' - Find a quota record.
44 cupsd_quota_t * /* O - Quota data */
46 cupsd_printer_t *p, /* I - Printer */
47 const char *username) /* I - User */
49 cupsd_quota_t *q, /* Quota data pointer */
50 match; /* Search data */
51 char *ptr; /* Pointer into username */
57 strlcpy(match.username, username, sizeof(match.username));
58 if ((ptr = strchr(match.username, '@')) != NULL)
59 *ptr = '\0'; /* Strip @domain/@KDC */
61 if ((q = (cupsd_quota_t *)cupsArrayFind(p->quotas, &match)) != NULL)
64 return (add_quota(p, username));
69 * 'cupsdFreeQuotas()' - Free quotas for a printer.
73 cupsdFreeQuotas(cupsd_printer_t *p) /* I - Printer */
75 cupsd_quota_t *q; /* Current quota record */
81 for (q = (cupsd_quota_t *)cupsArrayFirst(p->quotas);
83 q = (cupsd_quota_t *)cupsArrayNext(p->quotas))
86 cupsArrayDelete(p->quotas);
93 * 'cupsdUpdateQuota()' - Update quota data for the specified printer and user.
96 cupsd_quota_t * /* O - Quota data */
98 cupsd_printer_t *p, /* I - Printer */
99 const char *username, /* I - User */
100 int pages, /* I - Number of pages */
101 int k) /* I - Number of kilobytes */
103 cupsd_quota_t *q; /* Quota data */
104 cupsd_job_t *job; /* Current job */
105 time_t curtime; /* Current time */
106 ipp_attribute_t *attr; /* Job attribute */
112 if (!p->k_limit && !p->page_limit)
115 if ((q = cupsdFindQuota(p, username)) == NULL)
118 cupsdLogMessage(CUPSD_LOG_DEBUG,
119 "cupsdUpdateQuota: p=%s username=%s pages=%d k=%d",
120 p->name, username, pages, k);
122 curtime = time(NULL);
124 if (curtime < q->next_update)
126 q->page_count += pages;
133 curtime -= p->quota_period;
141 for (job = (cupsd_job_t *)cupsArrayFirst(Jobs);
143 job = (cupsd_job_t *)cupsArrayNext(Jobs))
146 * We only care about the current printer/class and user...
149 if (_cups_strcasecmp(job->dest, p->name) != 0 ||
150 _cups_strcasecmp(job->username, q->username) != 0)
154 * Make sure attributes are loaded; we always call cupsdLoadJob() to ensure
155 * the access_time member is updated so the job isn't unloaded right away...
158 if (!cupsdLoadJob(job))
161 if ((attr = ippFindAttribute(job->attrs, "time-at-completion",
162 IPP_TAG_INTEGER)) == NULL)
163 if ((attr = ippFindAttribute(job->attrs, "time-at-processing",
164 IPP_TAG_INTEGER)) == NULL)
165 attr = ippFindAttribute(job->attrs, "time-at-creation",
168 if (attr->values[0].integer < curtime)
171 * This job is too old to count towards the quota, ignore it...
174 if (JobAutoPurge && !job->printer && job->state_value > IPP_JOB_STOPPED)
175 cupsdDeleteJob(job, CUPSD_JOB_PURGE);
180 if (q->next_update == 0)
181 q->next_update = attr->values[0].integer + p->quota_period;
183 if ((attr = ippFindAttribute(job->attrs, "job-media-sheets-completed",
184 IPP_TAG_INTEGER)) != NULL)
185 q->page_count += attr->values[0].integer;
187 if ((attr = ippFindAttribute(job->attrs, "job-k-octets",
188 IPP_TAG_INTEGER)) != NULL)
189 q->k_count += attr->values[0].integer;
197 * 'add_quota()' - Add a quota record for this printer and user.
200 static cupsd_quota_t * /* O - Quota data */
201 add_quota(cupsd_printer_t *p, /* I - Printer */
202 const char *username) /* I - User */
204 cupsd_quota_t *q; /* New quota data */
205 char *ptr; /* Pointer into username */
212 p->quotas = cupsArrayNew((cups_array_func_t)compare_quotas, NULL);
217 if ((q = calloc(1, sizeof(cupsd_quota_t))) == NULL)
220 strlcpy(q->username, username, sizeof(q->username));
221 if ((ptr = strchr(q->username, '@')) != NULL)
222 *ptr = '\0'; /* Strip @domain/@KDC */
224 cupsArrayAdd(p->quotas, q);
231 * 'compare_quotas()' - Compare two quota records...
234 static int /* O - Result of comparison */
235 compare_quotas(const cupsd_quota_t *q1, /* I - First quota record */
236 const cupsd_quota_t *q2) /* I - Second quota record */
238 return (_cups_strcasecmp(q1->username, q2->username));
243 * End of "$Id: quotas.c 9793 2011-05-20 03:49:49Z mike $".