WebSVN – HelenOS – Diff – /branches/dynload/uspace/app/bdsh/cmds/modules/cat/cat.c


/* Copyright (c) 2008, Tim Post <tinkertim@gmail.com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * Neither the name of the original program's authors nor the names of its
 * contributors may be used to endorse or promote products derived from this
 * software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
 
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <getopt.h>
#include <string.h>
#include <fcntl.h>
 
#include "config.h"
#include "util.h"
#include "errors.h"
#include "entry.h"
#include "cat.h"
#include "cmds.h"
 
static char *cmdname = "cat";
#define CAT_VERSION "0.0.1"
#define CAT_DEFAULT_BUFLEN 1024
 
static char *cat_oops = "That option is not yet supported\n";
 
static struct option const long_options[] = {
    { "help", no_argument, 0, 'h' },
    { "version", no_argument, 0, 'v' },
    { "head", required_argument, 0, 'H' },
    { "tail", required_argument, 0, 't' },
    { "buffer", required_argument, 0, 'b' },
    { "more", no_argument, 0, 'm' },
    { 0, 0, 0, 0 }
};
 
/* Dispays help for cat in various levels */
void help_cmd_cat(unsigned int level)
{
    if (level == HELP_SHORT) {
        printf("`%s' shows the contents of files\n", cmdname);
    } else {
        help_cmd_cat(HELP_SHORT);
        printf(
        "Usage:  %s [options] <file1> [file2] [...]\n"
        "Options:\n"
        "  -h, --help       A short option summary\n"
        "  -v, --version    Print version information and exit\n"
        "  -H, --head ##    Print only the first ## bytes\n"
        "  -t, --tail ##    Print only the last ## bytes\n"
        "  -b, --buffer ##  Set the read buffer size to ##\n"
        "  -m, --more       Pause after each screen full\n"
        "Currently, %s is under development, some options don't work.\n",
        cmdname, cmdname);
    }
 
    return;
}
 
static unsigned int cat_file(const char *fname, size_t blen)
{
    int fd, bytes = 0, count = 0, reads = 0;
    off_t total = 0;
    char *buff = NULL;
 
    fd = open(fname, O_RDONLY);
    if (fd < 0) {
        printf("Unable to open %s\n", fname);
        return 1;
    }
 
    total = lseek(fd, 0, SEEK_END);
    lseek(fd, 0, SEEK_SET);
 
    if (NULL == (buff = (char *) malloc(blen + 1))) {
        close(fd);
        printf("Unable to allocate enough memory to read %s\n",
            fname);
        return 1;
    }
 
    do {
        bytes = read(fd, buff, blen);
        if (bytes > 0) {
            count += bytes;
            buff[bytes] = '\0';
            printf("%s", buff);
            reads++;
        }
    } while (bytes > 0);
 
    close(fd);
    if (bytes == -1) {
        printf("Error reading %s\n", fname);
        free(buff);
        return 1;
    }
 
    free(buff);
 
    return 0;
}
 
/* Main entry point for cat, accepts an array of arguments */
int cmd_cat(char **argv)
{
    unsigned int argc, i, ret = 0, buffer = 0;
    int c, opt_ind;
 
    argc = cli_count_args(argv);
 
    for (c = 0, optind = 0, opt_ind = 0; c != -1;) {
        c = getopt_long(argc, argv, "hvmH:t:b:", long_options, &opt_ind);
        switch (c) {
        case 'h':
            help_cmd_cat(HELP_LONG);
            return CMD_SUCCESS;
        case 'v':
            printf("%s\n", CAT_VERSION);
            return CMD_SUCCESS;
        case 'H':
            printf(cat_oops);
            return CMD_FAILURE;
        case 't':
            printf(cat_oops);
            return CMD_FAILURE;
        case 'b':
            printf(cat_oops);
            break;
        case 'm':
            printf(cat_oops);
            return CMD_FAILURE;
        }
    }
 
    argc -= optind;
 
    if (argc < 1) {
        printf("%s - incorrect number of arguments. Try `%s --help'\n",
            cmdname, cmdname);
        return CMD_FAILURE;
    }
 
    if (buffer <= 0)
        buffer = CAT_DEFAULT_BUFLEN;
 
    for (i = optind; argv[i] != NULL; i++)
        ret += cat_file(argv[i], buffer);
 
    if (ret)
        return CMD_FAILURE;
    else
        return CMD_SUCCESS;
}
 
 

