author | Sergey Poznyakoff <gray@gnu.org> | 2018-05-15 05:50:08 (GMT) |
---|---|---|
committer | Sergey Poznyakoff <gray@gnu.org> | 2018-05-15 06:05:31 (GMT) |
commit | b5f4388a2da1b94ce8c8e45a990fd51b2f52dae4 (patch) (side-by-side diff) | |
tree | 28bcf8cf313309636357c470faed4456837db7c8 | |
download | genrc-b5f4388a2da1b94ce8c8e45a990fd51b2f52dae4.tar.gz genrc-b5f4388a2da1b94ce8c8e45a990fd51b2f52dae4.tar.bz2 |
Initial commit
-rw-r--r-- | .gitignore | 36 | ||||
-rw-r--r-- | .gitmodules | 3 | ||||
-rw-r--r-- | COPYING | 674 | ||||
-rw-r--r-- | Makefile.am | 1 | ||||
-rw-r--r-- | configure.ac | 35 | ||||
m--------- | grecs | 0 | ||||
-rw-r--r-- | src/.gitignore | 1 | ||||
-rw-r--r-- | src/Makefile.am | 32 | ||||
-rw-r--r-- | src/com_reload.c | 17 | ||||
-rw-r--r-- | src/com_restart.c | 12 | ||||
-rw-r--r-- | src/com_start.c | 121 | ||||
-rw-r--r-- | src/com_status.c | 38 | ||||
-rw-r--r-- | src/com_stop.c | 66 | ||||
-rw-r--r-- | src/err.c | 47 | ||||
-rw-r--r-- | src/genrc.c | 469 | ||||
-rw-r--r-- | src/genrc.h | 137 | ||||
-rw-r--r-- | src/match_exact.c | 20 | ||||
-rw-r--r-- | src/match_glob.c | 22 | ||||
-rw-r--r-- | src/match_pcre.c | 60 | ||||
-rw-r--r-- | src/match_regex.c | 61 | ||||
-rw-r--r-- | src/pid_config.c | 91 | ||||
-rw-r--r-- | src/pid_file.c | 70 | ||||
-rw-r--r-- | src/pid_grep.c | 80 | ||||
-rw-r--r-- | src/pid_proc.c | 335 | ||||
-rw-r--r-- | src/pid_ps.c | 137 | ||||
-rw-r--r-- | src/pidfrom.c | 52 | ||||
-rw-r--r-- | src/pidlist.c | 121 | ||||
-rw-r--r-- | src/procscan.c | 95 | ||||
-rw-r--r-- | src/sentinel.c | 207 | ||||
-rw-r--r-- | src/transform.c | 560 |
30 files changed, 3600 insertions, 0 deletions
diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..438eec5 --- a/dev/null +++ b/.gitignore @@ -0,0 +1,36 @@ +/ABOUT-NLS +/ABOUT-NLS~ +*.a +*.la +*.lo +*.o +*.tar.* +*~ +.bootstrap +.deps +.emacs* +.gdbinit +.libs +ABOUT-NLS +ChangeLog +INSTALL +Makefile +Makefile.in +TAGS +aclocal.m4 +autom4te.cache/ +config.h +config.h.in +config.log +config.status +configure +core +gnu +m4 +stamp-h1 +autoscan.log +compile +depcomp +install-sh +missing +ylwrap diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..fea8f96 --- a/dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "grecs"] + path = grecs + url = git://git.gnu.org.ua/grecs.git @@ -0,0 +1,674 @@ + GNU GENERAL PUBLIC LICENSE + Version 3, 29 June 2007 + + Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/> + Everyone is permitted to copy and distribute verbatim copies + of this license document, but changing it is not allowed. + + Preamble + + The GNU General Public License is a free, copyleft license for +software and other kinds of works. + + The licenses for most software and other practical works are designed +to take away your freedom to share and change the works. By contrast, +the GNU General Public License is intended to guarantee your freedom to +share and change all versions of a program--to make sure it remains free +software for all its users. We, the Free Software Foundation, use the +GNU General Public License for most of our software; it applies also to +any other work released this way by its authors. You can apply it to +your programs, too. + + When we speak of free software, we are referring to freedom, not +price. Our General Public Licenses are designed to make sure that you +have the freedom to distribute copies of free software (and charge for +them if you wish), that you receive source code or can get it if you +want it, that you can change the software or use pieces of it in new +free programs, and that you know you can do these things. + + To protect your rights, we need to prevent others from denying you +these rights or asking you to surrender the rights. Therefore, you have +certain responsibilities if you distribute copies of the software, or if +you modify it: responsibilities to respect the freedom of others. + + For example, if you distribute copies of such a program, whether +gratis or for a fee, you must pass on to the recipients the same +freedoms that you received. You must make sure that they, too, receive +or can get the source code. And you must show them these terms so they +know their rights. + + Developers that use the GNU GPL protect your rights with two steps: +(1) assert copyright on the software, and (2) offer you this License +giving you legal permission to copy, distribute and/or modify it. + + For the developers' and authors' protection, the GPL clearly explains +that there is no warranty for this free software. For both users' and +authors' sake, the GPL requires that modified versions be marked as +changed, so that their problems will not be attributed erroneously to +authors of previous versions. + + Some devices are designed to deny users access to install or run +modified versions of the software inside them, although the manufacturer +can do so. This is fundamentally incompatible with the aim of +protecting users' freedom to change the software. The systematic +pattern of such abuse occurs in the area of products for individuals to +use, which is precisely where it is most unacceptable. Therefore, we +have designed this version of the GPL to prohibit the practice for those +products. If such problems arise substantially in other domains, we +stand ready to extend this provision to those domains in future versions +of the GPL, as needed to protect the freedom of users. + + Finally, every program is threatened constantly by software patents. +States should not allow patents to restrict development and use of +software on general-purpose computers, but in those that do, we wish to +avoid the special danger that patents applied to a free program could +make it effectively proprietary. To prevent this, the GPL assures that +patents cannot be used to render the program non-free. + + The precise terms and conditions for copying, distribution and +modification follow. + + TERMS AND CONDITIONS + + 0. Definitions. + + "This License" refers to version 3 of the GNU General Public License. + + "Copyright" also means copyright-like laws that apply to other kinds of +works, such as semiconductor masks. + + "The Program" refers to any copyrightable work licensed under this +License. Each licensee is addressed as "you". "Licensees" and +"recipients" may be individuals or organizations. + + To "modify" a work means to copy from or adapt all or part of the work +in a fashion requiring copyright permission, other than the making of an +exact copy. The resulting work is called a "modified version" of the +earlier work or a work "based on" the earlier work. + + A "covered work" means either the unmodified Program or a work based +on the Program. + + To "propagate" a work means to do anything with it that, without +permission, would make you directly or secondarily liable for +infringement under applicable copyright law, except executing it on a +computer or modifying a private copy. Propagation includes copying, +distribution (with or without modification), making available to the +public, and in some countries other activities as well. + + To "convey" a work means any kind of propagation that enables other +parties to make or receive copies. Mere interaction with a user through +a computer network, with no transfer of a copy, is not conveying. + + An interactive user interface displays "Appropriate Legal Notices" +to the extent that it includes a convenient and prominently visible +feature that (1) displays an appropriate copyright notice, and (2) +tells the user that there is no warranty for the work (except to the +extent that warranties are provided), that licensees may convey the +work under this License, and how to view a copy of this License. If +the interface presents a list of user commands or options, such as a +menu, a prominent item in the list meets this criterion. + + 1. Source Code. + + The "source code" for a work means the preferred form of the work +for making modifications to it. "Object code" means any non-source +form of a work. + + A "Standard Interface" means an interface that either is an official +standard defined by a recognized standards body, or, in the case of +interfaces specified for a particular programming language, one that +is widely used among developers working in that language. + + The "System Libraries" of an executable work include anything, other +than the work as a whole, that (a) is included in the normal form of +packaging a Major Component, but which is not part of that Major +Component, and (b) serves only to enable use of the work with that +Major Component, or to implement a Standard Interface for which an +implementation is available to the public in source code form. A +"Major Component", in this context, means a major essential component +(kernel, window system, and so on) of the specific operating system +(if any) on which the executable work runs, or a compiler used to +produce the work, or an object code interpreter used to run it. + + The "Corresponding Source" for a work in object code form means all +the source code needed to generate, install, and (for an executable +work) run the object code and to modify the work, including scripts to +control those activities. However, it does not include the work's +System Libraries, or general-purpose tools or generally available free +programs which are used unmodified in performing those activities but +which are not part of the work. For example, Corresponding Source +includes interface definition files associated with source files for +the work, and the source code for shared libraries and dynamically +linked subprograms that the work is specifically designed to require, +such as by intimate data communication or control flow between those +subprograms and other parts of the work. + + The Corresponding Source need not include anything that users +can regenerate automatically from other parts of the Corresponding +Source. + + The Corresponding Source for a work in source code form is that +same work. + + 2. Basic Permissions. + + All rights granted under this License are granted for the term of +copyright on the Program, and are irrevocable provided the stated +conditions are met. This License explicitly affirms your unlimited +permission to run the unmodified Program. The output from running a +covered work is covered by this License only if the output, given its +content, constitutes a covered work. This License acknowledges your +rights of fair use or other equivalent, as provided by copyright law. + + You may make, run and propagate covered works that you do not +convey, without conditions so long as your license otherwise remains +in force. You may convey covered works to others for the sole purpose +of having them make modifications exclusively for you, or provide you +with facilities for running those works, provided that you comply with +the terms of this License in conveying all material for which you do +not control copyright. Those thus making or running the covered works +for you must do so exclusively on your behalf, under your direction +and control, on terms that prohibit them from making any copies of +your copyrighted material outside their relationship with you. + + Conveying under any other circumstances is permitted solely under +the conditions stated below. Sublicensing is not allowed; section 10 +makes it unnecessary. + + 3. Protecting Users' Legal Rights From Anti-Circumvention Law. + + No covered work shall be deemed part of an effective technological +measure under any applicable law fulfilling obligations under article +11 of the WIPO copyright treaty adopted on 20 December 1996, or +similar laws prohibiting or restricting circumvention of such +measures. + + When you convey a covered work, you waive any legal power to forbid +circumvention of technological measures to the extent such circumvention +is effected by exercising rights under this License with respect to +the covered work, and you disclaim any intention to limit operation or +modification of the work as a means of enforcing, against the work's +users, your or third parties' legal rights to forbid circumvention of +technological measures. + + 4. Conveying Verbatim Copies. + + You may convey verbatim copies of the Program's source code as you +receive it, in any medium, provided that you conspicuously and +appropriately publish on each copy an appropriate copyright notice; +keep intact all notices stating that this License and any +non-permissive terms added in accord with section 7 apply to the code; +keep intact all notices of the absence of any warranty; and give all +recipients a copy of this License along with the Program. + + You may charge any price or no price for each copy that you convey, +and you may offer support or warranty protection for a fee. + + 5. Conveying Modified Source Versions. + + You may convey a work based on the Program, or the modifications to +produce it from the Program, in the form of source code under the +terms of section 4, provided that you also meet all of these conditions: + + a) The work must carry prominent notices stating that you modified + it, and giving a relevant date. + + b) The work must carry prominent notices stating that it is + released under this License and any conditions added under section + 7. This requirement modifies the requirement in section 4 to + "keep intact all notices". + + c) You must license the entire work, as a whole, under this + License to anyone who comes into possession of a copy. This + License will therefore apply, along with any applicable section 7 + additional terms, to the whole of the work, and all its parts, + regardless of how they are packaged. This License gives no + permission to license the work in any other way, but it does not + invalidate such permission if you have separately received it. + + d) If the work has interactive user interfaces, each must display + Appropriate Legal Notices; however, if the Program has interactive + interfaces that do not display Appropriate Legal Notices, your + work need not make them do so. + + A compilation of a covered work with other separate and independent +works, which are not by their nature extensions of the covered work, +and which are not combined with it such as to form a larger program, +in or on a volume of a storage or distribution medium, is called an +"aggregate" if the compilation and its resulting copyright are not +used to limit the access or legal rights of the compilation's users +beyond what the individual works permit. Inclusion of a covered work +in an aggregate does not cause this License to apply to the other +parts of the aggregate. + + 6. Conveying Non-Source Forms. + + You may convey a covered work in object code form under the terms +of sections 4 and 5, provided that you also convey the +machine-readable Corresponding Source under the terms of this License, +in one of these ways: + + a) Convey the object code in, or embodied in, a physical product + (including a physical distribution medium), accompanied by the + Corresponding Source fixed on a durable physical medium + customarily used for software interchange. + + b) Convey the object code in, or embodied in, a physical product + (including a physical distribution medium), accompanied by a + written offer, valid for at least three years and valid for as + long as you offer spare parts or customer support for that product + model, to give anyone who possesses the object code either (1) a + copy of the Corresponding Source for all the software in the + product that is covered by this License, on a durable physical + medium customarily used for software interchange, for a price no + more than your reasonable cost of physically performing this + conveying of source, or (2) access to copy the + Corresponding Source from a network server at no charge. + + c) Convey individual copies of the object code with a copy of the + written offer to provide the Corresponding Source. This + alternative is allowed only occasionally and noncommercially, and + only if you received the object code with such an offer, in accord + with subsection 6b. + + d) Convey the object code by offering access from a designated + place (gratis or for a charge), and offer equivalent access to the + Corresponding Source in the same way through the same place at no + further charge. You need not require recipients to copy the + Corresponding Source along with the object code. If the place to + copy the object code is a network server, the Corresponding Source + may be on a different server (operated by you or a third party) + that supports equivalent copying facilities, provided you maintain + clear directions next to the object code saying where to find the + Corresponding Source. Regardless of what server hosts the + Corresponding Source, you remain obligated to ensure that it is + available for as long as needed to satisfy these requirements. + + e) Convey the object code using peer-to-peer transmission, provided + you inform other peers where the object code and Corresponding + Source of the work are being offered to the general public at no + charge under subsection 6d. + + A separable portion of the object code, whose source code is excluded +from the Corresponding Source as a System Library, need not be +included in conveying the object code work. + + A "User Product" is either (1) a "consumer product", which means any +tangible personal property which is normally used for personal, family, +or household purposes, or (2) anything designed or sold for incorporation +into a dwelling. In determining whether a product is a consumer product, +doubtful cases shall be resolved in favor of coverage. For a particular +product received by a particular user, "normally used" refers to a +typical or common use of that class of product, regardless of the status +of the particular user or of the way in which the particular user +actually uses, or expects or is expected to use, the product. A product +is a consumer product regardless of whether the product has substantial +commercial, industrial or non-consumer uses, unless such uses represent +the only significant mode of use of the product. + + "Installation Information" for a User Product means any methods, +procedures, authorization keys, or other information required to install +and execute modified versions of a covered work in that User Product from +a modified version of its Corresponding Source. The information must +suffice to ensure that the continued functioning of the modified object +code is in no case prevented or interfered with solely because +modification has been made. + + If you convey an object code work under this section in, or with, or +specifically for use in, a User Product, and the conveying occurs as +part of a transaction in which the right of possession and use of the +User Product is transferred to the recipient in perpetuity or for a +fixed term (regardless of how the transaction is characterized), the +Corresponding Source conveyed under this section must be accompanied +by the Installation Information. But this requirement does not apply +if neither you nor any third party retains the ability to install +modified object code on the User Product (for example, the work has +been installed in ROM). + + The requirement to provide Installation Information does not include a +requirement to continue to provide support service, warranty, or updates +for a work that has been modified or installed by the recipient, or for +the User Product in which it has been modified or installed. Access to a +network may be denied when the modification itself materially and +adversely affects the operation of the network or violates the rules and +protocols for communication across the network. + + Corresponding Source conveyed, and Installation Information provided, +in accord with this section must be in a format that is publicly +documented (and with an implementation available to the public in +source code form), and must require no special password or key for +unpacking, reading or copying. + + 7. Additional Terms. + + "Additional permissions" are terms that supplement the terms of this +License by making exceptions from one or more of its conditions. +Additional permissions that are applicable to the entire Program shall +be treated as though they were included in this License, to the extent +that they are valid under applicable law. If additional permissions +apply only to part of the Program, that part may be used separately +under those permissions, but the entire Program remains governed by +this License without regard to the additional permissions. + + When you convey a copy of a covered work, you may at your option +remove any additional permissions from that copy, or from any part of +it. (Additional permissions may be written to require their own +removal in certain cases when you modify the work.) You may place +additional permissions on material, added by you to a covered work, +for which you have or can give appropriate copyright permission. + + Notwithstanding any other provision of this License, for material you +add to a covered work, you may (if authorized by the copyright holders of +that material) supplement the terms of this License with terms: + + a) Disclaiming warranty or limiting liability differently from the + terms of sections 15 and 16 of this License; or + + b) Requiring preservation of specified reasonable legal notices or + author attributions in that material or in the Appropriate Legal + Notices displayed by works containing it; or + + c) Prohibiting misrepresentation of the origin of that material, or + requiring that modified versions of such material be marked in + reasonable ways as different from the original version; or + + d) Limiting the use for publicity purposes of names of licensors or + authors of the material; or + + e) Declining to grant rights under trademark law for use of some + trade names, trademarks, or service marks; or + + f) Requiring indemnification of licensors and authors of that + material by anyone who conveys the material (or modified versions of + it) with contractual assumptions of liability to the recipient, for + any liability that these contractual assumptions directly impose on + those licensors and authors. + + All other non-permissive additional terms are considered "further +restrictions" within the meaning of section 10. If the Program as you +received it, or any part of it, contains a notice stating that it is +governed by this License along with a term that is a further +restriction, you may remove that term. If a license document contains +a further restriction but permits relicensing or conveying under this +License, you may add to a covered work material governed by the terms +of that license document, provided that the further restriction does +not survive such relicensing or conveying. + + If you add terms to a covered work in accord with this section, you +must place, in the relevant source files, a statement of the +additional terms that apply to those files, or a notice indicating +where to find the applicable terms. + + Additional terms, permissive or non-permissive, may be stated in the +form of a separately written license, or stated as exceptions; +the above requirements apply either way. + + 8. Termination. + + You may not propagate or modify a covered work except as expressly +provided under this License. Any attempt otherwise to propagate or +modify it is void, and will automatically terminate your rights under +this License (including any patent licenses granted under the third +paragraph of section 11). + + However, if you cease all violation of this License, then your +license from a particular copyright holder is reinstated (a) +provisionally, unless and until the copyright holder explicitly and +finally terminates your license, and (b) permanently, if the copyright +holder fails to notify you of the violation by some reasonable means +prior to 60 days after the cessation. + + Moreover, your license from a particular copyright holder is +reinstated permanently if the copyright holder notifies you of the +violation by some reasonable means, this is the first time you have +received notice of violation of this License (for any work) from that +copyright holder, and you cure the violation prior to 30 days after +your receipt of the notice. + + Termination of your rights under this section does not terminate the +licenses of parties who have received copies or rights from you under +this License. If your rights have been terminated and not permanently +reinstated, you do not qualify to receive new licenses for the same +material under section 10. + + 9. Acceptance Not Required for Having Copies. + + You are not required to accept this License in order to receive or +run a copy of the Program. Ancillary propagation of a covered work +occurring solely as a consequence of using peer-to-peer transmission +to receive a copy likewise does not require acceptance. However, +nothing other than this License grants you permission to propagate or +modify any covered work. These actions infringe copyright if you do +not accept this License. Therefore, by modifying or propagating a +covered work, you indicate your acceptance of this License to do so. + + 10. Automatic Licensing of Downstream Recipients. + + Each time you convey a covered work, the recipient automatically +receives a license from the original licensors, to run, modify and +propagate that work, subject to this License. You are not responsible +for enforcing compliance by third parties with this License. + + An "entity transaction" is a transaction transferring control of an +organization, or substantially all assets of one, or subdividing an +organization, or merging organizations. If propagation of a covered +work results from an entity transaction, each party to that +transaction who receives a copy of the work also receives whatever +licenses to the work the party's predecessor in interest had or could +give under the previous paragraph, plus a right to possession of the +Corresponding Source of the work from the predecessor in interest, if +the predecessor has it or can get it with reasonable efforts. + + You may not impose any further restrictions on the exercise of the +rights granted or affirmed under this License. For example, you may +not impose a license fee, royalty, or other charge for exercise of +rights granted under this License, and you may not initiate litigation +(including a cross-claim or counterclaim in a lawsuit) alleging that +any patent claim is infringed by making, using, selling, offering for +sale, or importing the Program or any portion of it. + + 11. Patents. + + A "contributor" is a copyright holder who authorizes use under this +License of the Program or a work on which the Program is based. The +work thus licensed is called the contributor's "contributor version". + + A contributor's "essential patent claims" are all patent claims +owned or controlled by the contributor, whether already acquired or +hereafter acquired, that would be infringed by some manner, permitted +by this License, of making, using, or selling its contributor version, +but do not include claims that would be infringed only as a +consequence of further modification of the contributor version. For +purposes of this definition, "control" includes the right to grant +patent sublicenses in a manner consistent with the requirements of +this License. + + Each contributor grants you a non-exclusive, worldwide, royalty-free +patent license under the contributor's essential patent claims, to +make, use, sell, offer for sale, import and otherwise run, modify and +propagate the contents of its contributor version. + + In the following three paragraphs, a "patent license" is any express +agreement or commitment, however denominated, not to enforce a patent +(such as an express permission to practice a patent or covenant not to +sue for patent infringement). To "grant" such a patent license to a +party means to make such an agreement or commitment not to enforce a +patent against the party. + + If you convey a covered work, knowingly relying on a patent license, +and the Corresponding Source of the work is not available for anyone +to copy, free of charge and under the terms of this License, through a +publicly available network server or other readily accessible means, +then you must either (1) cause the Corresponding Source to be so +available, or (2) arrange to deprive yourself of the benefit of the +patent license for this particular work, or (3) arrange, in a manner +consistent with the requirements of this License, to extend the patent +license to downstream recipients. "Knowingly relying" means you have +actual knowledge that, but for the patent license, your conveying the +covered work in a country, or your recipient's use of the covered work +in a country, would infringe one or more identifiable patents in that +country that you have reason to believe are valid. + + If, pursuant to or in connection with a single transaction or +arrangement, you convey, or propagate by procuring conveyance of, a +covered work, and grant a patent license to some of the parties +receiving the covered work authorizing them to use, propagate, modify +or convey a specific copy of the covered work, then the patent license +you grant is automatically extended to all recipients of the covered +work and works based on it. + + A patent license is "discriminatory" if it does not include within +the scope of its coverage, prohibits the exercise of, or is +conditioned on the non-exercise of one or more of the rights that are +specifically granted under this License. You may not convey a covered +work if you are a party to an arrangement with a third party that is +in the business of distributing software, under which you make payment +to the third party based on the extent of your activity of conveying +the work, and under which the third party grants, to any of the +parties who would receive the covered work from you, a discriminatory +patent license (a) in connection with copies of the covered work +conveyed by you (or copies made from those copies), or (b) primarily +for and in connection with specific products or compilations that +contain the covered work, unless you entered into that arrangement, +or that patent license was granted, prior to 28 March 2007. + + Nothing in this License shall be construed as excluding or limiting +any implied license or other defenses to infringement that may +otherwise be available to you under applicable patent law. + + 12. No Surrender of Others' Freedom. + + If conditions are imposed on you (whether by court order, agreement or +otherwise) that contradict the conditions of this License, they do not +excuse you from the conditions of this License. If you cannot convey a +covered work so as to satisfy simultaneously your obligations under this +License and any other pertinent obligations, then as a consequence you may +not convey it at all. For example, if you agree to terms that obligate you +to collect a royalty for further conveying from those to whom you convey +the Program, the only way you could satisfy both those terms and this +License would be to refrain entirely from conveying the Program. + + 13. Use with the GNU Affero General Public License. + + Notwithstanding any other provision of this License, you have +permission to link or combine any covered work with a work licensed +under version 3 of the GNU Affero General Public License into a single +combined work, and to convey the resulting work. The terms of this +License will continue to apply to the part which is the covered work, +but the special requirements of the GNU Affero General Public License, +section 13, concerning interaction through a network will apply to the +combination as such. + + 14. Revised Versions of this License. + + The Free Software Foundation may publish revised and/or new versions of +the GNU General Public License from time to time. Such new versions will +be similar in spirit to the present version, but may differ in detail to +address new problems or concerns. + + Each version is given a distinguishing version number. If the +Program specifies that a certain numbered version of the GNU General +Public License "or any later version" applies to it, you have the +option of following the terms and conditions either of that numbered +version or of any later version published by the Free Software +Foundation. If the Program does not specify a version number of the +GNU General Public License, you may choose any version ever published +by the Free Software Foundation. + + If the Program specifies that a proxy can decide which future +versions of the GNU General Public License can be used, that proxy's +public statement of acceptance of a version permanently authorizes you +to choose that version for the Program. + + Later license versions may give you additional or different +permissions. However, no additional obligations are imposed on any +author or copyright holder as a result of your choosing to follow a +later version. + + 15. Disclaimer of Warranty. + + THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY +APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT +HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY +OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, +THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR +PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM +IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF +ALL NECESSARY SERVICING, REPAIR OR CORRECTION. + + 16. Limitation of Liability. + + IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING +WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS +THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY +GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE +USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF +DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD +PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), +EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF +SUCH DAMAGES. + + 17. Interpretation of Sections 15 and 16. + + If the disclaimer of warranty and limitation of liability provided +above cannot be given local legal effect according to their terms, +reviewing courts shall apply local law that most closely approximates +an absolute waiver of all civil liability in connection with the +Program, unless a warranty or assumption of liability accompanies a +copy of the Program in return for a fee. + + END OF TERMS AND CONDITIONS + + How to Apply These Terms to Your New Programs + + If you develop a new program, and you want it to be of the greatest +possible use to the public, the best way to achieve this is to make it +free software which everyone can redistribute and change under these terms. + + To do so, attach the following notices to the program. It is safest +to attach them to the start of each source file to most effectively +state the exclusion of warranty; and each file should have at least +the "copyright" line and a pointer to where the full notice is found. + + <one line to give the program's name and a brief idea of what it does.> + Copyright (C) <year> <name of author> + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. + +Also add information on how to contact you by electronic and paper mail. + + If the program does terminal interaction, make it output a short +notice like this when it starts in an interactive mode: + + <program> Copyright (C) <year> <name of author> + This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'. + This is free software, and you are welcome to redistribute it + under certain conditions; type `show c' for details. + +The hypothetical commands `show w' and `show c' should show the appropriate +parts of the General Public License. Of course, your program's commands +might be different; for a GUI interface, you would use an "about box". + + You should also get your employer (if you work as a programmer) or school, +if any, to sign a "copyright disclaimer" for the program, if necessary. +For more information on this, and how to apply and follow the GNU GPL, see +<http://www.gnu.org/licenses/>. + + The GNU General Public License does not permit incorporating your program +into proprietary programs. If your program is a subroutine library, you +may consider it more useful to permit linking proprietary applications with +the library. If this is what you want to do, use the GNU Lesser General +Public License instead of this License. But first, please read +<http://www.gnu.org/philosophy/why-not-lgpl.html>. diff --git a/Makefile.am b/Makefile.am new file mode 100644 index 0000000..31e9e5f --- a/dev/null +++ b/Makefile.am @@ -0,0 +1 @@ +SUBDIRS = grecs src diff --git a/configure.ac b/configure.ac new file mode 100644 index 0000000..764cddc --- a/dev/null +++ b/configure.ac @@ -0,0 +1,35 @@ +# -*- Autoconf -*- +# Process this file with autoconf to produce a configure script. + +AC_PREREQ([2.69]) +AC_INIT([genrc], [0.1], [gray@gnu.org]) +AC_CONFIG_SRCDIR([src/genrc.c]) +AC_CONFIG_HEADERS([config.h]) +AM_INIT_AUTOMAKE([1.11 foreign silent-rules]) +AC_CONFIG_MACRO_DIR(grecs/am) + +# Enable silent rules by default +AM_SILENT_RULES([yes]) + +# Checks for programs. +AC_PROG_CC +AC_PROG_RANLIB + +# Checks for libraries. +AC_CHECK_LIB(pcre, main) + +# Checks for header files. +AC_CHECK_HEADERS([getopt.h pcre.h]) + +# Checks for library functions. +AC_CHECK_FUNCS([getdtablesize]) + +GRECS_SETUP(grecs, [all-parsers git2chg]) + +AM_CONDITIONAL([COND_PCRE], + [test "$ac_cv_header_pcre_h" = yes && test "$ac_cv_lib_pcre_main" = yes]) + +AC_CONFIG_FILES([Makefile + src/Makefile]) + +AC_OUTPUT diff --git a/grecs b/grecs new file mode 160000 +Subproject 27ca9f5819d0e44173a5f61a241cd066ad15713 diff --git a/src/.gitignore b/src/.gitignore new file mode 100644 index 0000000..3d36647 --- a/dev/null +++ b/src/.gitignore @@ -0,0 +1 @@ +genrc diff --git a/src/Makefile.am b/src/Makefile.am new file mode 100644 index 0000000..fe24f3e --- a/dev/null +++ b/src/Makefile.am @@ -0,0 +1,32 @@ +bin_PROGRAMS = genrc +genrc_SOURCES = \ + genrc.c\ + genrc.h\ + err.c\ + pidfrom.c\ + pid_file.c\ + pid_config.c\ + pid_grep.c\ + pid_proc.c\ + pid_ps.c\ + pidlist.c\ + procscan.c\ + com_status.c\ + com_start.c\ + com_stop.c\ + com_restart.c\ + com_reload.c\ + transform.c\ + match_exact.c\ + match_glob.c\ + match_regex.c\ + sentinel.c + +AM_CPPFLAGS = @GRECS_INCLUDES@ +LDADD = @GRECS_LDADD@ + +if COND_PCRE + genrc_SOURCES += match_pcre.c + AM_CPPFLAGS += -DHAVE_PCRE=1 +endif + diff --git a/src/com_reload.c b/src/com_reload.c new file mode 100644 index 0000000..7ff1206 --- a/dev/null +++ b/src/com_reload.c @@ -0,0 +1,17 @@ +#include "genrc.h" + +int +com_reload(void) +{ + PIDLIST pids; + + if (genrc_no_reload) + return com_restart(); + pidlist_init(&pids); + if (get_pid_list(genrc_pid_closure, &pids)) + return 1; + pidlist_kill(&pids, genrc_signal_reload); + return 0; +} + + diff --git a/src/com_restart.c b/src/com_restart.c new file mode 100644 index 0000000..b3588a1 --- a/dev/null +++ b/src/com_restart.c @@ -0,0 +1,12 @@ +#include "genrc.h" + +int +com_restart(void) +{ + int rc; + rc = com_stop(); + if (rc == 0) + rc = com_start(); + return rc; +} + diff --git a/src/com_start.c b/src/com_start.c new file mode 100644 index 0000000..73bb2ec --- a/dev/null +++ b/src/com_start.c @@ -0,0 +1,121 @@ +#include "genrc.h" + +void +report_exec_error(int rc, char const *program) +{ + if (WIFEXITED(rc)) { + if (WEXITSTATUS(rc)) { + genrc_error("%s exited with status %d", + program, WEXITSTATUS(rc)); + } + } else if (WIFSIGNALED(rc)) { + char const *coremsg = ""; +#ifdef WCOREDUMP + if (WCOREDUMP(rc)) + coremsg = " (core dumped)"; +#endif + genrc_error("%s terminated on signal %d%s", + program, WTERMSIG(rc), coremsg); + } else if (WIFSTOPPED(rc)) { + genrc_error("%s stopped on signal %d", + program, WSTOPSIG(rc)); + } else { + genrc_error("%s terminated with unrecognized status: %d", + program, rc); + } +} + +typedef void (*SIGHANDLER)(int); + +void +sigchld(int sig) +{ +} + +int +timedwaitpid(pid_t pid, int *status) +{ + struct timeval now, stoptime, ttw; + int rc = -1; + SIGHANDLER oldsig; + + oldsig = signal(SIGCHLD, sigchld); + gettimeofday(&stoptime, NULL); + stoptime.tv_sec += genrc_timeout; + while (1) { + pid_t p; + + p = waitpid(pid, status, WNOHANG); + if (p == pid) { + rc = 0; + break; + } + if (p < 0 && errno != EINTR) { + system_error(errno, "waitpid"); + break; + } + + gettimeofday(&now, NULL); + if (timercmp(&now, &stoptime, >=)) + break; + timersub(&stoptime, &now, &ttw); + if (select(0, NULL, NULL, NULL, &ttw) < 0) { + if (errno != EINTR) { + system_error(errno, "select"); + break; + } + } + + } + signal(SIGCHLD, oldsig); + if (rc) { + kill(pid, SIGKILL); + } + return rc; +} + +int +com_start(void) +{ + pid_t pid; + int status; + PIDLIST pids; + char *p; + + pidlist_init(&pids); + if (get_pid_list(genrc_pid_closure, &pids) == 0) { + int running = pids.pidc > 0; + pidlist_free(&pids); + if (running) { + genrc_error("%s is already running", genrc_program); + return 1; + } + } + + if ((p = getenv("GENRC_SENTINEL")) && *p == '1') + return sentinel(); + + pid = fork(); + if (pid == -1) { + system_error(errno, "fork"); + return 1; + } + if (pid == 0) { + char *argv[] = { SHELL, "-c", NULL, NULL }; + argv[2] = genrc_command; + execvp(SHELL, argv); + system_error(errno, "failed to exec %s", genrc_program); + exit(127); + } + + if (timedwaitpid(pid, &status)) { + genrc_error("timed out waiting for %s to return", + genrc_program); + return 1; + } + if (!(WIFEXITED(status) && WEXITSTATUS(status) == 0)) { + report_exec_error(status, genrc_program); + return 1; + } + return 0; +} diff --git a/src/com_status.c b/src/com_status.c new file mode 100644 index 0000000..f68c6ef --- a/dev/null +++ b/src/com_status.c @@ -0,0 +1,38 @@ +#include "genrc.h" + +int +com_status(void) +{ + PIDLIST pids; + int rc; + + pidlist_init(&pids); + rc = get_pid_list(genrc_pid_closure, &pids); + if (rc == 0) { + if (pids.pidc == 0) + printf("%s is not running\n", genrc_program); + else { + printf("%s is running ", genrc_program); + if (pids.pidc == 1) { + printf("(pid %lu)", + (unsigned long) pids.pidv[0]); + } else { + int i; + int delim = ' '; + printf("(pids"); + for (i = 0; i < pids.pidc; i++) { + printf("%c%lu", delim, + (unsigned long) pids.pidv[i]); + delim = ','; + } + putchar(')'); + } + putchar('\n'); + } + } else { + printf("status unknown\n"); + } + return 0; +} + + diff --git a/src/com_stop.c b/src/com_stop.c new file mode 100644 index 0000000..5227358 --- a/dev/null +++ b/src/com_stop.c @@ -0,0 +1,66 @@ +#include "genrc.h" + +static void +timermul(struct timeval *a, int b) +{ + a->tv_sec *= b; + a->tv_usec *= b; + a->tv_sec = a->tv_sec + a->tv_usec / 1000000; + a->tv_usec %= 1000000; +} + +#define MIN_POLL_TTW 20000 + +int +com_stop(void) +{ + PIDLIST pids; + struct timeval stoptime, before, after, ttw, maxttw; + int ratio; + + pidlist_init(&pids); + if (get_pid_list(genrc_pid_closure, &pids)) { + genrc_error("program status unknown"); + return 1; + } + + if (pids.pidc == 0) { + genrc_error("%s not running", genrc_program); + return 1; + } + + gettimeofday(&stoptime, NULL); + stoptime.tv_sec += genrc_timeout; + ratio = 1; + while (pids.pidc) { + gettimeofday(&before, NULL); + if (timercmp(&before, &stoptime, >)) + break; + pidlist_kill(&pids, genrc_signal_stop); + if (get_pid_list(genrc_pid_closure, &pids)) + break; + if (pids.pidc == 0) + return 0; + gettimeofday(&after, NULL); + if (timercmp(&after, &stoptime, >=)) + break; + if (ratio < 10) + ++ratio; + timersub(&stoptime, &after, &maxttw); + timersub(&after, &before, &ttw); + timermul(&ttw, ratio); + if (ttw.tv_sec < 0 || ttw.tv_usec < 0) + ttw.tv_sec = ttw.tv_usec = 0; + if (timercmp(&ttw, &maxttw, >)) + ttw = maxttw; + if (ttw.tv_sec == 0 && ttw.tv_usec < MIN_POLL_TTW) + ttw.tv_usec = MIN_POLL_TTW; + if (select(0, NULL, NULL, NULL, &ttw) < 0 && errno != EINTR) { + system_error(errno, "select"); + break; + } + } + return 1; +} + + diff --git a/src/err.c b/src/err.c new file mode 100644 index 0000000..539a7b8 --- a/dev/null +++ b/src/err.c @@ -0,0 +1,47 @@ +#include "genrc.h" + +char *progname = "genrc"; + +void +setprogname(char const *s) +{ + progname = xstrdup(s); +} + +void +genrc_error(char const *fmt, ...) +{ + va_list ap; + fprintf(stderr, "%s: ", progname); + va_start(ap, fmt); + vfprintf(stderr, fmt, ap); + va_end(ap); + fputc('\n', stderr); +} + +void +usage_error(char const *fmt, ...) +{ + va_list ap; + fprintf(stderr, "%s: ", progname); + va_start(ap, fmt); + vfprintf(stderr, fmt, ap); + va_end(ap); + fputc('\n', stderr); + exit(1); +} + +void +system_error(int ec, char const *fmt, ...) +{ + va_list ap; + fprintf(stderr, "%s: ", progname); + va_start(ap, fmt); + vfprintf(stderr, fmt, ap); + va_end(ap); + fprintf(stderr, ": %s\n", strerror(ec)); +} + + + + diff --git a/src/genrc.c b/src/genrc.c new file mode 100644 index 0000000..64dd81a --- a/dev/null +++ b/src/genrc.c @@ -0,0 +1,469 @@ +#include "genrc.h" +#include <sys/ioctl.h> + +char *genrc_command; +char *genrc_program; +char *genrc_pid_from; +unsigned genrc_timeout = 5; +int genrc_no_reload; +int genrc_signal_stop = SIGTERM; +int genrc_signal_reload = SIGHUP; +GENRC_PID_CLOSURE *genrc_pid_closure; +char *genrc_pidfile; + + +enum { + OPT_USAGE = 256, + OPT_VERSION, + OPT_SIGNAL_RELOAD, + OPT_NO_RELOAD, + OPT_SIGNAL_STOP +}; + +struct option longopts[] = { + { "help", no_argument, 0, 'h' }, + { "usage", no_argument, 0, OPT_USAGE }, + { "command", required_argument, 0, 'c' }, + { "program", required_argument, 0, 'p' }, + { "pid-from", required_argument, 0, 'P' }, + { "pidfile", required_argument, 0, 'F' }, + { "timeout", required_argument, 0, 't' }, + { "signal-reload", required_argument, 0, OPT_SIGNAL_RELOAD }, + { "no-reload", no_argument, 0, OPT_NO_RELOAD }, + { "signal-stop", required_argument, 0, OPT_SIGNAL_STOP }, + { "sentinel", no_argument, 0, 'S' }, + { "version", no_argument, 0, OPT_VERSION }, + { NULL } +}; +char shortopts[] = "c:hF:P:p:St:"; + +struct sigdefn { + char const *sig_name; + int sig_no; +}; + +#define S(s) { #s, s } +static struct sigdefn sigdefn[] = { + S (SIGHUP), + S (SIGINT), + S (SIGQUIT), + S (SIGILL), + S (SIGTRAP), + S (SIGABRT), + S (SIGIOT), + S (SIGBUS), + S (SIGFPE), + S (SIGKILL), + S (SIGUSR1), + S (SIGSEGV), + S (SIGUSR2), + S (SIGPIPE), + S (SIGALRM), + S (SIGTERM), +#ifdef SIGSTKFLT + S (SIGSTKFLT), +#endif + S (SIGCHLD), + S (SIGCONT), + S (SIGSTOP), + S (SIGTSTP), + S (SIGTTIN), + S (SIGTTOU), +#ifdef SIGURG + S (SIGURG), +#endif +#ifdef SIGXCPU + S (SIGXCPU), +#endif +#ifdef SIGXFSZ + S (SIGXFSZ), +#endif +#ifdef SIGVTALRM + S (SIGVTALRM), +#endif +#ifdef SIGPROF + S (SIGPROF), +#endif +#ifdef SIGWINCH + S (SIGWINCH), +#endif +#ifdef SIGPOLL + S (SIGPOLL), +#endif +#ifdef SIGIO + S (SIGIO), +#endif +#ifdef SIGPWR + S (SIGPWR), +#endif +#ifdef SIGSYS + S (SIGSYS), +#endif + {NULL} +}; +#undef S + +static int +is_numeric_str(char const *s) +{ + while (*s) { + if (!isdigit(*s)) + return 0; + s++; + } + return 1; +} + +int +sig_name_to_str(char const *s) +{ + if (is_numeric_str(s)) { + char *end; + unsigned long n; + errno = 0; + n = strtoul(s, &end, 10); + if (errno || *end || n > UINT_MAX) + return -1; + return n; + } else { + struct sigdefn *sd; + + for (sd = sigdefn; sd->sig_name; sd++) { + if (s[0] == 's' || s[0] == 'S') { + if (strcasecmp(sd->sig_name, s) == 0) + return sd->sig_no; + } else if (strcasecmp(sd->sig_name + 3, s) == 0) + return sd->sig_no; + } + } + return -1; +} + +char const *help_msg[] = { + "Usage: genrc [OPTIONS] COMMAND", + "generic rc program\n", + "COMMANDs are:\n", + " start start the program", + " stop stop the program", + " restart restart the running program", + " reload send a reload signal to the program", + " status display the program status", + "", + "OPTIONs are:\n", + " -h, --help display this help list", + " --usage display short usage information", + " -c, --command=COMMAND command line to run", + " -p, --program=PROGRAM name of the program to run", + " -P, --pid-from=SOURCE where to look for PIDs of the running programs", + " -F, --pidfile=NAME name of the PID file", + " (same as --pid-from=FILE:NAME)", + " -t, --timeout=SECONDS time to wait for the program to start up or", + " terminate", + " --signal-reload=SIG signal to send on reload (default: SIGHUP)", + " setting to 0 is equivalent to --no-reload", + " --no-reload makes reload equivalent to restart", + " --signal-stop=SIG signal to send in order to terminate the program", + " (default: SIGTERM)", + " --sentinel PROGRAM runs in foreground; disconnect from the", + " controlling terminal, run it and act as a sentinel", + " --version display program version and exist", + "", + "Influential environment variables and corresponding options:", + "", + " GENRC_COMMAND=COMMAND --command=COMMAND", + " GENRC_PROGRAM=NAME --program=NAME", + " GENRC_PID_FROM=SOURCE --pid-from=SOURCE", + " GENRC_TIMEOUT=SECONDS --timeout=SECONDS", + " GENRC_SENTINEL=1 --sentinel", + "", + " At least one of COMMAND or PROGRAM must be given.", + " If PROGRAM is supplied, but COMMAND is not, COMMAND is set to PROGRAM", + " Otherwise, if COMMAND is set, but PROGRAM is not, PROGRAM is set to the", + " first word (in the shell sense) in COMMAND.", + "", + "PID sources:", + "", + " FILE:<NAME>", + " Read PID from the file <NAME>", + "", + " CONFIG:<LANG>:<FILENAME>:<FQRN>", + " Name of the PID file is stored in relation <FQRN> of the configuration", + " file <FILENAME>, written in language <LANG>", + "", + " GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>][;...]", + " Grep for the first line in <FILE> that matches <RX>. If found, process", + " replace the matched portion according to <REPL> and <FLAGS>. Use", + " the resulting string as PID. More sed expressions can be supplied", + " separated with semicolons.", + "", + " PROC[:[<EXE>][:<FLAGS>]]", + " Look for matching program in /proc/<PID>/*. If <EXE> is not supplied", + " or empty, program name from --program will be used. <FLAGS> are:", + " e exact match", + " g glob pattern match", + " x extended POSIX regexp match (default)", + " p PCRE match", + " i case-insensitive match", + "", + " c match entire command line", + " r match real executable name (instead of argv0)", + " a signal all matching PIDs", + "", + " PS:[:[<EXE>][:<FLAGS>]]", + " Look for matching program in the output of 'ps -ef'. <EXE> and <FLAGS>", + " as described above", + "", + NULL +}; + +void +help(void) +{ + int i; + + for (i = 0; help_msg[i]; i++) + puts(help_msg[i]); +} + +char const *usage_msg[] = { + "genrc", + "[-h]", + "[-F PIDFILE]", + "[-P SOURCE]", + "[-c COMMAND]", + "[-p PROGRAM]", + "[-t SECONDS]", + "[--command=COMMAND]", + "[--help]", + "[--no-reload]", + "[--pid-from=SOURCE]", + "[--pidfile=PIDFILE]", + "[--program=PROGRAM]", + "[--sentinel]", + "[--signal-reload=SIG]", + "[--signal-stop=SIG]", + "[--timeout=SECONDS]", + "[--usage]", + "{", + "start", + "|", + "stop", + "|", + "restart", + "|", + "reload", + "|", + "status", + "}", + NULL +}; + +static int +screen_width(void) +{ + struct winsize ws; + ws.ws_col = ws.ws_row = 0; + if ((ioctl(1, TIOCGWINSZ, (char *) &ws) < 0) || ws.ws_col == 0) { + char *p = getenv("COLUMNS"); + if (p) + ws.ws_col = atoi(p); + if (ws.ws_col == 0) + ws.ws_col = 80; + } + return ws.ws_col; +} + +void +usage(void) +{ + int i, j, pos = 0; + int width = screen_width(); + if (width > 4) + width -= 4; + for (i = j = 0; usage_msg[i]; i++) { + int len = strlen(usage_msg[i]); + if (j > 0) + len++; + if (pos + len > width) { + putchar('\n'); + pos = strlen(usage_msg[0]) + 1; + printf("%*s", pos, ""); + j = 0; + } else if (j > 0) + putchar(' '); + j++; + printf("%s", usage_msg[i]); + pos += len; + } + putchar('\n'); +} + +static const char gplv3[] = + "License GPLv3+: GNU GPL version 3 or later " + "<http://gnu.org/licenses/gpl.html>\n" + "This is free software: you are free to change and redistribute it.\n" + "There is NO WARRANTY, to the extent permitted by law.\n\n"; + +void +version(void) +{ + printf("%s\n", PACKAGE_STRING); + printf("Copyryght (C) 2018 Sergey Poznyakoff\n"); + printf("%s", gplv3); +} + + +typedef int (*GENRC_COMMAND)(void); + +struct comtab { + char const *com_name; + GENRC_COMMAND com_fun; +}; + +static struct comtab comtab[] = { + { "start", com_start }, + { "stop", com_stop }, + { "restart", com_restart }, + { "reload", com_reload }, + { "status", com_status }, + { NULL, NULL } +}; + +GENRC_COMMAND +find_command(char const *s) +{ + struct comtab *c; + for (c = comtab; c->com_name; c++) + if (strcmp(c->com_name, s) == 0) + break; + return c->com_fun; +} + +int +main(int argc, char **argv) +{ + int c; + char *p; + int no_reload = 0; + GENRC_COMMAND command; + + setprogname(argv[0]); + + while ((c = getopt_long(argc, argv, shortopts, longopts, NULL)) + != EOF) { + switch (c) { + case 'h': + help(); + exit(0); + case OPT_USAGE: + usage(); + exit(0); + case OPT_VERSION: + version(); + exit(0); + case 'c': + setenv("GENRC_COMMAND", optarg, 1); + break; + case 'p': + setenv("GENRC_PROGRAM", optarg, 1); + break; + case 'P': + setenv("GENRC_PID_FROM", optarg, 1); + break; + case 'F': + genrc_pidfile = optarg; + p = xmalloc(6 + strlen(optarg)); + strcat(strcpy(p, "FILE:"), optarg); + setenv("GENRC_PID_FROM", p, 1); + free(p); + break; + case 't': + setenv("GENRC_TIMEOUT", optarg, 1); + break; + case 'S': + setenv("GENRC_SENTINEL", "1", 1); + break; + case OPT_NO_RELOAD: + no_reload = 1; + break; + case OPT_SIGNAL_RELOAD: + setenv("GENRC_SIGNAL_RELOAD", optarg, 1); + break; + case OPT_SIGNAL_STOP: + setenv("GENRC_SIGNAL_STOP", optarg, 1); + break; + default: + exit(1); + } + } + + if ((p = getenv("GENRC_COMMAND")) != NULL) + genrc_command = p; + if ((p = getenv("GENRC_PROGRAM")) != NULL) + genrc_program = p; + + if (no_reload) + genrc_no_reload = 1; + else if ((p = getenv("GENRC_SIGNAL_RELOAD")) != NULL) { + genrc_signal_reload = sig_name_to_str(p); + if (genrc_signal_reload == -1) + usage_error("%s: invalid signal number", p); + else if (genrc_signal_reload == 0) + genrc_no_reload = 1; + } + + if ((p = getenv("GENRC_SIGNAL_STOP")) != NULL) { + genrc_signal_stop = sig_name_to_str(p); + if (genrc_signal_stop <= 0) + usage_error("%s: invalid signal number", p); + } + + if ((p = getenv("GENRC_TIMEOUT")) != NULL) { + char *end; + unsigned long n; + errno = 0; + n = strtoul(p, &end, 10); + if (errno || *p || n > UINT_MAX) + usage_error("%s: invalid timeout", p); + if (n == 0) + genrc_no_reload = 1; + else + genrc_timeout = n; + } + + if (!genrc_command) { + if (genrc_program) { + genrc_command = xstrdup(genrc_program); + } else { + usage_error("either --command (GENRC_COMMAND) or --program (GENRC_PROGRAM) must be given"); + } + } else if (!genrc_program) { + struct wordsplit ws; + ws.ws_error = genrc_error; + if (wordsplit(genrc_command, &ws, + WRDSF_NOCMD|WRDSF_NOVAR| + WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR)) + exit(1); + genrc_program = xstrdup(ws.ws_wordv[0]); + wordsplit_free(&ws); + } + + if ((p = getenv("GENRC_PID_FROM")) != NULL) { + genrc_pid_closure = get_pid_closure(p); + } else { + genrc_pid_closure = get_pid_closure("PROC"); + } + + argc -= optind; + if (argc == 0) + usage_error("missing command verb"); + if (argc > 1) + usage_error("too many arguments"); + argv += optind; + + command = find_command(argv[0]); + if (!command) + usage_error("unrecognized command: %s", argv[0]); + exit(command()); +} + diff --git a/src/genrc.h b/src/genrc.h new file mode 100644 index 0000000..f782fd5 --- a/dev/null +++ b/src/genrc.h @@ -0,0 +1,137 @@ +#include <config.h> +#include <stdlib.h> +#include <stdio.h> +#include <unistd.h> +#include <getopt.h> +#include <signal.h> +#include <limits.h> +#include <errno.h> +#include <string.h> +#include <regex.h> +#include <ctype.h> +#include <sys/time.h> +#include <sys/select.h> +#include <sys/wait.h> +#include "grecs.h" +#include "wordsplit.h" + +void setprogname(char const *s); +void genrc_error(char const *fmt, ...); +void usage_error(char const *fmt, ...); +void system_error(int ec, char const *fmt, ...); + +#define xmalloc grecs_malloc +#define xzalloc grecs_zalloc +#define xcalloc grecs_calloc +#define xrealloc grecs_realloc +#define xstrdup grecs_strdup + +#define SHELL "/bin/sh" + +pid_t file_read_pid(char const *filename); + +typedef struct transform *TRANSFORM; +char *transform_string(TRANSFORM tf, const char *input); +char *transform_string_if_match(TRANSFORM tf, const char *input); +TRANSFORM compile_transform_expr(const char *expr, int cflags); + +struct pidlist { + pid_t *pidv; + size_t pidc; + size_t pidn; +}; +typedef struct pidlist PIDLIST; + +void pidlist_init(PIDLIST *); +void pidlist_free(PIDLIST *); +void pidlist_clear(PIDLIST *plist); +void pidlist_add(PIDLIST *, pid_t); +ssize_t pidlist_index(PIDLIST *plist, pid_t p); +int pidlist_member(PIDLIST *plist, pid_t p); +int pidlist_remove(PIDLIST *plist, size_t i); +void pidlist_kill(PIDLIST *plist, int sig); + +pid_t strtopid(char const *str); + +int pid_is_running(pid_t pid); + + +enum { + MATCH_REGEX, /* extended POSIX regexp match (default) */ + MATCH_PCRE, /* PCRE match (not implemented) */ + MATCH_GLOB, /* glob pattern match */ + MATCH_EXACT, /* exact match */ +}; +#define MATCH_DEFAULT MATCH_REGEX + +enum { + PROCF_ICASE = 0x01, /* ignore case */ + PROCF_ALL = 0x02, /* use all pids */ + PROCF_CMDLINE = 0x04, /* match against entire command line */ + PROCF_EXE = 0x08 /* match against executable name */ +}; +#define PROCF_DEFAULT 0 + +struct procscanbuf { + int match; /* match type */ + int flags; /* match flags */ + void *pattern; +}; + +typedef struct procscanbuf *PROCSCANBUF; + +PROCSCANBUF procscan_init(char const *pattern, char const *flagstr); +void procscan_free(PROCSCANBUF buf); +int procscan_match(PROCSCANBUF buf, char const *arg); + +void match_exact_init(PROCSCANBUF buf, char const *pattern); +void match_exact_free(PROCSCANBUF buf); +int match_exact(PROCSCANBUF buf, char const *arg); + +void match_glob_init(PROCSCANBUF buf, char const *pattern); +void match_glob_free(PROCSCANBUF buf); +int match_glob(PROCSCANBUF buf, char const *arg); + +void match_regex_init(PROCSCANBUF buf, char const *pattern); +void match_regex_free(PROCSCANBUF buf); +int match_regex(PROCSCANBUF buf, char const *arg); + +void match_pcre_init(PROCSCANBUF buf, char const *pattern); +void match_pcre_free(PROCSCANBUF buf); +int match_pcre(PROCSCANBUF buf, char const *arg); + + + +struct genrc_pid_closure { + char const *name; + int (*pid)(struct genrc_pid_closure *, PIDLIST *); +}; + +typedef struct genrc_pid_closure GENRC_PID_CLOSURE; +GENRC_PID_CLOSURE *get_pid_closure(char const *str); +int get_pid_list(GENRC_PID_CLOSURE *, PIDLIST *); + +GENRC_PID_CLOSURE *genrc_pid_file_init(int argc, char **argv); +GENRC_PID_CLOSURE *genrc_pid_config_init(int argc, char **argv); +GENRC_PID_CLOSURE *genrc_pid_grep_init(int argc, char **argv); +GENRC_PID_CLOSURE *genrc_pid_proc_init(int argc, char **argv); +GENRC_PID_CLOSURE *genrc_pid_ps_init(int argc, char **argv); + +extern char *genrc_command; +extern char *genrc_program; +extern char *genrc_pid_from; +extern unsigned genrc_timeout; +extern int genrc_no_reload; +extern int genrc_signal_reload; +extern int genrc_signal_stop; +extern GENRC_PID_CLOSURE *genrc_pid_closure; +extern char *genrc_pidfile; + +int sentinel(void); + +int com_start(void); +int com_status(void); +int com_stop(void); +int com_restart(void); +int com_reload(void); + diff --git a/src/match_exact.c b/src/match_exact.c new file mode 100644 index 0000000..49be17f --- a/dev/null +++ b/src/match_exact.c @@ -0,0 +1,20 @@ +#include "genrc.h" + +void +match_exact_init(PROCSCANBUF buf, char const *pattern) +{ + buf->pattern = xstrdup(pattern ? pattern : genrc_program); +} + +void +match_exact_free(PROCSCANBUF buf) +{ + free(buf->pattern); +} + +int +match_exact(PROCSCANBUF buf, char const *arg) +{ + return ((buf->flags & PROCF_ICASE) ? strcasecmp : strcmp) + ((char*)buf->pattern, arg); +} diff --git a/src/match_glob.c b/src/match_glob.c new file mode 100644 index 0000000..d2dfd71 --- a/dev/null +++ b/src/match_glob.c @@ -0,0 +1,22 @@ +#include "genrc.h" +#define _GNU_SOURCE 1 +#include <fnmatch.h> + +void +match_glob_init(PROCSCANBUF buf, char const *pattern) +{ + buf->pattern = xstrdup(pattern ? pattern : genrc_program); +} + +void +match_glob_free(PROCSCANBUF buf) +{ + free(buf->pattern); +} + +int +match_glob(PROCSCANBUF buf, char const *arg) +{ + return fnmatch((char*)buf->pattern, arg, + (buf->flags & PROCF_ICASE) ? FNM_CASEFOLD : 0); +} diff --git a/src/match_pcre.c b/src/match_pcre.c new file mode 100644 index 0000000..55c4714 --- a/dev/null +++ b/src/match_pcre.c @@ -0,0 +1,60 @@ +#include "genrc.h" +#include <pcre.h> + +static char * +progname_pattern(char const *prog) +{ + grecs_txtacc_t acc; + char *ret; + static char const specials[] = "\\[]?.*+(){}|"; + + acc = grecs_txtacc_create(); + grecs_txtacc_grow_char(acc, '^'); + if (prog[0] != '/') + grecs_txtacc_grow_string(acc, "(?:/(?:.*/)?)?"); + for (; *prog; prog++) { + if (strchr(specials, *prog)) + grecs_txtacc_grow_char(acc, '\\'); + grecs_txtacc_grow_char(acc, *prog); + } + grecs_txtacc_grow_char(acc, '$'); + grecs_txtacc_grow_char(acc, 0); + ret = grecs_txtacc_finish(acc, 1); + grecs_txtacc_free(acc); + return ret; +} + +void +match_pcre_init(PROCSCANBUF buf, char const *pattern) +{ + pcre *pcre; + char const *error; + int error_offset; + char *tmp = NULL; + int cflags = 0; + if (buf->flags & PROCF_ICASE) + cflags |= PCRE_CASELESS; + if (!pattern) { + pattern = tmp = progname_pattern(genrc_program); + } + pcre = pcre_compile(pattern, cflags, &error, &error_offset, 0); + free(tmp); + if (!pcre) + usage_error("%s at offset %d", error, error_offset); + buf->pattern = pcre; +} + +void +match_pcre_free(PROCSCANBUF buf) +{ + pcre_free ((pcre*)buf->pattern); +} + +int +match_pcre(PROCSCANBUF buf, char const *arg) +{ + int rc = pcre_exec((pcre*)buf->pattern, 0, arg, strlen(arg), + 0, 0, NULL, 0); + return !(rc >= 0); + /* FIXME: error message unless rc == PCRE_ERROR_NOMATCH */ +} diff --git a/src/match_regex.c b/src/match_regex.c new file mode 100644 index 0000000..fc666c9 --- a/dev/null +++ b/src/match_regex.c @@ -0,0 +1,61 @@ +#include "genrc.h" + +static char * +progname_pattern(char const *prog) +{ + grecs_txtacc_t acc; + char *ret; + static char const specials[] = "\\[]?.*+(){}|"; + + acc = grecs_txtacc_create(); + grecs_txtacc_grow_char(acc, '^'); + if (prog[0] != '/') + grecs_txtacc_grow_string(acc, "(/(.*/)?)?"); + for (; *prog; prog++) { + if (strchr(specials, *prog)) + grecs_txtacc_grow_char(acc, '\\'); + grecs_txtacc_grow_char(acc, *prog); + } + grecs_txtacc_grow_char(acc, '$'); + grecs_txtacc_grow_char(acc, 0); + ret = grecs_txtacc_finish(acc, 1); + grecs_txtacc_free(acc); + return ret; +} + +void +match_regex_init(PROCSCANBUF buf, char const *pattern) +{ + regex_t *rx = xmalloc(sizeof(*rx)); + int rc; + char *tmp = NULL; + int cflags = REG_EXTENDED|REG_NOSUB; + if (buf->flags & PROCF_ICASE) + cflags |= REG_ICASE; + if (!pattern) { + pattern = tmp = progname_pattern(genrc_program); + } + + rc = regcomp(rx, pattern, cflags); + free(tmp); + if (rc) { + char errbuf[512]; + regerror(rc, rx, errbuf, sizeof(errbuf)); + usage_error("invalid regular expression: %s", errbuf); + } + buf->pattern = rx; +} + +void +match_regex_free(PROCSCANBUF buf) +{ + regfree((regex_t*)buf->pattern); + free(buf->pattern); +} + +int +match_regex(PROCSCANBUF buf, char const *arg) +{ + return regexec((regex_t*)buf->pattern, arg, 0, NULL, 0); +} + diff --git a/src/pid_config.c b/src/pid_config.c new file mode 100644 index 0000000..b646cc0 --- a/dev/null +++ b/src/pid_config.c @@ -0,0 +1,91 @@ +#include "genrc.h" + +typedef struct grecs_node *(*GRECS_PARSER_FUNC)(const char *, int); + +struct config_pid_closure { + struct genrc_pid_closure generic; + char *filename; + char *fqrn; + GRECS_PARSER_FUNC parser; +}; + +static int +pid_config_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + struct config_pid_closure *config_clos = + (struct config_pid_closure *)clos; + struct grecs_node *tree, *node; + pid_t pid; + + grecs_parser_fun = config_clos->parser; + tree = grecs_parse(config_clos->filename); + if (!tree) + exit(1); + node = grecs_find_node(tree, config_clos->fqrn); + if (!node) + usage_error("%s: relation %s not found", + config_clos->filename, + config_clos->fqrn); + if (node->type != grecs_node_stmt) + usage_error("%s: relation %s is not a statement", + config_clos->filename, + config_clos->fqrn); + if (node->v.value->type != GRECS_TYPE_STRING) + usage_error("%s: relation %s is not a string", + config_clos->filename, + config_clos->fqrn); + + pid = file_read_pid(node->v.value->v.string); + grecs_tree_free(tree); + + if (pid == -1) + return -1; + if (pid > 0) + pidlist_add(plist, pid); + + return 0; +} + +struct langtab { + char const *lang_name; + GRECS_PARSER_FUNC lang_parser; +}; + +static struct langtab langtab[] = { + { "GRECS", grecs_grecs_parser }, + { "META1", grecs_meta1_parser }, + { "BIND", grecs_bind_parser }, + { "DHCPD", grecs_dhcpd_parser }, + { "GIT", grecs_git_parser }, + { "PATH", grecs_path_parser }, + { NULL, NULL } +}; + +static GRECS_PARSER_FUNC +find_parser(const char *lang) +{ + struct langtab *p; + for (p = langtab; p->lang_name; p++) + if (strcmp(p->lang_name, lang) == 0) + break; + return p->lang_parser; +} + +GENRC_PID_CLOSURE * +genrc_pid_config_init(int argc, char **argv) +{ + struct config_pid_closure *clos; + + if (argc != 4) + usage_error("expected format: CONFIG:<LANG>:<FILE>:<FQRN>"); + + clos = xmalloc(sizeof(*clos)); + clos->generic.pid = pid_config_get; + clos->parser = find_parser(argv[1]); + if (!clos->parser) + usage_error("%s: unsupported parser type", argv[1]); + clos->filename = xstrdup(argv[2]); + clos->fqrn = xstrdup(argv[3]); + return (GENRC_PID_CLOSURE *) clos; +} + diff --git a/src/pid_file.c b/src/pid_file.c new file mode 100644 index 0000000..344de12 --- a/dev/null +++ b/src/pid_file.c @@ -0,0 +1,70 @@ +#include "genrc.h" + +struct file_pid_closure { + struct genrc_pid_closure generic; + char *filename; +}; + +pid_t +file_read_pid(char const *filename) +{ + FILE *fp; + char buf[256]; + int n, ec, ok; + unsigned long pid; + char *end; + + fp = fopen(filename, "r"); + if (!fp) { + if (errno == ENOENT) + return 0; + system_error(errno, "can't open file %s", filename); + return -1; + } + ok = fgets(buf, sizeof(buf), fp) != NULL; + ec = errno; + fclose(fp); + if (!ok) { + system_error(ec, "error reading from %s", filename); + return -1; + } + n = strlen(buf); + if (n == 0) { + genrc_error("invalid line read from %s", filename); + return -1; + } + if (buf[n-1] == '\n') + buf[n-1] = 0; + errno = 0; + pid = strtoul(buf, &end, 10); + if (errno || *end) { + genrc_error("invalid line read from %s", filename); + return -1; + } + return pid_is_running(pid) ? pid : 0; +} + +static int +pid_file_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + struct file_pid_closure *file_clos = (struct file_pid_closure *)clos; + pid_t pid = file_read_pid(file_clos->filename); + if (pid == -1) + return -1; + if (pid > 0) + pidlist_add(plist, pid); + return 0; +} + +GENRC_PID_CLOSURE * +genrc_pid_file_init(int argc, char **argv) +{ + struct file_pid_closure *clos; + + if (argc != 2) + usage_error("%s: expected format: FILE:<FILENAME>", argv[0]); + clos = xmalloc(sizeof(*clos)); + clos->generic.pid = pid_file_get; + clos->filename = xstrdup(argv[1]); + return (GENRC_PID_CLOSURE *)clos; +} diff --git a/src/pid_grep.c b/src/pid_grep.c new file mode 100644 index 0000000..8869a2d --- a/dev/null +++ b/src/pid_grep.c @@ -0,0 +1,80 @@ +#include "genrc.h" + +/* +GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>] + +Grep for the first line in <FILE> that matches <RX>. If found, process the +string using the SED expression and return the result. +*/ + +struct grep_pid_closure { + struct genrc_pid_closure generic; + char *filename; + TRANSFORM tf; +}; + +static int +pid_grep_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + struct grep_pid_closure *grep_clos = (struct grep_pid_closure *)clos; + FILE *fp; + char *buf = NULL; + size_t size = 0; + ssize_t n; + char *res = NULL; + pid_t pid; + + fp = fopen(grep_clos->filename, "r"); + if (!fp) { + if (errno == ENOENT) + return 0; + system_error(errno, "can't open file %s", grep_clos->filename); + return -1; + } + + while ((n = grecs_getline(&buf, &size, fp)) > 0) { + buf[--n] = 0; + if (n == 0) + continue; + res = transform_string_if_match(grep_clos->tf, buf); + if (res) + break; + } + free(buf); + fclose(fp); + + if (!res) { + usage_error("%s: no match for %s", + grep_clos->filename, + grep_clos->generic.name); + return -1; + } + + pid = file_read_pid(res); + free(res); + + if (pid == -1) + return -1; + if (pid > 0) + pidlist_add(plist, pid); + + return 0; +} + +GENRC_PID_CLOSURE * +genrc_pid_grep_init(int argc, char **argv) +{ + struct grep_pid_closure *clos; + + if (argc != 3) + usage_error("expected format: GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>];..."); + + clos = xmalloc(sizeof(*clos)); + clos->generic.pid = pid_grep_get; + clos->filename = xstrdup(argv[1]); + clos->tf = compile_transform_expr(argv[2], REG_EXTENDED); + + return (GENRC_PID_CLOSURE *) clos; +} + + diff --git a/src/pid_proc.c b/src/pid_proc.c new file mode 100644 index 0000000..e460dff --- a/dev/null +++ b/src/pid_proc.c @@ -0,0 +1,335 @@ +#include "genrc.h" +#include <glob.h> +#include <sys/stat.h> + + +char * +catfile(char const *dir, char const *file) +{ + char *ret; + size_t dlen, flen; + + dlen = strlen(dir); + if (dlen > 0 && dir[dlen-1] == '/') + dlen--; + flen = strlen(file); + ret = xmalloc(dlen + flen + 2); + memcpy(ret, dir, dlen); + ret[dlen] = '/'; + strcpy(ret + dlen + 1, file); + return ret; +} + +#define INITIAL_READLINK_SIZE 128 + +static int +areadlink(const char *name, char **pbuf) +{ + size_t status = 0; + char *buf; + size_t size; + ssize_t linklen; + + size = INITIAL_READLINK_SIZE; + buf = malloc(size); + if (!buf) + return -1; + while (1) { + char *p; + size_t newsize = size << 1; + if (newsize < size) { + status = ENAMETOOLONG; + break; + } + size = newsize; + p = realloc(buf, size); + if (!p) + free(buf); + buf = p; + if (!buf) { + status = ENOMEM; + break; + } + + linklen = readlink(name, buf, size); + if (linklen < 0 && errno != ERANGE) { + status = errno; + break; + } + + if ((size_t) linklen < size) { + buf[linklen++] = '\0'; + status = 0; + break; + } + } + + if (status) { + free(buf); + errno = status; + return -1; + } + *pbuf = buf; + return 0; +} + +pid_t +strtopid(char const *str) +{ + unsigned long n; + char *end; + errno = 0; + n = strtoul(str, &end, 10); + if (errno || *end) + return -1; + return (pid_t) n; +} + +struct proc_pid_closure { + struct genrc_pid_closure generic; + PROCSCANBUF buf; +}; + +pid_t +parent_pid(pid_t p) +{ + FILE *fp; + char *filename = NULL; + size_t s = 0; + char *line = NULL; + ssize_t n; + struct wordsplit ws; + pid_t pid; + + grecs_asprintf(&filename, &s, "/proc/%lu/stat", (unsigned long)p); + fp = fopen(filename, "r"); + if (!fp) { + system_error(errno, "can't open file %s", filename); + free(filename); + return -1; + } + s = 0; + n = grecs_getline(&line, &s, fp); + fclose(fp); + if (n == -1) { + system_error(errno, "error reading from %s", filename); + free(filename); + return -1; + } + free(filename); + if (n == 0) + return 0; + + ws.ws_delim = " "; + ws.ws_error = genrc_error; + if (wordsplit(line, &ws, + WRDSF_NOCMD|WRDSF_NOVAR|WRDSF_QUOTE| + WRDSF_DELIM|WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR)) + exit(1); + pid = strtopid(ws.ws_wordv[3]); + wordsplit_free(&ws); + return pid; +} + +static void +parent_pids_filter(PIDLIST *plist) +{ + int more; + size_t i; + + do { + more = 0; + for (i = 0; i < plist->pidc; ) { + pid_t ppid = parent_pid(plist->pidv[i]); + if (ppid > 1 && pidlist_member(plist, ppid)) { + pidlist_remove(plist, i); + more = 1; + } else + i++; + } + } while (more); +} + +static int +match_exe(struct proc_pid_closure *clos, char const *dir) +{ + char *fname = catfile(dir, "exe"); + char *exename; + struct stat st; + int rc = 0; + + if (lstat(fname, &st)) { + if (errno != ENOENT) + system_error(errno, "stat %s", fname); + rc = -1; + } else { + if (!S_ISLNK(st.st_mode)) { + rc = -1; + } else if (areadlink(fname, &exename)) { + if (errno != ENOENT) + system_error(errno, "readlink %s", fname); + rc = -1; + } + } + free(fname); + if (rc == 0) { + rc = procscan_match(clos->buf, exename); + free(exename); + } + return rc; +} + +static int +match_cmdline(struct proc_pid_closure *clos, char const *dir) +{ + char *fname = catfile(dir, "cmdline"); + FILE *fp; + int c; + int nul = 0; + grecs_txtacc_t acc; + int rc; + char *cmdline; + + fp = fopen(fname, "r"); + if (!fp) { + system_error(errno, "fopen %s", fname); + free(fname); + return -1; + } + acc = grecs_txtacc_create(); + + while ((c = fgetc(fp)) != EOF) { + if (nul) { + grecs_txtacc_grow_char(acc, ' '); + nul = 0; + } + if (c == 0) + nul = 1; + else + grecs_txtacc_grow_char(acc, c); + } + fclose(fp); + free(fname); + + grecs_txtacc_grow_char(acc, 0); + cmdline = grecs_txtacc_finish(acc, 0); + + rc = procscan_match(clos->buf, cmdline); + grecs_txtacc_free(acc); + return rc; +} + +static int +match_argv0(struct proc_pid_closure *clos, char const *dir) +{ + char *fname = catfile(dir, "cmdline"); + FILE *fp; + int c; + grecs_txtacc_t acc; + int rc; + char *argv0; + + fp = fopen(fname, "r"); + if (!fp) { + system_error(errno, "fopen %s", fname); + free(fname); + return -1; + } + acc = grecs_txtacc_create(); + + while ((c = fgetc(fp)) != EOF) { + grecs_txtacc_grow_char(acc, c); + if (c == 0) + break; + } + fclose(fp); + free(fname); + + argv0 = grecs_txtacc_finish(acc, 0); + + rc = procscan_match(clos->buf, argv0); + grecs_txtacc_free(acc); + return rc; +} + +static void +scandir(struct proc_pid_closure *clos, char const *dir, PIDLIST *plist) +{ + int rc; + struct stat st; + + if (stat(dir, &st)) { + if (errno != ENOENT) + system_error(errno, "stat %s", dir); + return; + } + if (!S_ISDIR(st.st_mode)) + return; + + if (clos->buf->flags & PROCF_CMDLINE) { + rc = match_cmdline(clos, dir); + } else if (clos->buf->flags & PROCF_EXE) { + rc = match_exe(clos, dir); + } else { + rc = match_argv0(clos, dir); + } + + if (rc == 0) { + pidlist_add(plist, strtopid(dir + sizeof("/proc"))); + } +} + + +static int +globerr(const char *epath, int ec) +{ + system_error(ec, "%s", epath); + return 0; +} + +static int +pid_proc_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + struct proc_pid_closure *proc_clos = (struct proc_pid_closure *)clos; + glob_t gl; + char const *dirpat = "/proc/[0-9]*"; + int i; + + switch (glob(dirpat, GLOB_NOSORT|GLOB_ONLYDIR, globerr, &gl)) { + case GLOB_NOSPACE: + grecs_alloc_die(); + case GLOB_ABORTED: + genrc_error("read error during glob scan"); + exit(1); + case GLOB_NOMATCH: + return 0; + } + + for (i = 0; i < gl.gl_pathc; i++) { + scandir(proc_clos, gl.gl_pathv[i], plist); + } + globfree(&gl); + + if (plist->pidc > 1 && !(proc_clos->buf->flags & PROCF_ALL)) + parent_pids_filter(plist); + + return 0; +} + +GENRC_PID_CLOSURE * +genrc_pid_proc_init(int argc, char **argv) +{ + struct proc_pid_closure *clos; + + if (argc > 3) + usage_error("expected format: PROC[:[<EXE>][:<FLAGS>]]"); + + clos = xzalloc(sizeof *clos); + + clos->buf = procscan_init((argc >= 2 && argv[1][0] != 0) + ? argv[1] : NULL, + argc > 2 ? argv[2] : NULL); + clos->generic.pid = pid_proc_get; + + return (GENRC_PID_CLOSURE *)clos; +} diff --git a/src/pid_ps.c b/src/pid_ps.c new file mode 100644 index 0000000..573304d --- a/dev/null +++ b/src/pid_ps.c @@ -0,0 +1,137 @@ +#include "genrc.h" + +struct ps_pid_closure { + struct genrc_pid_closure generic; + PROCSCANBUF buf; +}; + +enum { + UID_COL = 0, + PID_COL, + PPID_COL, + C_COL, + STIME_COL, + TTY_COL, + TIME_COL, + CMD_COL +}; + +static int +match_cmdline(struct ps_pid_closure *clos, struct wordsplit const *ws) +{ + grecs_txtacc_t acc; + size_t i; + int rc; + char *cmdline; + + acc = grecs_txtacc_create(); + grecs_txtacc_grow_string(acc, ws->ws_wordv[CMD_COL]); + for (i = CMD_COL + 1; i < ws->ws_wordc; i++) { + grecs_txtacc_grow_char(acc, ' '); + grecs_txtacc_grow_string(acc, ws->ws_wordv[i]); + } + grecs_txtacc_grow_char(acc, 0); + cmdline = grecs_txtacc_finish(acc, 0); + rc = procscan_match(clos->buf, cmdline); + grecs_txtacc_free(acc); + return rc; +} + +static int +match_argv0(struct ps_pid_closure *clos, struct wordsplit const *ws) +{ + return procscan_match(clos->buf, ws->ws_wordv[CMD_COL]); +} + +static int +match(struct ps_pid_closure *clos, struct wordsplit const *ws) +{ + return ((clos->buf->flags & PROCF_CMDLINE) + ? match_cmdline : match_argv0)(clos, ws); +} + +static int +pid_ps_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + struct ps_pid_closure *ps_clos = (struct ps_pid_closure *)clos; + FILE *fp; + struct wordsplit ws; + int wsflags; + ssize_t n; + char *buf = NULL; + size_t size = 0; + size_t line = 0; + PIDLIST pplist; + + fp = popen("ps -ef", "r"); + if (!fp) { + system_error(errno, "failed to run ps"); + return -1; + } + + ws.ws_delim = " "; + ws.ws_error = genrc_error; + wsflags = WRDSF_NOCMD + | WRDSF_NOVAR + | WRDSF_QUOTE + | WRDSF_DELIM + | WRDSF_SQUEEZE_DELIMS + | WRDSF_ENOMEMABRT + | WRDSF_SHOWERR + | WRDSF_ERROR; + pidlist_init(&pplist); + while ((n = grecs_getline(&buf, &size, fp)) > 0) { + line++; + if (line == 1) + continue; + buf[n-1] = 0; + if (wordsplit(buf, &ws, wsflags)) + exit(1); + wsflags |= WRDSF_REUSE; + if (match(ps_clos, &ws) == 0) { + pid_t ppid = strtopid(ws.ws_wordv[PPID_COL]); + if (!pidlist_member(&pplist, ppid)) + pidlist_add(&pplist, ppid); + pidlist_add(plist, strtopid(ws.ws_wordv[PID_COL])); + } + } + fclose(fp); + free(buf); + if (wsflags & WRDSF_REUSE) + wordsplit_free(&ws); + if (plist->pidc > 1 && !(ps_clos->buf->flags & PROCF_ALL)) { + size_t i; + + for (i = 0; i < pplist.pidc; ) { + if (pidlist_member(plist, pplist.pidv[i])) + i++; + else { + pidlist_remove(&pplist, i); + } + } + pidlist_free(plist); + *plist = pplist; + } + + return 0; +} + +GENRC_PID_CLOSURE * +genrc_pid_ps_init(int argc, char **argv) +{ + struct ps_pid_closure *clos; + + if (argc > 3) + usage_error("expected format: PS[:[<EXE>][:<FLAGS>]]"); + + clos = xzalloc(sizeof *clos); + + clos->buf = procscan_init((argc >= 2 && argv[1][0] != 0) + ? argv[1] : genrc_program, + argc > 2 ? argv[2] : NULL); + if (clos->buf->flags & PROCF_EXE) + usage_error("flag r isn't compatible with PS"); + clos->generic.pid = pid_ps_get; + + return (GENRC_PID_CLOSURE *)clos; +} diff --git a/src/pidfrom.c b/src/pidfrom.c new file mode 100644 index 0000000..ab1f623 --- a/dev/null +++ b/src/pidfrom.c @@ -0,0 +1,52 @@ +#include "genrc.h" + +struct genrc_pid_source { + char *name; + GENRC_PID_CLOSURE *(*init)(int argc, char **argv); +}; + +struct genrc_pid_source sourcetab[] = { + { "FILE", genrc_pid_file_init }, + { "CONFIG", genrc_pid_config_init }, + { "GREP", genrc_pid_grep_init }, + { "PROC", genrc_pid_proc_init }, + { "PS", genrc_pid_ps_init }, + { NULL } +}; + +GENRC_PID_CLOSURE * +get_pid_closure(char const *str) +{ + struct genrc_pid_source *s; + struct wordsplit ws; + GENRC_PID_CLOSURE *pc; + + ws.ws_error = genrc_error; + ws.ws_delim = ":"; + if (wordsplit(str, &ws, + WRDSF_NOCMD|WRDSF_NOVAR|WRDSF_QUOTE| + WRDSF_DELIM|WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR)) + exit(1); + + if (ws.ws_wordc == 0) + usage_error("GENRC_PID_FROM argument is empty"); + for (s = sourcetab; ; s++) { + if (!s->name) + usage_error("%s: unsupported PID source", str); + if (strcmp(s->name, ws.ws_wordv[0]) == 0) + break; + } + pc = s->init(ws.ws_wordc, ws.ws_wordv); + if (!pc) + exit(1); + pc->name = xstrdup(str); + return pc; +} + +int +get_pid_list(GENRC_PID_CLOSURE *clos, PIDLIST *plist) +{ + pidlist_clear(plist); + return clos->pid(clos, plist); +} + diff --git a/src/pidlist.c b/src/pidlist.c new file mode 100644 index 0000000..cf47c2e --- a/dev/null +++ b/src/pidlist.c @@ -0,0 +1,121 @@ +#include "genrc.h" + +void * +x2nrealloc(void *p, size_t *pn, size_t s) +{ + size_t n = *pn; + + if (!p) { + if (!n) { + /* The approximate size to use for initial small + allocation requests, when the invoking code + specifies an old size of zero. 64 bytes is + the largest "small" request for the + GNU C library malloc. */ + enum { DEFAULT_MXFAST = 64 }; + + n = DEFAULT_MXFAST / s; + n += !n; + } + } else { + /* Set N = ceil (1.5 * N) so that progress is made if N == 1. + Check for overflow, so that N * S stays in size_t range. + The check is slightly conservative, but an exact check isn't + worth the trouble. */ + if ((size_t) -1 / 3 * 2 / s <= n) + grecs_alloc_die(); + n += (n + 1) / 2; + } + + *pn = n; + return grecs_realloc(p, n * s); +} + +void +pidlist_init(PIDLIST *plist) +{ + plist->pidv = NULL; + plist->pidc = 0; + plist->pidn = 0; +} + +void +pidlist_free(PIDLIST *plist) +{ + free(plist->pidv); +} + +void +pidlist_clear(PIDLIST *plist) +{ + plist->pidc = 0; +} + +void +pidlist_add(PIDLIST *plist, pid_t p) +{ + if (p <= 0) + return; + if (plist->pidc == plist->pidn) { + plist->pidv = x2nrealloc(plist->pidv, &plist->pidn, + sizeof(plist->pidv[0])); + } + plist->pidv[plist->pidc++] = p; +} + +ssize_t +pidlist_index(PIDLIST *plist, pid_t p) +{ + size_t i; + for (i = 0; i < plist->pidc; i++) { + if (plist->pidv[i] == p) + return i; + } + return -1; +} + +int +pidlist_member(PIDLIST *plist, pid_t p) +{ + return pidlist_index(plist, p) != -1; +} + +int +pidlist_remove(PIDLIST *plist, size_t i) +{ + if (i == -1) + return 0; + + if (i < plist->pidc - 1) { + memmove(plist->pidv + i, plist->pidv + i + 1, + (plist->pidc - i - 1) * sizeof(plist->pidv[0])); + } + plist->pidc--; + return 1; +} + +void +pidlist_kill(PIDLIST *plist, int sig) +{ + size_t i; + + for (i = 0; i < plist->pidc; i++) { + if (kill(plist->pidv[i], sig)) { + if (errno != ESRCH) + system_error(errno, "kill %lu", + (unsigned long) plist->pidv[i]); + } + } +} + +int +pid_is_running(pid_t pid) +{ + if (kill(pid, 0)) { + if (errno == ESRCH) + return 0; + system_error(errno, "kill %lu", (unsigned long)pid); + exit(1); + } + return 1; +} diff --git a/src/procscan.c b/src/procscan.c new file mode 100644 index 0000000..1425f6a --- a/dev/null +++ b/src/procscan.c @@ -0,0 +1,95 @@ +#include "genrc.h" + +struct flagdfn { + int letter; /* flag letter */ + int is_match; /* is this flag a match mode */ + int value; /* flag value */ +}; + +static struct flagdfn flagtab[] = { + { 'e', 1, MATCH_EXACT }, + { 'g', 1, MATCH_GLOB }, + { 'x', 1, MATCH_REGEX }, + { 'p', 1, MATCH_PCRE }, + { 'i', 0, PROCF_ICASE }, + { 'a', 0, PROCF_ALL }, + { 'c', 0, PROCF_CMDLINE }, + { 'r', 0, PROCF_EXE }, + { 0 } +}; + +static struct flagdfn * +find_flag(int let) +{ + struct flagdfn *p; + + let = tolower(let); + for (p = flagtab; p->letter; p++) + if (p->letter == let) + return p; + return NULL; +} + +struct match_type { + void (*match_init)(PROCSCANBUF, char const *); + void (*match_free)(PROCSCANBUF); + int (*match_func)(PROCSCANBUF, char const *); +}; + +struct match_type match_types[] = { + [MATCH_REGEX] = { match_regex_init, match_regex_free, match_regex }, +#ifdef HAVE_PCRE + [MATCH_PCRE] = { match_pcre_init, match_pcre_free, match_pcre }, +#endif + [MATCH_GLOB] = { match_glob_init, match_glob_free, match_glob }, + [MATCH_EXACT] = { match_exact_init, match_exact_free, match_exact } +}; + +PROCSCANBUF +procscan_init(char const *pattern, char const *flagstr) +{ + PROCSCANBUF buf; + + buf = xmalloc(sizeof(buf)); + buf->match = MATCH_DEFAULT; + buf->flags = PROCF_DEFAULT; + if (flagstr) { + char const *p; + for (p = flagstr; *p; p++) { + struct flagdfn *flag = find_flag(*p); + if (!flag) + usage_error("%s: unknown flag %c", + flagstr, *p); + if (flag->is_match) { + if (match_types[flag->value].match_init) + buf->match = flag->value; + else + usage_error("match type '%c' is not available", *p); + } + else if (isupper(*p)) + buf->flags &= ~flag->value; + else + buf->flags |= flag->value; + } + } + match_types[buf->match].match_init(buf, pattern); + return buf; +} + +void +procscan_free(PROCSCANBUF buf) +{ + match_types[buf->match].match_free(buf); + free(buf); +} + +int +procscan_match(PROCSCANBUF buf, char const *arg) +{ + if (!arg) + return 1; + return match_types[buf->match].match_func(buf, arg); +} + + + diff --git a/src/sentinel.c b/src/sentinel.c new file mode 100644 index 0000000..0fe0714 --- a/dev/null +++ b/src/sentinel.c @@ -0,0 +1,207 @@ +#include "genrc.h" +#include <unistd.h> +#include <fcntl.h> +#include <syslog.h> + +static void +xpipe(int p[2]) +{ + if (pipe(p)) { + system_error(errno, "pipe"); + exit(1); + } +} + +#define max(a,b) ((a) > (b) ? (a) : (b)) + +static void +write_pid_file(pid_t pid) +{ + if (genrc_pidfile) { + FILE *fp; + fp = fopen(genrc_pidfile, "w"); + fprintf(fp, "%lu\n", (unsigned long)pid); + fclose(fp); + } +} + +#define LOGBUFSIZE 1024 + +struct log_buffer { + char buf[LOGBUFSIZE]; + size_t pos; + int prio; +}; + +static void +log_buffer_init(struct log_buffer *lb, int prio) +{ + lb->buf[sizeof(lb->buf) - 1] = 0; + lb->pos = 0; + lb->prio = prio; +} + +static void +log_buffer_read(int fd, struct log_buffer *lb) +{ + int rc; + + rc = read(fd, lb->buf + lb->pos, 1); + if (rc == -1) { + syslog(LOG_ERR, "read: %m"); + } else if (rc == 1) { + if (lb->pos == sizeof(lb->buf)-1) { + syslog(lb->prio, "%s", lb->buf); + lb->pos = 0; + } else if (lb->buf[lb->pos] == '\n') { + lb->buf[lb->pos] = 0; + syslog(lb->prio, "%s", lb->buf); + lb->pos = 0; + } else + lb->pos++; + } +} + +void +wait_loop(pid_t child, int out, int err) +{ + fd_set rdset; + int nfd = (out > err ? out : err) + 1; + struct log_buffer obuf, ebuf; + + openlog(genrc_program, LOG_PID, LOG_DAEMON); + log_buffer_init(&obuf, LOG_INFO); + log_buffer_init(&ebuf, LOG_ERR); + + while (1) { + int rc, status; + + if (waitpid(child, &status, WNOHANG) == child) { + if (genrc_pidfile) + unlink(genrc_pidfile); + if (WIFEXITED(status)) { + syslog(LOG_INFO, "%s exited with status %d", + genrc_program, WEXITSTATUS(status)); + _exit(WEXITSTATUS(status)); + } else if (WIFSIGNALED(status)) { + char const *coremsg = ""; +#ifdef WCOREDUMP + if (WCOREDUMP(status)) + coremsg = " (core dumped)"; +#endif + syslog(LOG_INFO, "%s terminated on signal %d%s", + genrc_program, WTERMSIG(status), coremsg); + } else if (WIFSTOPPED(status)) { + syslog(LOG_INFO, "%s stopped on signal %d", + genrc_program, WSTOPSIG(status)); + } else { + syslog(LOG_INFO, "%s terminated; status %d", + genrc_program, rc); + } + break; + } + + FD_ZERO(&rdset); + FD_SET(out, &rdset); + FD_SET(err, &rdset); + rc = select(nfd, &rdset, NULL, NULL, NULL); + if (rc == -1) { + if (errno == EINTR) + continue; + syslog(LOG_CRIT, "select: %m"); + syslog(LOG_CRIT, "killing %lu and terminating", + (unsigned long) child); + kill(child, SIGKILL); + break; + } + + if (FD_ISSET(out, &rdset)) { + log_buffer_read(out, &obuf); + } + if (FD_ISSET(err, &rdset)) { + log_buffer_read(err, &ebuf); + } + } + _exit(1); +} + +pid_t +start_command(int p[]) +{ + int errpipe[2], outpipe[2]; + pid_t pid; + + xpipe(errpipe); + xpipe(outpipe); + pid = fork(); + if (pid == -1) { + system_error(errno, "pipe"); + return -1; + } + + if (pid == 0) { + char *argv[] = { SHELL, "-c", NULL, NULL }; + int i; + + close(0); + close(1); + close(2); + open("/dev/null", O_RDWR); + dup(outpipe[1]); + dup(errpipe[1]); + + i = max(max(outpipe[0],outpipe[1]), + max(errpipe[0],errpipe[1])); + while (--i > 2) + close(i); + + argv[2] = genrc_command; + execvp(SHELL, argv); + system_error(errno, "failed to exec %s", genrc_program); + exit(127); + } + + write_pid_file(pid); + + close(outpipe[1]); + close(errpipe[1]); + + p[0] = outpipe[0]; + p[1] = errpipe[0]; + return pid; +} + +int +sentinel(void) +{ + pid_t pid; + int p[2]; + + /* Detach from the controlling terminal */ + pid = fork(); + if (pid == -1) { + system_error(errno, "fork"); + return -1; + } + if (pid) /* master */ + return 0; + /* Run as a session leader */ + setsid(); + + pid = fork(); + if (pid == -1) { + system_error(errno, "fork"); + exit(1); + } + if (pid) + _exit(0); + + /* Grand-child */ + pid = start_command(p); + if (pid == -1) + _exit(127); + wait_loop(pid, p[0], p[1]); + _exit(1); +} + + diff --git a/src/transform.c b/src/transform.c new file mode 100644 index 0000000..d3cccb1 --- a/dev/null +++ b/src/transform.c @@ -0,0 +1,560 @@ +/* This file is part of genrc. + Copyright (C) 2006-2018 Sergey Poznyakoff. + (using my implementation for the GNU tar). + + Genrc is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3, or (at your option) + any later version. + + Genrc is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with genrc. If not, see <http://www.gnu.org/licenses/>. */ + +#include "genrc.h" + +enum transform_type { + transform_first, + transform_global +}; + +enum replace_segm_type { + segm_literal, /* Literal segment */ + segm_backref, /* Back-reference segment */ + segm_case_ctl /* Case control segment (GNU extension) */ +}; + +enum case_ctl_type { + ctl_stop, /* Stop case conversion */ + ctl_upcase_next, /* Turn the next character to uppercase */ + ctl_locase_next, /* Turn the next character to lowercase */ + ctl_upcase, /* Turn the replacement to uppercase until ctl_stop */ + ctl_locase /* Turn the replacement to lowercase until ctl_stop */ +}; + +struct replace_segm { + struct replace_segm *next; + enum replace_segm_type type; + union { + struct { + char *ptr; + size_t size; + } literal; /* type == segm_literal */ + size_t ref; /* type == segm_backref */ + enum case_ctl_type ctl; /* type == segm_case_ctl */ + } v; +}; + +struct transform { + struct transform *next; + enum transform_type transform_type; + unsigned match_number; + regex_t regex; + /* Compiled replacement expression */ + struct replace_segm *repl_head, *repl_tail; + size_t segm_count; /* Number of elements in the above list */ +}; + +struct transform_list { + struct transform *head, *tail; +}; + +static struct transform * +new_transform(struct transform_list *tlist) +{ + struct transform *p = xzalloc(sizeof *p); + if (tlist->tail) + tlist->tail->next = p; + else + tlist->head = p; + tlist->tail = p; + return p; +} + +static struct replace_segm * +add_segment(struct transform *tf) +{ + struct replace_segm *segm = xmalloc(sizeof *segm); + segm->next = NULL; + if (tf->repl_tail) + tf->repl_tail->next = segm; + else + tf->repl_head = segm; + tf->repl_tail = segm; + tf->segm_count++; + return segm; +} + +static void +add_literal_segment(struct transform *tf, char *str, char *end) +{ + size_t len = end - str; + if (len) { + struct replace_segm *segm = add_segment(tf); + segm->type = segm_literal; + segm->v.literal.ptr = xmalloc(len + 1); + memcpy(segm->v.literal.ptr, str, len); + segm->v.literal.ptr[len] = 0; + segm->v.literal.size = len; + } +} + +static void +add_char_segment(struct transform *tf, int chr) +{ + struct replace_segm *segm = add_segment(tf); + segm->type = segm_literal; + segm->v.literal.ptr = xmalloc(2); + segm->v.literal.ptr[0] = chr; + segm->v.literal.ptr[1] = 0; + segm->v.literal.size = 1; +} + +static void +add_backref_segment(struct transform *tf, size_t ref) +{ + struct replace_segm *segm = add_segment(tf); + segm->type = segm_backref; + segm->v.ref = ref; +} + +static void +add_case_ctl_segment(struct transform *tf, enum case_ctl_type ctl) +{ + struct replace_segm *segm = add_segment(tf); + segm->type = segm_case_ctl; + segm->v.ctl = ctl; +} + +static const char * +parse_transform_expr(struct transform_list *tlist, + const char *expr, int cflags) +{ + int delim; + int i, j, rc; + char *str, *beg, *cur; + const char *p; + struct transform *tf = new_transform(tlist); + + if (expr[0] != 's') + usage_error("invalid transform expression: %s", expr); + + delim = expr[1]; + + /* Scan regular expression */ + for (i = 2; expr[i] && expr[i] != delim; i++) + if (expr[i] == '\\' && expr[i + 1]) + i++; + + if (expr[i] != delim) + usage_error("missing 2nd delimiter in position %d of expression %s", + i, expr); + + /* Scan replacement expression */ + for (j = i + 1; expr[j] && expr[j] != delim; j++) + if (expr[j] == '\\' && expr[j + 1]) + j++; + + if (expr[j] != delim) + usage_error("missing trailing delimiter in position %d of expression %s", + j, expr); + + /* Check flags */ + tf->transform_type = transform_first; + for (p = expr + j + 1; *p && *p != ';'; p++) + switch (*p) { + case 'g': + tf->transform_type = transform_global; + break; + + case 'i': + cflags |= REG_ICASE; + break; + + case 'x': + cflags |= REG_EXTENDED; + break; + + case 'b': + cflags &= ~REG_EXTENDED; + break; + + case '0': + case '1': + case '2': + case '3': + case '4': + case '5': + case '6': + case '7': + case '8': + case '9': + tf->match_number = strtoul(p, (char **)&p, 0); + p--; + break; + + default: + usage_error("unknown flag in transform expression: %c", + *p); + } + + if (*p == ';') + p++; + + /* Extract and compile regex */ + str = xmalloc(i - 1); + memcpy(str, expr + 2, i - 2); + str[i - 2] = 0; + + rc = regcomp(&tf->regex, str, cflags); + + if (rc) { + char errbuf[512]; + regerror(rc, &tf->regex, errbuf, sizeof(errbuf)); + usage_error("invalid transform expression: %s", errbuf); + } + + if (str[0] == '^' || str[strlen(str) - 1] == '$') + tf->transform_type = transform_first; + + free(str); + + /* Extract and compile replacement expr */ + i++; + str = xmalloc(j - i + 1); + memcpy(str, expr + i, j - i); + str[j - i] = 0; + + for (cur = beg = str; *cur;) { + if (*cur == '\\') { + size_t n; + + add_literal_segment(tf, beg, cur); + switch (*++cur) { + case '0': + case '1': + case '2': + case '3': + case '4': + case '5': + case '6': + case '7': + case '8': + case '9': + n = strtoul(cur, &cur, 10); + if (n > tf->regex.re_nsub) + usage_error("invalid transform replacement: " + "back reference out of range"); + add_backref_segment(tf, n); + break; + + case '\\': + add_char_segment(tf, '\\'); + cur++; + break; + + case 'a': + add_char_segment(tf, '\a'); + cur++; + break; + + case 'b': + add_char_segment(tf, '\b'); + cur++; + break; + + case 'f': + add_char_segment(tf, '\f'); + cur++; + break; + + case 'n': + add_char_segment(tf, '\n'); + cur++; + break; + + case 'r': + add_char_segment(tf, '\r'); + cur++; + break; + + case 't': + add_char_segment(tf, '\t'); + cur++; + break; + + case 'v': + add_char_segment(tf, '\v'); + cur++; + break; + + case '&': + add_char_segment(tf, '&'); + cur++; + break; + + case 'L': + /* Turn the replacement to lowercase until + a `\U' or `\E' is found, */ + add_case_ctl_segment(tf, ctl_locase); + cur++; + break; + + case 'l': + /* Turn the next character to lowercase, */ + add_case_ctl_segment(tf, ctl_locase_next); + cur++; + break; + + case 'U': + /* Turn the replacement to uppercase until + a `\L' or `\E' is found, */ + add_case_ctl_segment(tf, ctl_upcase); + cur++; + break; + + case 'u': + /* Turn the next character to uppercase, */ + add_case_ctl_segment(tf, ctl_upcase_next); + cur++; + break; + + case 'E': + /* Stop case conversion started by + `\L' or `\U'. */ + add_case_ctl_segment(tf, ctl_stop); + cur++; + break; + + default: + if (*cur == delim) + add_char_segment(tf, delim); + else { + char buf[2]; + buf[0] = '\\'; + buf[1] = *cur; + add_literal_segment(tf, buf, buf + 2); + } + cur++; + break; + } + beg = cur; + } else if (*cur == '&') { + add_literal_segment(tf, beg, cur); + add_backref_segment(tf, 0); + beg = ++cur; + } else + cur++; + } + add_literal_segment(tf, beg, cur); + + return p; +} + +TRANSFORM +compile_transform_expr(const char *expr, int cflags) +{ + struct transform_list tlist = { NULL, NULL }; + while (*expr) + expr = parse_transform_expr(&tlist, expr, cflags); + return tlist.head; +} + +/* Run case conversion specified by CASE_CTL on array PTR of SIZE + characters. Returns pointer to statically allocated storage. */ +static const char * +run_case_conv(enum case_ctl_type case_ctl, const char *ptr, size_t size) +{ + static char *case_ctl_buffer; + static size_t case_ctl_bufsize; + char *p; + + if (case_ctl_bufsize < size) { + case_ctl_bufsize = size; + case_ctl_buffer = xrealloc(case_ctl_buffer, case_ctl_bufsize); + } + memcpy(case_ctl_buffer, ptr, size); + switch (case_ctl) { + case ctl_upcase_next: + case_ctl_buffer[0] = toupper(case_ctl_buffer[0]); + break; + + case ctl_locase_next: + case_ctl_buffer[0] = tolower(case_ctl_buffer[0]); + break; + + case ctl_upcase: + for (p = case_ctl_buffer; p < case_ctl_buffer + size; p++) + *p = toupper(*p); + break; + + case ctl_locase: + for (p = case_ctl_buffer; p < case_ctl_buffer + size; p++) + *p = tolower(*p); + break; + + case ctl_stop: + break; + } + return case_ctl_buffer; +} + + +struct case_ctl_state { + enum case_ctl_type case_ctl; /* Current case conversion op */ + enum case_ctl_type save_ctl; /* Saved case_ctl for \u and \l */ +}; + +/* Reset case conversion after a single-char operation */ +static void +case_ctl_reset(struct case_ctl_state *state) +{ + if (state->case_ctl == ctl_upcase_next + || state->case_ctl == ctl_locase_next) { + state->case_ctl = state->save_ctl; + state->save_ctl = ctl_stop; + } +} + +static void +repl_loop(struct transform *tf, grecs_txtacc_t acc, + const char *input, + regmatch_t *rmp, + struct case_ctl_state *state) +{ + const char *ptr; + struct replace_segm *segm; + + for (segm = tf->repl_head; segm; segm = segm->next) { + switch (segm->type) { + case segm_literal: + /* Literal segment */ + if (state->case_ctl == ctl_stop) + ptr = segm->v.literal.ptr; + else { + ptr = run_case_conv(state->case_ctl, + segm->v.literal.ptr, + segm->v.literal.size); + case_ctl_reset(state); + } + grecs_txtacc_grow(acc, ptr, segm->v.literal.size); + break; + + case segm_backref: + /* Back-reference segment */ + if (rmp[segm->v.ref].rm_so != -1 + && rmp[segm->v.ref].rm_eo != -1) { + size_t size = rmp[segm->v.ref].rm_eo - + rmp[segm->v.ref].rm_so; + ptr = input + rmp[segm->v.ref].rm_so; + if (state->case_ctl != ctl_stop) { + ptr = run_case_conv(state->case_ctl, + ptr, size); + case_ctl_reset(state); + } + grecs_txtacc_grow(acc, ptr, size); + } + break; + + case segm_case_ctl: + switch (segm->v.ctl) { + case ctl_upcase_next: + case ctl_locase_next: + switch (state->save_ctl) { + case ctl_stop: + case ctl_upcase: + case ctl_locase: + state->save_ctl = state->case_ctl; + default: + break; + } + /*FALL THROUGH */ + + case ctl_upcase: + case ctl_locase: + case ctl_stop: + state->case_ctl = segm->v.ctl; + } + } + } +} + +void +_single_transform_name_to_acc(struct transform *tf, grecs_txtacc_t acc, + const char *input) +{ + regmatch_t *rmp; + int rc; + size_t nmatches = 0; + struct case_ctl_state ctl_state = { ctl_stop, ctl_stop }; + + rmp = xmalloc((tf->regex.re_nsub + 1) * sizeof(*rmp)); + + while (*input) { + size_t disp; + + rc = regexec(&tf->regex, input, tf->regex.re_nsub + 1, rmp, 0); + + if (rc == 0) { + disp = rmp[0].rm_eo; + + if (rmp[0].rm_so) + grecs_txtacc_grow(acc, input, rmp[0].rm_so); + + nmatches++; + if (tf->match_number && nmatches < tf->match_number) { + grecs_txtacc_grow(acc, input, disp); + input += disp; + continue; + } + repl_loop(tf, acc, input, rmp, &ctl_state); + } else { + disp = strlen(input); + grecs_txtacc_grow(acc, input, disp); + } + + input += disp; + + if (tf->transform_type == transform_first) { + grecs_txtacc_grow_string(acc, input); + break; + } + } + + free(rmp); + grecs_txtacc_grow_char(acc, 0); +} + +char * +transform_string(TRANSFORM tf, const char *input) +{ + if (tf) { + grecs_txtacc_t acc = grecs_txtacc_create(); + char *buf = NULL; + + for (; tf; tf = tf->next) { + _single_transform_name_to_acc(tf, acc, input); + if (buf) + grecs_txtacc_free_string(acc, buf); + buf = grecs_txtacc_finish(acc, 0); + input = buf; + } + buf = xstrdup(buf); + grecs_txtacc_free(acc); + return buf; + } + return xstrdup(input); +} + +char * +transform_string_if_match(TRANSFORM tf, const char *input) +{ + int rc = regexec(&tf->regex, input, 0, NULL, 0); + if (rc == 0) + return transform_string(tf, input); + return NULL; +} |