o MizK@sUdZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZmZddlmZddlmZddlmZddlmZd d gegd Zeed <d ZdZdZdZdZdZe e!Z"ddZ#ddZ$ddZ%dLddZ&dLddZ'de(de e(dee(e(fd e)fd!d"Z*dMd#d$Z+d%e(d&e(d dfd'd(Z,dNd)d*Z-d+d,Z.d e ee(ee(e(fee(ffd-d.Z/d/ee ee e(dee(e(fd0ee(e(fd1ee e(d2ed eee(f d3d4Z0d/eee(d eee(fd5d6Z1d/eee(d7e(dee(e(fd0ee(e(fd2ed eee(f d8d9Z2d:eee(d eee(fd;d<Z3d:eee(d dfd=d>Z4d?e)d@e)dAee(d dfdBdCZ5dLdDe e(d dfdEdFZ6dGe(dHed2edIe7d df dJdKZ8dS)Oz-Mounts: Configure mount points and swap filesN)DictListOptionalTuplecast) performancesubputil)Cloud)Config) MetaSchema) PER_INSTANCE cc_mountsall)iddistros frequencyactivate_by_schema_keysmetaz&^([x]{0,1}[shv]d[a-z][0-9]*|sr[0-9]+)$z^.+:.*z /etc/fstabzcomment=cloudconfigi@cCs8|dvrdSdD]}||r|ddkrdSqdS)N)amirootswapT) ephemeralebs:F) startswithfind)nameenumnamer! %sNz+Ignoring nonexistent default named mount %sFz&Device %s already defined in fstab: %sT)r<r=)rGrHrIr!r!r"sanitized_devname_is_valids rKc Cs`|dur td}|d}d||d}|dur|dur|}n2|rKt|}|j|j}||d<|dur>tt|d|}n||dkrJt|d}n|durQ|}||d<|dtkr^|}n|d tkridt}n t t |tt}t||}||d <i} | D]\} } t | trd | t| | <q| | | <qtd | d | d | d| d| d|S)Ntotalna)availmax_inmemrOg?maxsizez%s MBzCsuggest %s swap for %s memory with '%s' disk given max=%s [max=%s]'rQrP)r read_meminfor)statvfsf_frsizef_bfreeminintGBroundmathsqrtitems isinstanceMBr<r=) memsizemaxsizefsyssugg_maxinforWrOminsizerUpinfokvr!r!r"suggested_swapsizesN            rlfnamerUcsdfdd}tj|}t|t|ddkr0tddd|gtd d |gd krAtd krA|||d nz|||dWntjy^t d|||d Ynwtj |rkt |dz td|gWdStjyt |w)zSize is in MiB.z6Failed to create swapfile '%s' of size %sMB via %s: %sc std|||dkrddd||g}n|dkr%ddd|dd |g}ntd z tj|d d WdStjyP}zt||||t|d}~ww) Nz3Creating swapfile in '%s' on fstype '%s' using '%s' fallocatez-lz%sMddz if=/dev/zerozof=%szbs=1Mzcount=%sz:Missing dependency: 'dd' and 'fallocate' are not availableT)capture)r<r=rProcessExecutionErrorrgr del_file)rmrUmethodcmdeerrmsgfstyper!r" create_swaps4 z$create_swapfile..create_swapr;btrfstruncatez-s0chattrz+Cxfs)rRrornz4fallocate swap creation failed, will attempt with ddimkswapN)r)r*dirnamer ensure_dirget_mount_inforkernel_versionrqr<rgr+chmodrr)rmrUryswap_dirr!rvr"create_swapfiles0      rcCstj|}t|dkr2ztd}Wnty%t dYdSwt |t |||d}tt |d}|sCt ddSt dt||Wd|S1sYwY|S) z fname: full path string of filename to setup size: the size to create. set to "auto" for recommended maxsize: the maximum size autorLz)Not creating swap: failed to read meminfoN)rerdrcrz'Not creating swap: suggested size was 0zSetting up swap file)r)r*rr(lowerr rVIOErrorr<r=rrlr[rTimedr)rmrUrdrrcmibsizer!r!r"setup_swapfiles,        rc CsZt|ts tddS|dd}|dd}|dd}|r"|s)tddStj|rstjd s=td ||Sz$t d  D]}| |d rYtd ||WSqEtd |Wnt yrtd||YSwz t|tr|dkrt |}t|trt |}t|||dWSt y}z td|WYd}~dSd}~ww)zahandle the swap config, calling setup_swap if necessary. return None or (filename, size) z%input for swap config was not a dict.Nfilenamez /swap.imgrUrrdzno need to setup swapz /proc/swapsz:swap file %s exists, but no /proc/swaps exists, being safe zswap file %s already in usez+swap file %s exists, but not in /proc/swapsz.swap file %s exists. Error reading /proc/swapsr)rmrUrdzfailed to setup swap: %s)radictr<warningr?r=r)r*r+r load_text_file splitlinesr Exceptionr( human2bytesr)swapcfgrmrUrdlinerur!r!r"handle_swapcfg/sP              rcCslg}i}g}tjtr1ttD]}t|vr||q| }|r0|||d<||q|||fS)asParse /etc/fstab. Parse fstab, ignoring any lines containing "comment=cloudconfig". :return: A 3-tuple containing: - A list of lines exactly as they appear in fstab - A dictionary with key being the first token in the line and value being the entire line - A list of any lines that were ignored due to "comment=cloudconfig" r) r)r*r+ FSTAB_PATHr rr MNT_COMMENTappendr4) fstab_linesrI fstab_removedrtoksr!r!r" parse_fstab_s      rmountsdevice_aliasesdefault_fieldscloudc Csg}|D]Y}t|tstd|qt|d}t||j|d}t|||r0|g|dd} n|} t| D]\} } | durE|| | | <q6t| | | | <q6| |t | d7} | | q|S)aSanitize mounts to ensure we can work with devices in config. Specifically: - Ensure the mounts configuration is a list of lists - Transform and sanitize device names - Ensure all tokens are strings - Add default options to any lines without options z%Mount option not a list, ignoring: %srrCr;N) ralistr<rr(rFdevice_name_to_devicerK enumeratelenr) rrIrrr updated_linesrstartsanitized_devname updated_lineindextokenr!r!r"sanitize_mounts_configurationys&     rcCslg}g}|dddD]#}|ddus|d|vr)td|d||dq ||q |dddS)zRemove any entries that have a device name that doesn't exist. If the second field of a mount line is None (not the string, the value), we skip it along with any other entries that came before it that share the same device name. Nrr;rz$Skipping nonexistent device named %s)r<r=r)ractlist dev_denylistrr!r!r"remove_nonexistent_devicess rdefault_mount_optionsc st|}ddd|ddggdfD]5d}t||j|d}t|||s&qtt|d<tfd d |D}|rAt d |q| q|S) zAdd default mounts to the user provided mount configuration. Add them only if no other entry has the same device name r:z/mntrr|2)rnonerswr|r|rrc3s |] }|ddkVqdS)rNr!).0cfgm default_mountr!r" s z,add_default_mounts_to_cfg..z-Not including %s, already previously included) copydeepcopyrFrrKrr(anyr<r=r) rrrIrr new_mountsrrHdefault_already_existsr!rr"add_default_mounts_to_cfgs&     rrcCsdd|DS)z=Add "comment=cloudconfig" to the mount options of each entry.cSs8g|]}|dd|ddtg|ddqS)N,rR)rrentryr!r!r" s*zadd_comment..r!rr!r!r" add_commentsrcCs(tdd|DrtddgdSdS)z1Call 'swapon -a' if any entry has a swap fs type.css|] }|ddkVqdS)rMrNr!rr!r!r"rsz*activate_swap_if_needed..swapon-aN)rrrr!r!r"activate_swap_if_neededsr uses_systemd changes_madedirscCsfd}|rd}nddtD}tt||}|r/tddg|r1tddgd Sd Sd S) zCall 'mount -a' if needed. If changes were made, always call 'mount -a'. Otherwise, call 'mount -a' if any of the directories in the mount list are not already mounted. FTcSsh|] }d|vr|dqS) mountpointr!)rvalr!r!r" s z"mount_if_needed..mountr systemctlz daemon-reloadN)r rvaluesboolset differencer)rrrdo_mount mount_pointsr!r!r"mount_if_neededs rds_remove_entryc stjtsdStg}|r||ttd }|}Wdn1s&wYg}d}|D]tfdd|DrAd}q1|q1z(|rmttd }| |Wdn1s_wYt dtWdSWdSt y}z t d|WYd}~dSd}~ww) NrFc3s|]}|vVqdSNr!rrr!r"rsz cleanup_fstab..Twz%Removed resource disk entries from %sz4Failed to clean resource disk entries from fstab: %s)r)r*r+rrropen readlinesr writelinesr<rgrr)rremove_entriesflines new_lineschangedrur!rr" cleanup_fstabs4       rrcfgargsc s|j}|r dnd}ddd|ddg}|d|}|dg}td |t\} } } |d i} t|| | ||} t| || | |} t| } t | } t |d i}|r^| |d d d ddg| sgtddSdd| D}dd| D}|D]}zt |Wqwtyt td|Yqwwdd|Ddd| DfddDfddD}| |dd| }t t||rtd|ntdt| t|t||dS)z Handle the mounts configuration.z:defaults,nofail,x-systemd.after=cloud-init.service,_netdevzdefaults,nobootwaitNrr|rmount_default_fieldsrzmounts configuration is %srrrrz No modifications to fstab neededcSsg|]}d|qS) )r5rr!r!r"rKszhandle..cSs"g|] }|ddr|dqS)r;r2)r)rdr!r!r"rMs"z Failed to make '%s' config-mountcSg|]}|ddqSrrreplacernr!r!r"rUcSrrrrr!r!r"rVrcg|] }|vrd|qS)z- r!)rdrop)saddsr!r"rXscr)z+ r!)radd)sdropsr!r"rXsz%s  zChanges to fstab: %szNo changes to /etc/fstab made.)distrorr?r<r=rrrrrrrr rrlogexcextendr5 write_filerrrr)rrrrrrhardcoded_defaultsrrrrIrr updated_cfgswapfile cfg_linesrrsopscontentsr!)rrr"handle!s`            rr)NNN)NN)9__doc__rloggingr^r)r$typingrrrrr cloudinitrrr cloudinit.cloudr cloudinit.configr cloudinit.config.schemar cloudinit.settingsr r__annotations__r@r&rrrbr\ getLogger__name__r<r#r'r1r9rFr(rrKrlrrrrrrrrrrrrrr!r!r!r"s       1  9 >(0     ".     "$ "