← Back to branch summary

~ubuntu-branches/ubuntu/wily/apparmor/wily

« back to all changes in this revision

Viewing changes to utils/Immunix/Repository.pm

  • Committer: Bazaar Package Importer
  • Author(s): Kees Cook
  • Date: 2011-04-27 10:38:07 UTC
  • mfrom: (5.1.118 natty)
  • Revision ID: james.westby@ubuntu.com-20110427103807-ym3rhwys6o84ith0
Tags: 2.6.1-2
debian/copyright: clarify for some full organization names.

Show diffs side-by-side

added added

removed removed

Lines of Context:
utils/Immunix/Repository.pm
 
1
# ----------------------------------------------------------------------
 
2
#    Copyright (c) 2008 Dominic Reynolds
 
3
#
 
4
#    This program is free software; you can redistribute it and/or
 
5
#    modify it under the terms of version 2 of the GNU General Public
 
6
#    License as published by the Free Software Foundation.
 
7
#
 
8
#    This program is distributed in the hope that it will be useful,
 
9
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
#    GNU General Public License for more details.
 
12
#
 
13
#
 
14
# ----------------------------------------------------------------------
 
15
 
 
16
package Immunix::Repository;
 
17
 
 
18
use strict;
 
19
use warnings;
 
20
 
 
21
use Carp;
 
22
use Cwd qw(cwd realpath);
 
23
use Data::Dumper;
 
24
use File::Basename;
 
25
use File::Temp qw/ tempfile tempdir /;
 
26
use Immunix::Config;
 
27
use Locale::gettext;
 
28
use POSIX;
 
29
use RPC::XML;
 
30
use RPC::XML::Client;
 
31
 
 
32
 
 
33
require Exporter;
 
34
our @ISA    = qw(Exporter);
 
35
our @EXPORT = qw(
 
36
    get_repo_client
 
37
    did_result_succeed
 
38
    get_result_error
 
39
    user_login
 
40
    user_register
 
41
    upload_profile
 
42
    fetch_profile_by_id
 
43
    fetch_profiles_by_user
 
44
    fetch_profiles_by_name
 
45
    fetch_profiles_by_name_and_user
 
46
    fetch_newer_profile
 
47
    get_repo_config
 
48
    set_repo_config
 
49
);
 
50
 
 
51
our %clients;
 
52
our %uid2login;
 
53
our $DEBUGGING = 0;
 
54
our $repo_cfg;
 
55
our $aa_cfg;
 
56
 
 
57
sub get_repo_client ($) {
 
58
    my $repo_url = shift;
 
59
    unless ( $clients{$repo_url} ) {
 
60
        $clients{$repo_url} = new RPC::XML::Client $repo_url;
 
61
    }
 
62
    return $clients{$repo_url};
 
63
}
 
64
 
 
65
sub did_result_succeed {
 
66
    my $result = shift;
 
67
 
 
68
    my $ref = ref $result;
 
69
    return ($ref && $ref ne "RPC::XML::fault") ? 1 : 0;
 
70
}
 
71
 
 
72
sub get_result_error {
 
73
    my $result = shift;
 
74
 
 
75
    if (ref $result) {
 
76
        if (ref $result eq "RPC::XML::fault") {
 
77
            $result = $result->string;
 
78
        } else {
 
79
            $result = $$result;
 
80
        }
 
81
    }
 
82
    return $result;
 
83
}
 
84
 
 
85
sub user_login ($$$) {
 
86
    my ($repo_url,$user,$pass) = @_;
 
87
    my ($status,$detail);
 
88
    my $repo_client = get_repo_client( $repo_url );
 
89
    if ( $repo_client ) {
 
90
        my $res = $repo_client->send_request('LoginConfirm', $user, $pass);
 
91
        if (did_result_succeed($res)) {
 
92
            $status = 1;
 
93
            $detail = "";
 
94
        } else {
 
95
            $status = 0;
 
96
            $detail = get_result_error($res);
 
97
        }
 
98
    }
 
99
    return $status,$detail;
 
100
}
 
101
 
 
102
 
 
103
sub user_register ($$$$) {
 
104
    my ($repo_url,$user,$pass,$email) = @_;
 
105
    my $repo_client = get_repo_client( $repo_url );
 
106
    my ($status,$detail);
 
107
    if ( $repo_client ) {
 
108
        my $res = $repo_client->send_request('Signup', $user, $pass, $email);
 
109
        if (did_result_succeed($res)) {
 
110
            $status = 1;
 
111
            $detail = "";
 
112
        } else {
 
113
            $status  = 0;
 
114
            $detail = get_result_error($res);
 
115
        }
 
116
    }
 
117
    return $status,$detail;
 
118
}
 
