-
Notifications
You must be signed in to change notification settings - Fork 125
/
fusioninventory-injector
executable file
·139 lines (111 loc) · 3.19 KB
/
fusioninventory-injector
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#!/usr/bin/perl
use strict;
use warnings;
use Compress::Zlib;
use English qw(-no_match_vars);
use Fcntl qw(:flock);
use Getopt::Long;
use LWP::UserAgent;
use Pod::Usage;
my %options = (
useragent => 'FusionInventory-Injector'
);
GetOptions(
\%options,
'help|h',
'directory|d=s',
'file|f=s',
'url|u=s',
'useragent=s',
'remove|r',
'verbose|v',
'stdin',
);
$OUTPUT_AUTOFLUSH = 1;
pod2usage(-verbose => 0, -exitstatus => 0) if $options{help};
if ($options{file}) {
die "file $options{file} does not exist"
unless -f $options{file};
loadfile($options{file});
} elsif ($options{stdin}) {
loadstdin();
} elsif ($options{directory}) {
die "directory $options{directory} does not exist"
unless -d $options{directory};
loaddirectory($options{directory});
} else {
pod2usage();
}
exit(0);
sub loadfile {
my $file = shift;
die "can't read file $file" unless -r $file;
print "Loading $file..." if $options{verbose};
open (my $fh, '<', $file) or die "can't open file $file: $ERRNO";
## no critic (ProhibitBitwise)
flock ($fh, LOCK_EX | LOCK_NB) or die "can't lock file $file: $ERRNO";
local $RS;
my $content = <$fh>;
close $fh or die "Can't close file $file: $ERRNO";
sendContent($content);
}
sub loaddirectory {
my $directory = shift;
die "can't read directory $directory" unless -r $directory;
opendir (my $dh, $directory)
or die "can't open directory $directory: $ERRNO";
foreach ( readdir($dh) ) {
loadfile("$directory/$_") if (/\.ocs$/);
}
closedir $dh;
}
sub loadstdin {
my $content;
undef $RS;
$content = <STDIN>;
sendContent($content);
}
sub sendContent {
my $content = shift;
my $ua = LWP::UserAgent->new(
agent => $options{useragent}
);
my $request = HTTP::Request->new( POST => $options{url} );
$request->header(
'Pragma' => 'no-cache',
'Content-type', 'Application/x-compress'
);
if (uncompress($content)) {
$content = uncompress($content);
}
$request->content(compress($content));
my $res = $ua->request($request);
if ($res->is_success()) {
print "OK\n" if $options{verbose};
print STDERR "Can't remove $options{file}: $ERRNO\n"
if ($options{remove} && (!unlink $options{file}));
} else {
if ($options{verbose}) {
print "ERROR: ";
print $res->status_line(), "\n";
}
}
}
__END__
=head1 NAME
fusioninventory-injector - A tool to push inventory in an OCS Inventory or compatible server.
=head1 SYNOPSIS
fusioninventory-injector [options] [--file <file>|--directory <directory|--stdin]
Options:
-h --help this menu
-d --directory load every .ocs files from a directory
-f --file load a speficic file
-u --url server URL
-r --remove remove succesfuly injected files
-v --verbose verbose mode
--stdin read data from STDIN
Examples:
fusioninventory-injector -v -f /tmp/toto-2010-09-10-11-42-22.ocs --url https://login:pw@server/ocsinventory
=head1 DESCRIPTION
This tool can be used to test your server, do benchmark or push inventory from
off-line machine.