/* * OpenPBS (Portable Batch System) v2.3 Software License * * Copyright (c) 1999-2000 Veridian Information Solutions, Inc. * All rights reserved. * * --------------------------------------------------------------------------- * For a license to use or redistribute the OpenPBS software under conditions * other than those described below, or to purchase support for this software, * please contact Veridian Systems, PBS Products Department ("Licensor") at: * * www.OpenPBS.org +1 650 967-4675 sales@OpenPBS.org * 877 902-4PBS (US toll-free) * --------------------------------------------------------------------------- * * This license covers use of the OpenPBS v2.3 software (the "Software") at * your site or location, and, for certain users, redistribution of the * Software to other sites and locations. Use and redistribution of * OpenPBS v2.3 in source and binary forms, with or without modification, * are permitted provided that all of the following conditions are met. * After December 31, 2001, only conditions 3-6 must be met: * * 1. Commercial and/or non-commercial use of the Software is permitted * provided a current software registration is on file at www.OpenPBS.org. * If use of this software contributes to a publication, product, or * service, proper attribution must be given; see www.OpenPBS.org/credit.html * * 2. Redistribution in any form is only permitted for non-commercial, * non-profit purposes. There can be no charge for the Software or any * software incorporating the Software. Further, there can be no * expectation of revenue generated as a consequence of redistributing * the Software. * * 3. Any Redistribution of source code must retain the above copyright notice * and the acknowledgment contained in paragraph 6, this list of conditions * and the disclaimer contained in paragraph 7. * * 4. Any Redistribution in binary form must reproduce the above copyright * notice and the acknowledgment contained in paragraph 6, this list of * conditions and the disclaimer contained in paragraph 7 in the * documentation and/or other materials provided with the distribution. * * 5. Redistributions in any form must be accompanied by information on how to * obtain complete source code for the OpenPBS software and any * modifications and/or additions to the OpenPBS software. The source code * must either be included in the distribution or be available for no more * than the cost of distribution plus a nominal fee, and all modifications * and additions to the Software must be freely redistributable by any party * (including Licensor) without restriction. * * 6. All advertising materials mentioning features or use of the Software must * display the following acknowledgment: * * "This product includes software developed by NASA Ames Research Center, * Lawrence Livermore National Laboratory, and Veridian Information * Solutions, Inc. * Visit www.OpenPBS.org for OpenPBS software support, * products, and information." * * 7. DISCLAIMER OF WARRANTY * * THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND. ANY EXPRESS * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NON-INFRINGEMENT * ARE EXPRESSLY DISCLAIMED. * * IN NO EVENT SHALL VERIDIAN CORPORATION, ITS AFFILIATED COMPANIES, OR THE * U.S. GOVERNMENT OR ANY OF ITS AGENCIES BE LIABLE FOR ANY DIRECT OR INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * * This license will be governed by the laws of the Commonwealth of Virginia, * without reference to its choice of law rules. */ /* * This file contines two main library entries: * pbs_selectjob() * pbs_selstat() * * * pbs_selectjob() - the SelectJob request * Return a list of job ids that meet certain selection criteria. */ #include /* the master config generated by configure */ #include #include #include #include "libpbs.h" #include "dis.h" #include "tcp.h" /* tcp_chan */ static int PBSD_select_put(int, int, struct attropl *, char *); static int PBSD_selectattr_put(int, int, struct attropl *, struct attrl *, char *); static char **PBSD_select_get(int *, int); char **pbs_selectjob_err( int c, struct attropl *attrib, char *extend, int *local_errno) { if (PBSD_select_put(c, PBS_BATCH_SelectJobs, attrib, extend) == 0) return (PBSD_select_get(local_errno, c)); else return ((char **)0); } /* END pbs_selectjob_err() */ char **pbs_selectjob( int c, struct attropl *attrib, char *extend) { pbs_errno = 0; if (PBSD_select_put(c, PBS_BATCH_SelectJobs, attrib, extend) == 0) return (PBSD_select_get(&pbs_errno, c)); else return ((char **)0); } /* END pbs_selectjob() */ /* * pbs_selstat() - Selectable status * Return status information for jobs that meet certain selection * criteria. This is a short-cut combination of pbs_selecljob() * and repeated pbs_statjob(). */ struct batch_status * pbs_selstat_err( int c, struct attropl *attrib, char *extend, int *local_errno) { if (PBSD_select_put(c, PBS_BATCH_SelStat, attrib, extend) == 0) return (PBSD_status_get(local_errno, c)); else return ((struct batch_status *)0); } /* END pbs_selstat_err() */ struct batch_status * pbs_selstat( int c, struct attropl *attrib, char *extend) { pbs_errno = 0; if (PBSD_select_put(c, PBS_BATCH_SelStat, attrib, extend) == 0) return (PBSD_status_get(&pbs_errno, c)); else return ((struct batch_status *)0); } /* END pbs_selstat() */ /* * pbs_selstatattr() - Selectable status with attribute request * Return selected status information for jobs that meet certain * selection criteria. This is pbs_selstat() with the ability * to only request certain attributes. */ struct batch_status * pbs_selstatattr_err( int c, struct attropl *attropl, struct attrl *attrib, char *extend, int *local_errno) { if (PBSD_selectattr_put(c, PBS_BATCH_SelStatAttr, attropl, attrib, extend) == 0) return (PBSD_status_get(local_errno, c)); else return ((struct batch_status *)0); } /* END pbs_selstatattr_err() */ struct batch_status * pbs_selstatattr( int c, struct attropl *attropl, struct attrl *attrib, char *extend) { pbs_errno = 0; if (PBSD_selectattr_put(c, PBS_BATCH_SelStatAttr, attropl, attrib, extend) == 0) return (PBSD_status_get(&pbs_errno, c)); else return ((struct batch_status *)0); } /* END pbs_selstat() */ static int PBSD_select_put( int c, int type, struct attropl *attrib, char *extend) { int rc = PBSE_NONE; int sock; struct tcp_chan *chan = NULL; pthread_mutex_lock(connection[c].ch_mutex); sock = connection[c].ch_socket; /* setup DIS support routines for following DIS calls */ if ((chan = DIS_tcp_setup(sock)) == NULL) { pthread_mutex_unlock(connection[c].ch_mutex); rc = PBSE_PROTOCOL; return rc; } else if ((rc = encode_DIS_ReqHdr(chan, type, pbs_current_user)) || (rc = encode_DIS_attropl(chan, attrib)) || (rc = encode_DIS_ReqExtend(chan, extend))) { connection[c].ch_errtxt = strdup(dis_emsg[rc]); pthread_mutex_unlock(connection[c].ch_mutex); DIS_tcp_cleanup(chan); return (PBSE_PROTOCOL); } pthread_mutex_unlock(connection[c].ch_mutex); /* write data */ if (DIS_tcp_wflush(chan)) { rc = PBSE_PROTOCOL; } DIS_tcp_cleanup(chan); return rc; } /* END PBSD_select_put() */ static int PBSD_selectattr_put( int c, int type, struct attropl *attropl, struct attrl *attrib, char *extend) { int rc = PBSE_NONE; int sock; struct tcp_chan *chan = NULL; pthread_mutex_lock(connection[c].ch_mutex); sock = connection[c].ch_socket; /* setup DIS support routines for following DIS calls */ if ((chan = DIS_tcp_setup(sock)) == NULL) { pthread_mutex_unlock(connection[c].ch_mutex); rc = PBSE_PROTOCOL; return rc; } else if ((rc = encode_DIS_ReqHdr(chan, type, pbs_current_user)) || (rc = encode_DIS_attropl(chan, attropl)) || (rc = encode_DIS_attrl(chan, attrib)) || (rc = encode_DIS_ReqExtend(chan, extend))) { connection[c].ch_errtxt = strdup(dis_emsg[rc]); pthread_mutex_unlock(connection[c].ch_mutex); DIS_tcp_cleanup(chan); return (PBSE_PROTOCOL); } pthread_mutex_unlock(connection[c].ch_mutex); /* write data */ if (DIS_tcp_wflush(chan)) { rc = PBSE_PROTOCOL; } DIS_tcp_cleanup(chan); return rc; } /* END PBSD_selectattr_put() */ static char **PBSD_select_get( int *local_errno, int c) { int i; struct batch_reply *reply; int njobs; char *sp; int stringtot; size_t totsize; struct brp_select *sr; char **retval = (char **)NULL; pthread_mutex_lock(connection[c].ch_mutex); /* read reply from stream */ reply = PBSD_rdrpy(local_errno, c); if (reply == NULL) { *local_errno = PBSE_PROTOCOL; } else if (reply->brp_choice != 0 && reply->brp_choice != BATCH_REPLY_CHOICE_Text && reply->brp_choice != BATCH_REPLY_CHOICE_Select) { *local_errno = PBSE_PROTOCOL; } else if (connection[c].ch_errno == 0) { /* process the reply -- first, build a linked list of the strings we extract from the reply, keeping track of the amount of space used... */ stringtot = 0; njobs = 0; sr = reply->brp_un.brp_select; while (sr != (struct brp_select *)NULL) { stringtot += strlen(sr->brp_jobid) + 1; njobs++; sr = sr->brp_next; } /* ...then copy all the strings into one of "Bob's structures", freeing all strings we just allocated... */ totsize = stringtot + (njobs + 1) * (sizeof(char *)); retval = (char **)calloc(1, totsize); if (retval == (char **)NULL) { *local_errno = PBSE_SYSTEM; pthread_mutex_unlock(connection[c].ch_mutex); PBSD_FreeReply(reply); return(NULL); } sr = reply->brp_un.brp_select; sp = (char *)retval + (njobs + 1) * sizeof(char *); for (i = 0; i < njobs; i++) { retval[i] = sp; strcpy(sp, sr->brp_jobid); sp += strlen(sp) + 1; sr = sr->brp_next; } retval[i] = (char *)NULL; } pthread_mutex_unlock(connection[c].ch_mutex); PBSD_FreeReply(reply); return(retval); } /* END PBSD_select_get() */