Rev 3535	Rev 4389
1	/* Copyright (c) 2008, Tim Post <tinkertim@gmail.com>	1	/* Copyright (c) 2008, Tim Post <tinkertim@gmail.com>
2	* All rights reserved.	2	* All rights reserved.
3	*	3	*
4	* Redistribution and use in source and binary forms, with or without	4	* Redistribution and use in source and binary forms, with or without
5	* modification, are permitted provided that the following conditions are met:	5	* modification, are permitted provided that the following conditions are met:
6	*	6	*
7	* Redistributions of source code must retain the above copyright notice, this	7	* Redistributions of source code must retain the above copyright notice, this
8	* list of conditions and the following disclaimer.	8	* list of conditions and the following disclaimer.
9	*	9	*
10	* Redistributions in binary form must reproduce the above copyright notice,	10	* Redistributions in binary form must reproduce the above copyright notice,
11	* this list of conditions and the following disclaimer in the documentation	11	* this list of conditions and the following disclaimer in the documentation
12	* and/or other materials provided with the distribution.	12	* and/or other materials provided with the distribution.
13	*	13	*
14	* Neither the name of the original program's authors nor the names of its	14	* Neither the name of the original program's authors nor the names of its
15	* contributors may be used to endorse or promote products derived from this	15	* contributors may be used to endorse or promote products derived from this
16	* software without specific prior written permission.	16	* software without specific prior written permission.
17	*	17	*
18	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"	18	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE	19	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE	20	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE	21	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22	* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR	22	* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF	23	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS	24	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN	25	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)	26	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE	27	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28	* POSSIBILITY OF SUCH DAMAGE.	28	* POSSIBILITY OF SUCH DAMAGE.
29	*/	29	*/
30		30
31	#include <stdio.h>	31	#include <stdio.h>
32	#include <stdlib.h>	32	#include <stdlib.h>
33	#include <unistd.h>	33	#include <unistd.h>
34	#include <getopt.h>	34	#include <getopt.h>
35	#include <string.h>	35	#include <string.h>
36	#include <fcntl.h>	36	#include <fcntl.h>
37		37
38	#include "config.h"	38	#include "config.h"
39	#include "util.h"	39	#include "util.h"
40	#include "errors.h"	40	#include "errors.h"
41	#include "entry.h"	41	#include "entry.h"
42	#include "cat.h"	42	#include "cat.h"
43	#include "cmds.h"	43	#include "cmds.h"
44		44
45	static char *cmdname = "cat";	45	static char *cmdname = "cat";
46	#define CAT_VERSION "0.0.1"	46	#define CAT_VERSION "0.0.1"
47	#define CAT_DEFAULT_BUFLEN 1024	47	#define CAT_DEFAULT_BUFLEN 1024
48		48
49	static char *cat_oops = "That option is not yet supported\n";	49	static char *cat_oops = "That option is not yet supported\n";
50		50
51	static struct option const long_options[] = {	51	static struct option const long_options[] = {
52	{ "help", no_argument, 0, 'h' },	52	{ "help", no_argument, 0, 'h' },
53	{ "version", no_argument, 0, 'v' },	53	{ "version", no_argument, 0, 'v' },
54	{ "head", required_argument, 0, 'H' },	54	{ "head", required_argument, 0, 'H' },
55	{ "tail", required_argument, 0, 't' },	55	{ "tail", required_argument, 0, 't' },
56	{ "buffer", required_argument, 0, 'b' },	56	{ "buffer", required_argument, 0, 'b' },
57	{ "more", no_argument, 0, 'm' },	57	{ "more", no_argument, 0, 'm' },
58	{ 0, 0, 0, 0 }	58	{ 0, 0, 0, 0 }
59	};	59	};
60		60
61	/* Dispays help for cat in various levels */	61	/* Dispays help for cat in various levels */
62	void help_cmd_cat(unsigned int level)	62	void help_cmd_cat(unsigned int level)
63	{	63	{
64	if (level == HELP_SHORT) {	64	if (level == HELP_SHORT) {
65	printf("`%s' shows the contents of files\n", cmdname);	65	printf("`%s' shows the contents of files\n", cmdname);
66	} else {	66	} else {
67	help_cmd_cat(HELP_SHORT);	67	help_cmd_cat(HELP_SHORT);
68	printf(	68	printf(
69	"Usage: %s [options] <file1> [file2] [...]\n"	69	"Usage: %s [options] <file1> [file2] [...]\n"
70	"Options:\n"	70	"Options:\n"
71	" -h, --help A short option summary\n"	71	" -h, --help A short option summary\n"
72	" -v, --version Print version information and exit\n"	72	" -v, --version Print version information and exit\n"
73	" -H, --head ## Print only the first ## bytes\n"	73	" -H, --head ## Print only the first ## bytes\n"
74	" -t, --tail ## Print only the last ## bytes\n"	74	" -t, --tail ## Print only the last ## bytes\n"
75	" -b, --buffer ## Set the read buffer size to ##\n"	75	" -b, --buffer ## Set the read buffer size to ##\n"
76	" -m, --more Pause after each screen full\n"	76	" -m, --more Pause after each screen full\n"
77	"Currently, %s is under development, some options don't work.\n",	77	"Currently, %s is under development, some options don't work.\n",
78	cmdname, cmdname);	78	cmdname, cmdname);
79	}	79	}
80		80
81	return;	81	return;
82	}	82	}
83		83
84	static unsigned int cat_file(const char *fname, size_t blen)	84	static unsigned int cat_file(const char *fname, size_t blen)
85	{	85	{
86	int fd, bytes = 0, count = 0, reads = 0;	86	int fd, bytes = 0, count = 0, reads = 0;
87	off_t total = 0;	87	off_t total = 0;
88	char *buff = NULL;	88	char *buff = NULL;
89		89
90	if (-1 == (fd = open(fname, O_RDONLY))) {	90	fd = open(fname, O_RDONLY);
-		91	if (fd < 0) {
91	printf("Unable to open %s\n", fname);	92	printf("Unable to open %s\n", fname);
92	return 1;	93	return 1;
93	}	94	}
94		95
95	total = lseek(fd, 0, SEEK_END);	96	total = lseek(fd, 0, SEEK_END);
96	lseek(fd, 0, SEEK_SET);	97	lseek(fd, 0, SEEK_SET);
97		98
98	if (NULL == (buff = (char *) malloc(blen + 1))) {	99	if (NULL == (buff = (char *) malloc(blen + 1))) {
99	close(fd);	100	close(fd);
100	printf("Unable to allocate enough memory to read %s\n",	101	printf("Unable to allocate enough memory to read %s\n",
101	fname);	102	fname);
102	return 1;	103	return 1;
103	}	104	}
104		105
105	do {	106	do {
106	bytes = read(fd, buff, blen);	107	bytes = read(fd, buff, blen);
107	if (bytes > 0) {	108	if (bytes > 0) {
108	count += bytes;	109	count += bytes;
109	buff[bytes] = '\0';	110	buff[bytes] = '\0';
110	printf("%s", buff);	111	printf("%s", buff);
111	reads++;	112	reads++;
112	}	113	}
113	} while (bytes > 0);	114	} while (bytes > 0);
114		115
115	close(fd);	116	close(fd);
116	if (bytes == -1) {	117	if (bytes == -1) {
117	printf("Error reading %s\n", fname);	118	printf("Error reading %s\n", fname);
118	free(buff);	119	free(buff);
119	return 1;	120	return 1;
120	}	121	}
121		122
122	free(buff);	123	free(buff);
123		124
124	return 0;	125	return 0;
125	}	126	}
126		127
127	/* Main entry point for cat, accepts an array of arguments */	128	/* Main entry point for cat, accepts an array of arguments */
128	int cmd_cat(char **argv)	129	int cmd_cat(char **argv)
129	{	130	{
130	unsigned int argc, i, ret = 0, buffer = 0;	131	unsigned int argc, i, ret = 0, buffer = 0;
131	int c, opt_ind;	132	int c, opt_ind;
132		133
133	argc = cli_count_args(argv);	134	argc = cli_count_args(argv);
134		135
135	for (c = 0, optind = 0, opt_ind = 0; c != -1;) {	136	for (c = 0, optind = 0, opt_ind = 0; c != -1;) {
136	c = getopt_long(argc, argv, "hvmH:t:b:", long_options, &opt_ind);	137	c = getopt_long(argc, argv, "hvmH:t:b:", long_options, &opt_ind);
137	switch (c) {	138	switch (c) {
138	case 'h':	139	case 'h':
139	help_cmd_cat(HELP_LONG);	140	help_cmd_cat(HELP_LONG);
140	return CMD_SUCCESS;	141	return CMD_SUCCESS;
141	case 'v':	142	case 'v':
142	printf("%s\n", CAT_VERSION);	143	printf("%s\n", CAT_VERSION);
143	return CMD_SUCCESS;	144	return CMD_SUCCESS;
144	case 'H':	145	case 'H':
145	printf(cat_oops);	146	printf(cat_oops);
146	return CMD_FAILURE;	147	return CMD_FAILURE;
147	case 't':	148	case 't':
148	printf(cat_oops);	149	printf(cat_oops);
149	return CMD_FAILURE;	150	return CMD_FAILURE;
150	case 'b':	151	case 'b':
151	printf(cat_oops);	152	printf(cat_oops);
152	break;	153	break;
153	case 'm':	154	case 'm':
154	printf(cat_oops);	155	printf(cat_oops);
155	return CMD_FAILURE;	156	return CMD_FAILURE;
156	}	157	}
157	}	158	}
158		159
159	argc -= optind;	160	argc -= optind;
160		161
161	if (argc < 1) {	162	if (argc < 1) {
162	printf("%s - incorrect number of arguments. Try `%s --help'\n",	163	printf("%s - incorrect number of arguments. Try `%s --help'\n",
163	cmdname, cmdname);	164	cmdname, cmdname);
164	return CMD_FAILURE;	165	return CMD_FAILURE;
165	}	166	}
166		167
167	if (buffer <= 0)	168	if (buffer <= 0)
168	buffer = CAT_DEFAULT_BUFLEN;	169	buffer = CAT_DEFAULT_BUFLEN;
169		170
170	for (i = optind; argv[i] != NULL; i++)	171	for (i = optind; argv[i] != NULL; i++)
171	ret += cat_file(argv[i], buffer);	172	ret += cat_file(argv[i], buffer);
172		173
173	if (ret)	174	if (ret)
174	return CMD_FAILURE;	175	return CMD_FAILURE;
175	else	176	else
176	return CMD_SUCCESS;	177	return CMD_SUCCESS;
177	}	178	}
178		179
179		180

Subversion Repositories HelenOS

(root)/branches/dynload/uspace/app/bdsh/cmds/modules/cat/cat.c – Rev 3535 → 4389