119
 
 
120
sub upload_profile ($$$$$$$) {
 
121
    my ($repo_url,$user,$pass,$distro,$pname,$profile,$changelog) = @_;
 
122
    my ($status,$detail);
 
123
    my $repo_client = get_repo_client( $repo_url );
 
124
    my $res = $repo_client->send_request( 'Create', $user, $pass, $distro,
 
125
                                          $pname, $profile, $changelog);
 
126
    if (did_result_succeed($res)) {
 
127
       $detail = $res->value;
 
128
       $status  = 1;
 
129
    } else {
 
130
       $detail = get_result_error($res);
 
131
       $status  = 0;
 
132
    }
 
133
    return $status,$detail;
 
134
}
 
135
 
 
136
sub fetch_profile_by_id ($$) {
 
137
    my ($repo_url,$id) = @_;
 
138
    my $repo_client = get_repo_client( $repo_url );
 
139
    my $repo_profile;
 
140
    my ($status,$detail);
 
141
    my $res = $repo_client->send_request('Show', $id);
 
142
    if (did_result_succeed($res)) {
 
143
        $status = 1;
 
144
        $detail = $res->value();
 
145
    } else {
 
146
        $status  = 0;
 
147
        $detail = get_result_error($res);
 
148
    }
 
149
 
 
150
    return $status, $detail;
 
151
}
 
152
 
 
153
 
 
154
sub fetch_profiles ($$$$) {
 
155
    my ($repo_url,$distro,$username,$fqdn) = @_;
 
156
    my $p_hash = {};
 
157
    my ($status,$detail);
 
158
    my $repo_client = get_repo_client( $repo_url );
 
159
    my $res =
 
160
      $repo_client->send_request('FindProfiles', $distro, $fqdn, $username);
 
161
    if (did_result_succeed($res)) {
 
162
        $status = 1;
 
163
        for my $p ( @$res ) {
 
164
            my $p_repo = $p->{profile}->value();
 
165
            $p_repo =~ s/flags=\(complain\)// if ( $p_repo );  #strip complain flag
 
166
            $p->{profile}             = $p_repo;
 
167
            $p->{user_id}             = $p->{user_id}->value();
 
168
            $p->{id}                  = $p->{id}->value();
 
169
            $p->{name}                = $p->{name}->value();
 
170
            $p->{created_at}          = $p->{created_at}->value();
 
171
            $p->{downloaded_count}    = $p->{downloaded_count}->value();
 
172
        }
 
173
        $detail = $res;
 
174
    } else {
 
175
        $status = 0;
 
176
        $detail = get_result_error($res);
 
177
    }
 
178
    return $status,$detail;
 
179
}
 
180
 
 
181
sub fetch_profiles_by_user ($$$) {
 
182
    my ($repo_url,$distro,$username) = @_;
 
183
    my $p_hash = {};
 
184
    my ($status,$detail) = fetch_profiles( $repo_url, $distro, $username, "" );
 
185
    if ( $status ) {
 
186
        for my $p ( @$detail ) {
 
187
            my $p_repo = $p->{profile};
 
188
            if ($p_repo ne "") {
 
189
                $p->{username} = $username;
 
190
                $p_hash->{$p->{name}} = $p;
 
191
            }
 
192
        }
 
193
    } else {
 
194
        return ($status,$detail);
 
195
    }
 
196
    return($status,$p_hash);
 
197
}
 
198
 
 
199
 
 
200
sub fetch_profiles_by_name_and_user ($$$$) {
 
201
    my ($repo_url,$distro,$fqdbin, $username) = @_;
 
202
    my $p_hash = {};
 
203
    my ($status,$detail) = fetch_profiles( $repo_url, $distro, $username, $fqdbin );
 
204
    if ( $status ) {
 
205
        for my $p ( @$detail ) {
 
206
            my $p_repo = $p->{profile}?$p->{profile}:"";
 
207
            $p_hash->{$p->{name}} = $p if ($p_repo ne "");
 
208
        }
 
209
    } else {
 
210
        return ($status,$detail);
 
211
    }
 
212
    return($status,$p_hash);
 
213
}
 
214
 
 
215
 
 
216
sub fetch_profiles_by_name ($$$) {
 
217
    my ($repo_url,$distro,$fqdbin) = @_;
 
218
    my ($status,$detail,$data);
 
219
    $detail = {};
 
220
    ($status,$data) = fetch_profiles( $repo_url, $distro, "", $fqdbin);
 
221
    if ($status) {
 
222
        my @uids;
 
223
        for my $p (@$data) {
 
224
            push @uids, $p->{user_id};
 
225
        }
 
226
        my ($status_unames,$unames) = fetch_usernames_from_uids($repo_url, @uids);
 
227
        if ( $status_unames ) {
 
228
            for my $p (@$data) {
 
229
                if ( $unames->{$p->{user_id}} ) {
 
230
                    $p->{username} = $unames->{$p->{user_id}};
 
231
                } else {
 
232
                    $p->{username} = "unkown-" . $p->{user_id};
 
233
                }
 
234
            }
 
235
 
 
236
        } else {
 
237
            print STDOUT "ERROR UID\n";
 
238
        }
 
239
        for my $p (@$data) {
 
240
            $p->{profile_type} = "REPOSITORY";
 
241
            $detail->{$p->{username}} = $p;
 
242
        }
 
243
    } else {
 
244
        $detail = $data;
 
245
    }
 
246
    return $status,$detail;
 
247
}
 
