summaryrefslogtreecommitdiffstats
path: root/bitbake/lib/bb/fetch/ssh.py
blob: 57874d5ba941c45842d10adab566df528d84c1d7 (plain)
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
#!/usr/bin/env python
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
'''
BitBake 'Fetch' implementations

This implementation is for Secure Shell (SSH), and attempts to comply with the
IETF secsh internet draft:
    http://tools.ietf.org/wg/secsh/draft-ietf-secsh-scp-sftp-ssh-uri/

    Currently does not support the sftp parameters, as this uses scp
    Also does not support the 'fingerprint' connection parameter.

Copyright (C) 2006  OpenedHand Ltd.

Based in part on svk.py:
    Copyright (C) 2006 Holger Hans Peter Freyther
    Based on svn.py:
        Copyright (C) 2003, 2004  Chris Larson
        Based on functions from the base bb module:
            Copyright 2003 Holger Schurig


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 2 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, write to the Free Software Foundation, Inc., 59 Temple
Place, Suite 330, Boston, MA 02111-1307 USA.
'''
import re, os
import bb
from   bb import data
from   bb.fetch import Fetch
from   bb.fetch import FetchError
from   bb.fetch import MissingParameterError


__pattern__ = re.compile(r'''
 \s*                 # Skip leading whitespace
 ssh://              # scheme
 (                   # Optional username/password block
  (?P<user>\S+)      # username
  (:(?P<pass>\S+))?  # colon followed by the password (optional)
 )?
 (?P<cparam>(;[^;]+)*)?  # connection parameters block (optional)
 @
 (?P<host>\S+?)          # non-greedy match of the host
 (:(?P<port>[0-9]+))?    # colon followed by the port (optional)
 /
 (?P<path>[^;]+)         # path on the remote system, may be absolute or relative,
                         # and may include the use of '~' to reference the remote home
                         # directory
 (?P<sparam>(;[^;]+)*)?  # parameters block (optional)
 $
''', re.VERBOSE)

class SSH(Fetch):
    '''Class to fetch a module or modules via Secure Shell'''

    def supports(self, url, d):
        return __pattern__.match(url) != None

    def localpath(self, url, d):
        m = __pattern__.match(url)
        path = m.group('path')
        host = m.group('host')
        lpath = os.path.join(data.getVar('DL_DIR', d, 1), host, os.path.basename(path))
        return lpath

    def go(self, d, urls = []):
        if not urls:
            urls = self.urls

        for url in urls:
            dldir = data.getVar('DL_DIR', d, 1)

            m = __pattern__.match(url)
            path = m.group('path')
            host = m.group('host')
            port = m.group('port')
            user = m.group('user')
            password = m.group('pass')

            ldir = os.path.join(dldir, host)
            lpath = os.path.join(ldir, os.path.basename(path))

            if not os.path.exists(ldir):
                os.makedirs(ldir)

            if port:
                port = '-P %s' % port
            else:
                port = ''

            if user:
                fr = user
                if password:
                    fr += ':%s' % password
                fr += '@%s' % host
            else:
                fr = host
            fr += ':%s' % path


            import commands
            cmd = 'scp -B -r %s %s %s/' % (
                port,
                commands.mkarg(fr),
                commands.mkarg(ldir)
            )

            (exitstatus, output) = commands.getstatusoutput(cmd)
            if exitstatus != 0:
                print output
                raise FetchError('Unable to fetch %s' % url)