summaryrefslogtreecommitdiffstats
path: root/regparse.h
blob: cd653f5a8e011547736aa244d9b2a708fbc6a657 (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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
#ifndef ONIGURUMA_REGPARSE_H
#define ONIGURUMA_REGPARSE_H
/**********************************************************************
  regparse.h -  Oniguruma (regular expression library)
**********************************************************************/
/*-
 * Copyright (c) 2002-2007  K.Kosako  <sndgk393 AT ybb DOT ne DOT jp>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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 "regint.h"

/* node type */
#define NT_STR         0
#define NT_CCLASS      1
#define NT_CTYPE       2
#define NT_CANY        3
#define NT_BREF        4
#define NT_QTFR        5
#define NT_ENCLOSE     6
#define NT_ANCHOR      7
#define NT_LIST        8
#define NT_ALT         9
#define NT_CALL       10

/* node type bit */
#define NTYPE2BIT(type)      (1<<(type))

#define BIT_NT_STR        NTYPE2BIT(NT_STR)
#define BIT_NT_CCLASS     NTYPE2BIT(NT_CCLASS)
#define BIT_NT_CTYPE      NTYPE2BIT(NT_CTYPE)
#define BIT_NT_CANY       NTYPE2BIT(NT_CANY)
#define BIT_NT_BREF       NTYPE2BIT(NT_BREF)
#define BIT_NT_QTFR       NTYPE2BIT(NT_QTFR)
#define BIT_NT_ENCLOSE    NTYPE2BIT(NT_ENCLOSE)
#define BIT_NT_ANCHOR     NTYPE2BIT(NT_ANCHOR)
#define BIT_NT_LIST       NTYPE2BIT(NT_LIST)
#define BIT_NT_ALT        NTYPE2BIT(NT_ALT)
#define BIT_NT_CALL       NTYPE2BIT(NT_CALL)

#define IS_NODE_TYPE_SIMPLE(type) \
  ((NTYPE2BIT(type) & (BIT_NT_STR | BIT_NT_CCLASS | BIT_NT_CTYPE |\
                       BIT_NT_CANY | BIT_NT_BREF)) != 0)

#define NTYPE(node)             ((node)->u.base.type)
#define SET_NTYPE(node, ntype)   (node)->u.base.type = (ntype)

#define NSTR(node)         (&((node)->u.str))
#define NCCLASS(node)      (&((node)->u.cclass))
#define NCTYPE(node)       (&((node)->u.ctype))
#define NBREF(node)        (&((node)->u.bref))
#define NQTFR(node)        (&((node)->u.qtfr))
#define NENCLOSE(node)     (&((node)->u.enclose))
#define NANCHOR(node)      (&((node)->u.anchor))
#define NCONS(node)        (&((node)->u.cons))
#define NCALL(node)        (&((node)->u.call))

#define NCAR(node)         (NCONS(node)->car)
#define NCDR(node)         (NCONS(node)->cdr)



#define ANCHOR_ANYCHAR_STAR_MASK (ANCHOR_ANYCHAR_STAR | ANCHOR_ANYCHAR_STAR_ML)
#define ANCHOR_END_BUF_MASK      (ANCHOR_END_BUF | ANCHOR_SEMI_END_BUF)

#define ENCLOSE_MEMORY           (1<<0)
#define ENCLOSE_OPTION           (1<<1)
#define ENCLOSE_STOP_BACKTRACK   (1<<2)

#define NODE_STR_MARGIN         16
#define NODE_STR_BUF_SIZE       24  /* sizeof(CClassNode) - sizeof(int)*4 */
#define NODE_BACKREFS_SIZE       6

#define NSTR_RAW                (1<<0) /* by backslashed number */
#define NSTR_AMBIG              (1<<1)
#define NSTR_DONT_GET_OPT_INFO  (1<<2)

#define NSTRING_LEN(node)             ((node)->u.str.end - (node)->u.str.s)
#define NSTRING_SET_RAW(node)          (node)->u.str.flag |= NSTR_RAW
#define NSTRING_CLEAR_RAW(node)        (node)->u.str.flag &= ~NSTR_RAW
#define NSTRING_SET_AMBIG(node)        (node)->u.str.flag |= NSTR_AMBIG
#define NSTRING_SET_DONT_GET_OPT_INFO(node) \
  (node)->u.str.flag |= NSTR_DONT_GET_OPT_INFO
#define NSTRING_IS_RAW(node)          (((node)->u.str.flag & NSTR_RAW)   != 0)
#define NSTRING_IS_AMBIG(node)        (((node)->u.str.flag & NSTR_AMBIG) != 0)
#define NSTRING_IS_DONT_GET_OPT_INFO(node) \
  (((node)->u.str.flag & NSTR_DONT_GET_OPT_INFO) != 0)

#define BACKREFS_P(br) \
  (IS_NOT_NULL((br)->back_dynamic) ? (br)->back_dynamic : (br)->back_static);

#define NQ_TARGET_ISNOT_EMPTY     0
#define NQ_TARGET_IS_EMPTY        1
#define NQ_TARGET_IS_EMPTY_MEM    2
#define NQ_TARGET_IS_EMPTY_REC    3

/* status bits */
#define NST_MIN_FIXED             (1<<0)
#define NST_MAX_FIXED             (1<<1)
#define NST_CLEN_FIXED            (1<<2)
#define NST_MARK1                 (1<<3)
#define NST_MARK2                 (1<<4)
#define NST_MEM_BACKREFED         (1<<5)
#define NST_STOP_BT_SIMPLE_REPEAT (1<<6)
#define NST_RECURSION             (1<<7)
#define NST_CALLED                (1<<8)
#define NST_ADDR_FIXED            (1<<9)
#define NST_NAMED_GROUP           (1<<10)
#define NST_NAME_REF              (1<<11)
#define NST_IN_REPEAT             (1<<12) /* STK_REPEAT is nested in stack. */
#define NST_NEST_LEVEL            (1<<13)
#define NST_BY_NUMBER             (1<<14) /* {n,m} */

#define SET_ENCLOSE_STATUS(node,f)      (node)->u.enclose.state |=  (f)
#define CLEAR_ENCLOSE_STATUS(node,f)    (node)->u.enclose.state &= ~(f)

#define IS_ENCLOSE_CALLED(en)          (((en)->state & NST_CALLED)        != 0)
#define IS_ENCLOSE_ADDR_FIXED(en)      (((en)->state & NST_ADDR_FIXED)    != 0)
#define IS_ENCLOSE_RECURSION(en)       (((en)->state & NST_RECURSION)     != 0)
#define IS_ENCLOSE_MARK1(en)           (((en)->state & NST_MARK1)         != 0)
#define IS_ENCLOSE_MARK2(en)           (((en)->state & NST_MARK2)         != 0)
#define IS_ENCLOSE_MIN_FIXED(en)       (((en)->state & NST_MIN_FIXED)     != 0)
#define IS_ENCLOSE_MAX_FIXED(en)       (((en)->state & NST_MAX_FIXED)     != 0)
#define IS_ENCLOSE_CLEN_FIXED(en)      (((en)->state & NST_CLEN_FIXED)    != 0)
#define IS_ENCLOSE_STOP_BT_SIMPLE_REPEAT(en) \
    (((en)->state & NST_STOP_BT_SIMPLE_REPEAT) != 0)
#define IS_ENCLOSE_NAMED_GROUP(en)     (((en)->state & NST_NAMED_GROUP)   != 0)

#define SET_CALL_RECURSION(node)       (node)->u.call.state |= NST_RECURSION
#define IS_CALL_RECURSION(cn)          (((cn)->state & NST_RECURSION)  != 0)
#define IS_CALL_NAME_REF(cn)           (((cn)->state & NST_NAME_REF)   != 0)
#define IS_BACKREF_NAME_REF(bn)        (((bn)->state & NST_NAME_REF)   != 0)
#define IS_BACKREF_NEST_LEVEL(bn)      (((bn)->state & NST_NEST_LEVEL) != 0)
#define IS_QUANTIFIER_IN_REPEAT(qn)    (((qn)->state & NST_IN_REPEAT)  != 0)
#define IS_QUANTIFIER_BY_NUMBER(qn)    (((qn)->state & NST_BY_NUMBER)  != 0)

#define CALLNODE_REFNUM_UNDEF  -1

typedef struct {
  NodeBase base;
  UChar* s;
  UChar* end;
  unsigned int flag;
  int    capa;    /* (allocated size - 1) or 0: use buf[] */
  UChar  buf[NODE_STR_BUF_SIZE];
} StrNode;

typedef struct {
  NodeBase base;
  int state;
  struct _Node* target;
  int lower;
  int upper;
  int greedy;
  int target_empty_info;
  struct _Node* head_exact;
  struct _Node* next_head_exact;
  int is_refered;     /* include called node. don't eliminate even if {0} */
#ifdef USE_COMBINATION_EXPLOSION_CHECK
  int comb_exp_check_num;  /* 1,2,3...: check,  0: no check  */
#endif
} QtfrNode;

typedef struct {
  NodeBase base;
  int state;
  int type;
  int regnum;
  OnigOptionType option;
  struct _Node*  target;
  AbsAddrType    call_addr;
  /* for multiple call reference */
  OnigDistance min_len; /* min length (byte) */
  OnigDistance max_len; /* max length (byte) */ 
  int char_len;         /* character length  */
  int opt_count;        /* referenced count in optimize_node_left() */
} EncloseNode;

#ifdef USE_SUBEXP_CALL

typedef struct {
  int           offset;
  struct _Node* target;
} UnsetAddr;

typedef struct {
  int        num;
  int        alloc;
  UnsetAddr* us;
} UnsetAddrList;

typedef struct {
  NodeBase base;
  int     state;
  int     group_num;
  UChar*  name;
  UChar*  name_end;
  struct _Node*  target;  /* EncloseNode : ENCLOSE_MEMORY */
  UnsetAddrList* unset_addr_list;
} CallNode;

#endif

typedef struct {
  NodeBase base;
  int  state;
  int  back_num;
  int  back_static[NODE_BACKREFS_SIZE];
  int* back_dynamic;
  int  nest_level;
} BRefNode;

typedef struct {
  NodeBase base;
  int type;
  struct _Node* target;
  int char_len;
} AnchorNode;

typedef struct {
  NodeBase base;
  struct _Node* car;
  struct _Node* cdr;
} ConsAltNode;

typedef struct {
  NodeBase base;
  int ctype;
  int not;
} CtypeNode;

typedef struct _Node {
  union {
    NodeBase     base;
    StrNode      str;
    CClassNode   cclass;
    QtfrNode     qtfr;
    EncloseNode  enclose;
    BRefNode     bref;
    AnchorNode   anchor;
    ConsAltNode  cons;
    CtypeNode    ctype;
#ifdef USE_SUBEXP_CALL
    CallNode     call;
#endif
  } u;
} Node;


#define NULL_NODE  ((Node* )0)

#define SCANENV_MEMNODES_SIZE               8
#define SCANENV_MEM_NODES(senv)   \
 (IS_NOT_NULL((senv)->mem_nodes_dynamic) ? \
    (senv)->mem_nodes_dynamic : (senv)->mem_nodes_static)

typedef struct {
  OnigOptionType   option;
  OnigCaseFoldType case_fold_flag;
  OnigEncoding     enc;
  const OnigSyntaxType* syntax;
  BitStatusType    capture_history;
  BitStatusType    bt_mem_start;
  BitStatusType    bt_mem_end;
  BitStatusType    backrefed_mem;
  UChar*           pattern;
  UChar*           pattern_end;
  UChar*           error;
  UChar*           error_end;
  regex_t*         reg;       /* for reg->names only */
  int              num_call;
#ifdef USE_SUBEXP_CALL
  UnsetAddrList*   unset_addr_list;
#endif
  int              num_mem;
#ifdef USE_NAMED_GROUP
  int              num_named;
#endif
  int              mem_alloc;
  Node*            mem_nodes_static[SCANENV_MEMNODES_SIZE];
  Node**           mem_nodes_dynamic;
#ifdef USE_COMBINATION_EXPLOSION_CHECK
  int num_comb_exp_check;
  int comb_exp_max_regnum;
  int curr_max_regnum;
  int has_recursion;
#endif
} ScanEnv;


#define IS_SYNTAX_OP(syn, opm)    (((syn)->op  & (opm)) != 0)
#define IS_SYNTAX_OP2(syn, opm)   (((syn)->op2 & (opm)) != 0)
#define IS_SYNTAX_BV(syn, bvm)    (((syn)->behavior & (bvm)) != 0)

#ifdef USE_NAMED_GROUP
typedef struct {
  int new_val;
} GroupNumRemap;

extern int    onig_renumber_name_table P_((regex_t* reg, GroupNumRemap* map));
#endif

extern int    onig_strncmp P_((const UChar* s1, const UChar* s2, int n));
extern void   onig_strcpy P_((UChar* dest, const UChar* src, const UChar* end));
extern void   onig_scan_env_set_error_string P_((ScanEnv* env, int ecode, UChar* arg, UChar* arg_end));
extern int    onig_scan_unsigned_number P_((UChar** src, const UChar* end, OnigEncoding enc));
extern void   onig_reduce_nested_quantifier P_((Node* pnode, Node* cnode));
extern void   onig_node_conv_to_str_node P_((Node* node, int raw));
extern int    onig_node_str_cat P_((Node* node, const UChar* s, const UChar* end));
extern int    onig_node_str_set P_((Node* node, const UChar* s, const UChar* end));
extern void   onig_node_free P_((Node* node));
extern Node*  onig_node_new_enclose P_((int type));
extern Node*  onig_node_new_anchor P_((int type));
extern Node*  onig_node_new_str P_((const UChar* s, const UChar* end));
extern Node*  onig_node_new_list P_((Node* left, Node* right));
extern Node*  onig_node_list_add P_((Node* list, Node* x));
extern Node*  onig_node_new_alt P_((Node* left, Node* right));
extern void   onig_node_str_clear P_((Node* node));
extern int    onig_free_node_list P_((void));
extern int    onig_names_free P_((regex_t* reg));
extern int    onig_parse_make_tree P_((Node** root, const UChar* pattern, const UChar* end, regex_t* reg, ScanEnv* env));
extern int    onig_free_shared_cclass_table P_((void));

#ifdef ONIG_DEBUG
#ifdef USE_NAMED_GROUP
extern int onig_print_names(FILE*, regex_t*);
#endif
#endif

#endif /* ONIGURUMA_REGPARSE_H */
u'HINFO', keyrecord=u'KEY', kxrecord=u'KX', locrecord='LOC', mdrecord=u'MD', minforecord=u'MINFO', mxrecord=u'MX', naptrrecord=u'NAPTR', nsrecord=u'NS', nsecrecord=u'NSEC', ntxtrecord=u'NTXT', ptrrecord=u'PTR', rrsigrecord=u'RRSIG', sshfprecord=u'SSHFP', srvrecord=u'SRV', txtrecord=u'TXT', ) # supported DNS classes, IN = internet, rest is almost never used _record_classes = (u'IN', u'CS', u'CH', u'HS') # attributes displayed by default for resource records _record_default_attributes = ['%srecord' % r for r in _record_types] _record_default_attributes.append('idnsname') # attributes displayed by default for zones _zone_default_attributes = [ 'idnsname', 'idnszoneactive', 'idnssoamname', 'idnssoarname', 'idnssoaserial', 'idnssoarefresh', 'idnssoaretry', 'idnssoaexpire', 'idnssoaminimum' ] # normalizer for admin email def _rname_normalizer(value): value = value.replace('@', '.') if not value.endswith('.'): value += '.' return value # build zone dn def _get_zone_dn(ldap, idnsname): rdn = ldap.make_rdn_from_attr('idnsname', idnsname) return ldap.make_dn_from_rdn(rdn, _zone_container_dn) # build dn for entry with record def _get_record_dn(ldap, zone, idnsname): parent_dn = _get_zone_dn(ldap, zone) if idnsname == '@' or idnsname == zone: return parent_dn rdn = ldap.make_rdn_from_attr('idnsname', idnsname) return ldap.make_dn_from_rdn(rdn, parent_dn) def dns_container_exists(ldap): """ See if the dns container exists. If not raise an exception. """ basedn = 'cn=dns,%s' % api.env.basedn try: ret = ldap.find_entries('(objectclass=*)', None, basedn, ldap.SCOPE_BASE) except errors.NotFound: raise errors.NotFound(reason=_('DNS is not configured')) return True class dns(Object): """DNS zone/SOA record object.""" label = _('DNS') takes_params = ( Str('idnsname', cli_name='name', label=_('Zone'), doc=_('Zone name (FQDN)'), normalizer=lambda value: value.lower(), primary_key=True, ), Str('idnssoamname', cli_name='name_server', label=_('Authoritative name server'), ), Str('idnssoarname', cli_name='admin_email', label=_('administrator e-mail address'), default_from=lambda idnsname: 'root.%s' % idnsname, normalizer=_rname_normalizer, ), Int('idnssoaserial?', cli_name='serial', label=_('SOA serial'), ), Int('idnssoarefresh?', cli_name='refresh', label=_('SOA refresh'), ), Int('idnssoaretry?', cli_name='retry', label=_('SOA retry'), ), Int('idnssoaexpire?', cli_name='expire', label=_('SOA expire'), ), Int('idnssoaminimum?', cli_name='minimum', label=_('SOA minimum'), ), Int('dnsttl?', cli_name='ttl', label=_('SOA time to live'), ), StrEnum('dnsclass?', cli_name='class', label=_('SOA class'), values=_record_classes, ), Flag('idnsallowdynupdate', cli_name='allow_dynupdate', label=_('allow dynamic update?'), ), Str('idnsupdatepolicy?', cli_name='update_policy', label=_('BIND update policy'), ), ) default_attributes = _zone_default_attributes json_friendly_attributes = ( 'default_attributes', 'label', 'name', 'takes_params' ) def __json__(self): json_dict = dict( (a, getattr(self, a)) for a in self.json_friendly_attributes ) if self.primary_key: json_dict['primary_key'] = self.primary_key.name json_dict['methods'] = [m for m in self.methods] return json_dict api.register(dns) class dns_add(crud.Create): """ Create new DNS zone/SOA record. """ def execute(self, *args, **options): ldap = self.Backend.ldap2 idnsname = args[0] dns_container_exists(ldap) # build entry attributes entry_attrs = self.args_options_2_entry(*args, **options) # build entry DN dn = _get_zone_dn(ldap, idnsname) # fill in required attributes entry_attrs['objectclass'] = ['top', 'idnsrecord', 'idnszone'] entry_attrs['idnszoneactive'] = 'TRUE' entry_attrs['idnsallowdynupdate'] = str( entry_attrs['idnsallowdynupdate'] ).upper() # fill default values, build SOA serial from current date soa_serial = int('%s01' % time.strftime('%Y%d%m')) entry_attrs.setdefault('idnssoaserial', soa_serial) entry_attrs.setdefault('idnssoarefresh', 3600) entry_attrs.setdefault('idnssoaretry', 900) entry_attrs.setdefault('idnssoaexpire', 1209600) entry_attrs.setdefault('idnssoaminimum', 3600) # create zone entry ldap.add_entry(dn, entry_attrs) # get zone entry with created attributes for output (dn, entry_attrs) = ldap.get_entry(dn, entry_attrs.keys()) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, *args, **options): entry_attrs = result['result'] idnsname = result['value'] textui.print_name(self.name) textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_dashed('Created DNS zone "%s".' % idnsname) api.register(dns_add) class dns_del(crud.Delete): """ Delete existing DNS zone/SOA record. """ def execute(self, *args, **options): ldap = self.api.Backend.ldap2 idnsname = args[0] dns_container_exists(ldap) # build zone entry DN dn = _get_zone_dn(ldap, idnsname) # just check if zone exists for now ldap.get_entry(dn, ['']) # retrieve all subentries of zone - records try: (entries, truncated) = ldap.find_entries( None, [''], dn, ldap.SCOPE_ONELEVEL ) except errors.NotFound: (entries, truncated) = (tuple(), False) # kill'em all, records first for e in entries: ldap.delete_entry(e[0]) ldap.delete_entry(dn) return dict(result=True, value=u'') def output_for_cli(self, textui, result, *args, **options): textui.print_name(self.name) textui.print_dashed('Deleted DNS zone "%s".' % args[0]) api.register(dns_del) class dns_mod(crud.Update): """ Modify DNS zone/SOA record. """ def execute(self, *args, **options): ldap = self.api.Backend.ldap2 idnsname = args[0] dns_container_exists(ldap) # build entry attributes, don't include idnsname! entry_attrs = self.args_options_2_entry(*tuple(), **options) entry_attrs['idnsallowdynupdate'] = str( entry_attrs['idnsallowdynupdate'] ).upper() # build entry DN dn = _get_zone_dn(ldap, idnsname) # update zone entry ldap.update_entry(dn, entry_attrs) # get zone entry with modified + default attributes for output (dn, entry_attrs) = ldap.get_entry( dn, (entry_attrs.keys() + _zone_default_attributes) ) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, *args, **options): entry_attrs = result['result'] idnsname = result['value'] textui.print_name(self.name) textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_dashed('Modified DNS zone "%s".' % idnsname) api.register(dns_mod) class dns_find(crud.Search): """ Search for DNS zones/SOA records. """ def execute(self, term, **options): ldap = self.api.Backend.ldap2 dns_container_exists(ldap) # build search filter filter = ldap.make_filter_from_attr('idnsname', term, exact=False) # select attributes we want to retrieve if options.get('all', False): attrs_list = ['*'] else: attrs_list = _zone_default_attributes # get matching entries try: (entries, truncated) = ldap.find_entries( filter, attrs_list, _zone_container_dn, ldap.SCOPE_ONELEVEL ) except errors.NotFound: (entries, truncated) = (tuple(), False) for e in entries: e[1]['dn'] = e[0] entries = tuple(e for (dn, e) in entries) return dict(result=entries, count=len(entries), truncated=truncated) def output_for_cli(self, textui, result, term, **options): entries = result['result'] truncated = result['truncated'] textui.print_name(self.name) for entry_attrs in entries: textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_plain('') textui.print_count( len(entries), '%i DNS zone matched.', '%i DNS zones matched.' ) if truncated: textui.print_dashed('These results are truncated.', below=False) textui.print_dashed( 'Please refine your search and try again.', above=False ) api.register(dns_find) class dns_show(crud.Retrieve): """ Display DNS zone/SOA record. """ def execute(self, idnsname, **options): ldap = self.api.Backend.ldap2 dns_container_exists(ldap) # build entry DN dn = _get_zone_dn(ldap, idnsname) # select attributes we want to retrieve if options.get('all', False): attrs_list = ['*'] else: attrs_list = _zone_default_attributes (dn, entry_attrs) = ldap.get_entry(dn, attrs_list) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, *args, **options): entry_attrs = result['result'] textui.print_name(self.name) textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) api.register(dns_show) class dns_enable(Command): """ Activate DNS zone. """ takes_args = ( Str('zone', cli_name='zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), ) has_output = output.standard_value def execute(self, zone): ldap = self.api.Backend.ldap2 dns_container_exists(ldap) # build entry DN dn = _get_zone_dn(ldap, zone) # activate! try: ldap.update_entry(dn, {'idnszoneactive': 'TRUE'}) except errors.EmptyModlist: pass return dict(result=True, value=zone) def output_for_cli(self, textui, result, zone): textui.print_name(self.name) textui.print_dashed('Activated DNS zone "%s".' % zone) api.register(dns_enable) class dns_disable(Command): """ Deactivate DNS zone. """ takes_args = ( Str('zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), ) has_output = output.standard_value def execute(self, zone): ldap = self.api.Backend.ldap2 dns_container_exists(ldap) # build entry DN dn = _get_zone_dn(ldap, zone) # deactivate! try: ldap.update_entry(dn, {'idnszoneactive': 'FALSE'}) except errors.EmptyModlist: pass return dict(result=True, value=zone) def output_for_cli(self, textui, result, zone): textui.print_name(self.name) textui.print_dashed('Deactivated DNS zone "%s".' % zone) api.register(dns_disable) class dns_add_rr(Command): """ Add new DNS resource record. """ takes_args = ( Str('zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), Str('idnsname', cli_name='resource', label=_('resource name'), default_from=lambda zone: zone.lower(), attribute=True, ), StrEnum('type', label=_('Record type'), values=_record_types, ), Str('data', label=_('Data'), doc=_('Type-specific data'), ), ) takes_options = ( Int('dnsttl?', cli_name='ttl', label=_('Time to live'), attribute=True, ), StrEnum('dnsclass?', cli_name='class', label=_('Class'), values=_record_classes, attribute=True, ), ) has_output = standard_entry def execute(self, zone, idnsname, type, data, **options): ldap = self.api.Backend.ldap2 attr = ('%srecord' % type).lower() dns_container_exists(ldap) # build entry DN dn = _get_record_dn(ldap, zone, idnsname) # get resource entry where to store the new record try: (dn, entry_attrs) = ldap.get_entry(dn, [attr]) except errors.NotFound: if idnsname != '@' and idnsname != zone: # resource entry doesn't exist, check if zone exists zone_dn = _get_zone_dn(ldap, zone) ldap.get_entry(zone_dn, ['']) # it does, create new resource entry # build entry attributes entry_attrs = self.args_options_2_entry( (idnsname, ), **options ) # fill in required attributes entry_attrs['objectclass'] = ['top', 'idnsrecord'] # fill in the record entry_attrs[attr] = data # create the entry ldap.add_entry(dn, entry_attrs) # get entry with created attributes for output (dn, entry_attrs) = ldap.get_entry(dn, entry_attrs.keys()) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) # zone doesn't exist raise # resource entry already exists, create a modlist for the new record # convert entry_attrs keys to lowercase #entry_attrs = dict( # (k.lower(), v) for (k, v) in entry_attrs.iteritems() #) # get new value for record attribute attr_value = entry_attrs.get(attr, []) attr_value.append(data) ldap.update_entry(dn, {attr: attr_value}) # get entry with updated attribute for output (dn, entry_attrs) = ldap.get_entry(dn, ['idnsname', attr]) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, zone, idnsname, type, data, **options): entry_attrs = result['result'] output = '"%s %s %s" to zone "%s"' % ( idnsname, type, data, zone, ) textui.print_name(self.name) textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_dashed('Added DNS resource record %s.' % output) api.register(dns_add_rr) class dns_del_rr(Command): """ Delete DNS resource record. """ takes_args = ( Str('zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), Str('idnsname', cli_name='resource', label=_('Resource name'), default_from=lambda zone: zone.lower(), attribute=True, ), StrEnum('type', label=_('Record type'), values=_record_types, ), Str('data', label=_('Data'), doc=_('Type-specific data'), ), ) has_output = standard_entry def execute(self, zone, idnsname, type, data, **options): ldap = self.api.Backend.ldap2 attr = ('%srecord' % type).lower() dns_container_exists(ldap) # build entry DN dn = _get_record_dn(ldap, zone, idnsname) # get resource entry with the record we're trying to delete (dn, entry_attrs) = ldap.get_entry(dn) # convert entry_attrs keys to lowercase entry_attrs = dict( (k.lower(), v) for (k, v) in entry_attrs.iteritems() ) # get new value for record attribute attr_value = entry_attrs.get(attr.lower(), []) try: attr_value.remove(data) except ValueError: raise errors.NotFound(reason=u'resource record not found') # check if it's worth to keep this entry in LDAP if 'idnszone' not in entry_attrs['objectclass']: # get a list of all meaningful record attributes record_attrs = [] for (k, v) in entry_attrs.iteritems(): if k.endswith('record') and v: record_attrs.append(k) # check if the list is empty if not record_attrs: # it's not ldap.delete_entry(dn) return dict(result={}, value=idnsname) ldap.update_entry(dn, {attr: attr_value}) # get entry with updated attribute for output (dn, entry_attrs) = ldap.get_entry(dn, ['idnsname', attr]) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, zone, idnsname, type, data, **options): output = '"%s %s %s" from zone "%s"' % ( idnsname, type, data, zone, ) entry_attrs = result['result'] textui.print_name(self.name) if entry_attrs: textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_dashed('Deleted DNS resource record %s' % output) api.register(dns_del_rr) class dns_find_rr(Command): """ Search for DNS resource records. """ takes_args = ( Str('zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), Str('criteria?', cli_name='criteria', label=_('Search criteria'), ), ) takes_options = ( Str('idnsname?', cli_name='resource', label=_('Resource name'), default_from=lambda zone: zone.lower(), ), StrEnum('type?', label=_('Record type'), values=_record_types, ), Str('data?', label=_('type-specific data'), ), ) has_output = standard_list_of_entries def execute(self, zone, term, **options): ldap = self.api.Backend.ldap2 if 'type' in options: attr = ('%srecord' % options['type']).lower() else: attr = None dns_container_exists(ldap) # build base dn for search base_dn = _get_zone_dn(ldap, zone) # build search keywords search_kw = {} if 'data' in options: if attr is not None: # user is looking for a certain record type search_kw[attr] = options['data'] else: # search in all record types for a in _record_default_attributes: search_kw[a] = term if 'idnsname' in options: idnsname = options['idnsname'] if idnsname == '@': search_kw['idnsname'] = zone else: search_kw['idnsname'] = idnsname # build search filter filter = ldap.make_filter(search_kw, rules=ldap.MATCH_ALL) if term: search_kw = {} for a in _record_default_attributes: search_kw[a] = term term_filter = ldap.make_filter(search_kw, exact=False) filter = ldap.combine_filters((filter, term_filter), ldap.MATCH_ALL) # select attributes we want to retrieve if options.get('all', False): attrs_list = ['*'] elif attr is not None: attrs_list = [attr] else: attrs_list = _record_default_attributes # get matching entries try: (entries, truncated) = ldap.find_entries( filter, attrs_list, base_dn ) except errors.NotFound: (entries, truncated) = (tuple(), False) # if the user is looking for a certain record type, don't display # entries that do not contain it if attr is not None: related_entries = [] for e in entries: entry_attrs = e[1] if attr in entry_attrs: related_entries.append(e) entries = related_entries for e in entries: e[1]['dn'] = e[0] entries = tuple(e for (dn, e) in entries) return dict(result=entries, count=len(entries), truncated=truncated) def output_for_cli(self, textui, result, zone, term, **options): entries = result['result'] truncated = result['truncated'] textui.print_name(self.name) for entry_attrs in entries: textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) textui.print_plain('') textui.print_count( len(entries), '%i DNS resource record matched.', '%i DNS resource records matched.' ) if truncated: textui.print_dashed('These results are truncated.', below=False) textui.print_dashed( 'Please refine your search and try again.', above=False ) api.register(dns_find_rr) class dns_show_rr(Command): """ Show existing DNS resource records. """ takes_args = ( Str('zone', label=_('Zone name'), normalizer=lambda value: value.lower(), ), Str('idnsname', cli_name='resource', label=_('Resource name'), normalizer=lambda value: value.lower(), ), ) has_output = standard_entry def execute(self, zone, idnsname, **options): # shows all records associated with resource ldap = self.api.Backend.ldap2 dns_container_exists(ldap) # build entry DN dn = _get_record_dn(ldap, zone, idnsname) # select attributes we want to retrieve if options.get('all', False): attrs_list = ['*'] else: attrs_list = _record_default_attributes (dn, entry_attrs) = ldap.get_entry(dn, attrs_list) entry_attrs['dn'] = dn return dict(result=entry_attrs, value=idnsname) def output_for_cli(self, textui, result, zone, idnsname, **options): entry_attrs = result['result'] textui.print_name(self.name) textui.print_attribute('dn', entry_attrs['dn']) del entry_attrs['dn'] textui.print_entry(entry_attrs) api.register(dns_show_rr) class dns_resolve(Command): """ Resolve a host name in DNS """ has_output = output.standard_value msg_summary = _('Found \'%(value)s\'') takes_args = ( Str('hostname', label=_('Hostname'), ), ) def execute(self, *args, **options): query=args[0] if query.find(api.env.domain) == -1 and query.find('.') == -1: query = '%s.%s.' % (query, api.env.domain) if query[-1] != '.': query = query + '.' rr = dnsclient.query(query, dnsclient.DNS_C_IN, dnsclient.DNS_T_A) self.log.debug('%s' % rr) if len(rr) == 0: raise errors.NotFound(reason=_('Host \'%(host)s\' not found' % {'host':query})) return dict(result=True, value=query) api.register(dns_resolve)