248
 
 
249
 
 
250
sub fetch_newer_profile ($$$$$) {
 
251
    my ($repo_url,$distro,$user,$id,$profile)  = @_;
 
252
    my $repo_client = get_repo_client( $repo_url );
 
253
    my $p;
 
254
    my ($status,$detail);
 
255
 
 
256
    if ($repo_client) {
 
257
        my $res =
 
258
          $repo_client->send_request('FindProfiles', $distro, $profile, $user);
 
259
        if (did_result_succeed($res)) {
 
260
            my @profiles;
 
261
            my @profile_list = @{$res->value};
 
262
            $status = 1;
 
263
 
 
264
            if (@profile_list) {
 
265
                if ($profile_list[0]->{id} > $id) {
 
266
                    $p = $profile_list[0];
 
267
                }
 
268
            }
 
269
            $detail = $p;
 
270
        } else {
 
271
            $status = 0;
 
272
            $detail = get_result_error($res);
 
273
        }
 
274
    }
 
275
    return $status,$detail;
 
276
}
 
277
 
 
278
sub fetch_usernames_from_uids ($) {
 
279
    my ($repo_url,@searchuids) = @_;
 
280
    my ($status,$result) = (1,{});
 
281
    my @uids;
 
282
 
 
283
    for my $uid ( @searchuids ) {
 
284
        if ( $uid2login{$uid} ) {
 
285
            $result->{$uid} = $uid2login{$uid};
 
286
        } else {
 
287
            push @uids, $uid;
 
288
        }
 
289
    }
 
290
    if (@uids) {
 
291
        my $repo_client = get_repo_client( $repo_url );
 
292
        #RPC::XML will serialize the array into XML with the is_utf8 flag set
 
293
        #which causes, HTTP:Message to fail.  Looping on the array elements
 
294
        #stops this from happening, and since these are all numbers it
 
295
        #will not cause problems.
 
296
        for my $foo (@uids) {
 
297
            Encode::_utf8_off($foo);
 
298
        }
 
299
        my $res = $repo_client->send_request('LoginNamesFromUserIds', [@uids]);
 
300
        if (did_result_succeed($res)) {
 
301
            my @usernames = @{ $res->value };
 
302
            for my $uid (@uids) {
 
303
                my $username = shift @usernames;
 
304
                $uid2login{$uid} = $username;
 
305
                $result->{$uid} = $uid2login{$uid};
 
306
            }
 
307
        } else {
 
308
            $status = 0;
 
309
            $result = get_result_error($res);
 
310
        }
 
311
    }
 
312
    return $status,$result;
 
313
}
 
314
 
 
315
sub get_repo_config {
 
316
    unless ( $repo_cfg ) {
 
317
       $repo_cfg = Immunix::Config::read_config("repository.conf");
 
318
    }
 
319
    unless ( $aa_cfg ) {
 
320
       $aa_cfg = Immunix::Config::read_config("logprof.conf");
 
321
    }
 
322
    return {
 
323
              "url"      => $aa_cfg->{repository}{url},
 
324
              "distro"   => $aa_cfg->{repository}{distro},
 
325
              "enabled"  => $repo_cfg->{repository}{enabled},
 
326
              "upload"   => $repo_cfg->{repository}{upload},
 
327
              "user"     => $repo_cfg->{repository}{user},
 
328
              "password" => $repo_cfg->{repository}{pass},
 
329
              "email"    => $repo_cfg->{repository}{email}
 
330
            };
 
331
}
 
332
 
 
333
sub set_repo_config ($) {
 
334
    my $cfg = shift;
 
335
    my ($url,$distro,$enabled,$upload,$user,$pass);
 
336
    unless ( $repo_cfg ) {
 
337
       $repo_cfg = Immunix::Config::read_config("repository.conf");
 
338
    }
 
339
    unless ( $aa_cfg ) {
 
340
       $aa_cfg = Immunix::Config::read_config("logprof.conf");
 
341
    }
 
342
    $repo_cfg->{repository}{enabled} = $cfg->{enabled} if ( $cfg->{enabled} );
 
343
    $repo_cfg->{repository}{upload}  = $cfg->{upload}  if ( $cfg->{upload} );
 
344
    $repo_cfg->{repository}{user}    = $cfg->{user}    if ( $cfg->{user} );
 
345
    $repo_cfg->{repository}{pass}    = $cfg->{password}if ( $cfg->{password} );
 
346
    $repo_cfg->{repository}{email}   = $cfg->{email}   if ( $cfg->{email} );
 
347
    $aa_cfg->{repository}{distro}    = $cfg->{distro}  if ( $cfg->{distro} );
 
348
    $aa_cfg->{repository}{url}       = $cfg->{url}     if ( $cfg->{url} );
 
349
    write_config("repository.conf", $repo_cfg);
 
350
    write_config("logprof.conf", $aa_cfg);
 
351
}
 
352
 
 
353
 
 
354
1;